"""Test Network model.""" import pytest from podcast.models import network from podcast.utils.exc import OwsError from tests.utils import db_operations def test_network_model(): """Test network model.""" network_name = 'Test network' network_model = network.Network(name=network_name) assert network_model.name == network_name def test_get_networks(): """Test get all the available networks.""" result = network.get_networks() assert result assert len(result['items']) == len(db_operations.network_data) def test_search_networks(): """Test search across networks.""" result = network.search_networks('Broccoli', [1, 2]) assert len(result) == 1 result_network = result[0] assert result_network['id'] == db_operations.network_data[0]['id'] assert result_network['name'] == db_operations.network_data[0]['name'] def test_get_megaphone_url(): """Test get megaphone url.""" result = network.get_megaphone_url(2) assert result == 'https://cms.megaphone.fm/api/networks/74756b62-7e5c-4728-895a-2f42f8969e15' def test_get_megaphone_url_not_found(): """Test get megaphone url raises .""" with pytest.raises(OwsError) as e: network.get_megaphone_url(200) assert e.value.message == 'Network not found'