def setUp(self): self.return_value = [ Gazette("4205902", date.today(), "https://queridodiario.ok.org.br/"), Gazette("4202909", date.today(), "https://queridodiario.ok.org.br/"), ] self.mock_data_gateway = MagicMock() self.mock_data_gateway.get_gazettes = MagicMock( return_value=self.return_value) self.gazette_access = GazetteAccess(self.mock_data_gateway)
class GazetteAccessTest(TestCase): def setUp(self): self.return_value = [ Gazette("4205902", date.today(), "https://queridodiario.ok.org.br/"), Gazette("4202909", date.today(), "https://queridodiario.ok.org.br/"), ] self.mock_data_gateway = MagicMock() self.mock_data_gateway.get_gazettes = MagicMock( return_value=self.return_value) self.gazette_access = GazetteAccess(self.mock_data_gateway) def test_create_gazette_access(self): self.assertIsNotNone(self.gazette_access, msg="Could not create GazetteAccess object") self.assertIsInstance( self.gazette_access, GazetteAccessInterface, msg= "The GazetteAccess object should implement GazetteAccessInterface", ) def test_get_gazettes(self): self.assertEqual(2, len(list(self.gazette_access.get_gazettes()))) self.mock_data_gateway.get_gazettes.assert_called_once() def test_get_gazettes_should_return_dictionary(self): expected_results = [ { "territory_id": "4205902", "date": date.today(), "url": "https://queridodiario.ok.org.br/", }, { "territory_id": "4202909", "date": date.today(), "url": "https://queridodiario.ok.org.br/", }, ] gazettes = self.gazette_access.get_gazettes() self.assertCountEqual(expected_results, gazettes) def test_should_foward_filter_to_gateway(self): gazette_access = GazetteAccess(self.mock_data_gateway) list( self.gazette_access.get_gazettes(filters=GazetteRequest( territory_id="4205902"))) self.mock_data_gateway.get_gazettes.assert_called_once_with( territory_id="4205902")
def test_should_foward_filter_to_gateway(self): gazette_access = GazetteAccess(self.mock_data_gateway) list( self.gazette_access.get_gazettes(filters=GazetteRequest( territory_id="4205902"))) self.mock_data_gateway.get_gazettes.assert_called_once_with( territory_id="4205902")
def test_should_foward_page_fields_filter_to_gateway(self): gazette_access = GazetteAccess(self.mock_data_gateway, self.mock_database_gateway) list( self.gazette_access.get_gazettes( filters=GazetteRequest(offset=10, size=100))[1]) self.mock_data_gateway.get_gazettes.assert_called_once_with( until=None, since=None, territory_id=None, querystring=None, offset=10, size=100, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="relevance", )
def test_should_foward_until_date_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) list( self.gazette_access.get_gazettes( filters=GazetteRequest(until=date.today()) )[1] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( until=date.today(), since=None, territory_id=None, keywords=None, offset=0, size=10, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", )
def setUp(self): self.return_value = [ Gazette( "4205902", date.today(), "https://queridodiario.ok.org.br/", "so'jsdogjeogjsdogjheogdfsdf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4205902", (date.today() - timedelta(days=1)), "https://queridodiario.ok.org.br/", "a;oijaeofdjewofijrogho490jhfeasd", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4205902", (date.today() + timedelta(days=1)), "https://queridodiario.ok.org.br/", "eolgpijdsfesuhrgfiuhsad", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", date.today(), "https://queridodiario.ok.org.br/", "ew;oigfdfsdjn;dajnorgf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", (date.today() - timedelta(days=1)), "https://queridodiario.ok.org.br/", "sdosauiydhbfeicneqiudnewf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", (date.today() + timedelta(days=1)), "https://queridodiario.ok.org.br/", "sdo;ifjwefonsdiasndiswabdisbfnidf", "My city", "My state", "highlight" "123,456", False, ), ] self.mock_data_gateway = MagicMock() self.mock_data_gateway.get_gazettes = MagicMock( return_value=(len(self.return_value), self.return_value) ) self.database_data = [ City( "Taquarinha Do Norte", "1235", "RN", OpennessLevel("1"), ["https://somewebsite.org", "https://anotherwebsite.org"], ), City( "Taquarinha Do Sul", "1236", "RS", OpennessLevel("3"), ["https://somewebsite.org", "https://anotherwebsite.org"], ), ] self.mock_database_gateway = MagicMock() self.mock_database_gateway.get_cities = MagicMock( return_value=self.database_data ) self.mock_database_gateway.get_city = MagicMock( return_value=self.database_data[0] ) self.gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway )
class GazetteAccessTest(TestCase): def setUp(self): self.return_value = [ Gazette( "4205902", date.today(), "https://queridodiario.ok.org.br/", "so'jsdogjeogjsdogjheogdfsdf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4205902", (date.today() - timedelta(days=1)), "https://queridodiario.ok.org.br/", "a;oijaeofdjewofijrogho490jhfeasd", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4205902", (date.today() + timedelta(days=1)), "https://queridodiario.ok.org.br/", "eolgpijdsfesuhrgfiuhsad", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", date.today(), "https://queridodiario.ok.org.br/", "ew;oigfdfsdjn;dajnorgf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", (date.today() - timedelta(days=1)), "https://queridodiario.ok.org.br/", "sdosauiydhbfeicneqiudnewf", "My city", "My state", "highlight" "123,456", False, ), Gazette( "4202909", (date.today() + timedelta(days=1)), "https://queridodiario.ok.org.br/", "sdo;ifjwefonsdiasndiswabdisbfnidf", "My city", "My state", "highlight" "123,456", False, ), ] self.mock_data_gateway = MagicMock() self.mock_data_gateway.get_gazettes = MagicMock( return_value=(len(self.return_value), self.return_value) ) self.database_data = [ City( "Taquarinha Do Norte", "1235", "RN", OpennessLevel("1"), ["https://somewebsite.org", "https://anotherwebsite.org"], ), City( "Taquarinha Do Sul", "1236", "RS", OpennessLevel("3"), ["https://somewebsite.org", "https://anotherwebsite.org"], ), ] self.mock_database_gateway = MagicMock() self.mock_database_gateway.get_cities = MagicMock( return_value=self.database_data ) self.mock_database_gateway.get_city = MagicMock( return_value=self.database_data[0] ) self.gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) def test_create_gazette_access(self): self.assertIsNotNone( self.gazette_access, msg="Could not create GazetteAccess object" ) self.assertIsInstance( self.gazette_access, GazetteAccessInterface, msg="The GazetteAccess object should implement GazetteAccessInterface", ) def test_get_gazettes(self): items_count, gazettes = self.gazette_access.get_gazettes() self.assertEqual(items_count, len(self.return_value)) self.assertEqual(len(self.return_value), len(gazettes)) self.mock_data_gateway.get_gazettes.assert_called_once() def test_get_cities(self): cities = self.gazette_access.get_cities() self.assertEqual(len(self.database_data), len(cities)) self.mock_database_gateway.get_cities.assert_called_once() self.assertCountEqual([vars(city) for city in self.database_data], cities) def test_get_city(self): city = self.gazette_access.get_city() self.assertEqual(vars(self.database_data[0]), city) self.mock_database_gateway.get_city.assert_called_once() def test_get_gazettes_should_return_dictionary(self): expected_results = [ { "territory_id": gazette.territory_id, "date": gazette.date, "url": gazette.url, "checksum": gazette.checksum, "territory_name": gazette.territory_name, "state_code": gazette.state_code, "edition": gazette.edition, "is_extra_edition": gazette.is_extra_edition, "highlight_texts": gazette.highlight_texts, "file_raw_txt": gazette.file_raw_txt, } for gazette in self.return_value ] _, gazettes = self.gazette_access.get_gazettes() self.assertCountEqual(expected_results, gazettes) def test_should_foward_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) list( self.gazette_access.get_gazettes( filters=GazetteRequest(territory_id="4205902") )[1] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( territory_id="4205902", since=None, until=None, keywords=None, offset=0, size=10, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", ) def test_should_foward_since_date_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) list( self.gazette_access.get_gazettes( filters=GazetteRequest(since=date.today()) )[1] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( since=date.today(), until=None, territory_id=None, keywords=None, offset=0, size=10, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", ) def test_should_foward_until_date_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) list( self.gazette_access.get_gazettes( filters=GazetteRequest(until=date.today()) )[1] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( until=date.today(), since=None, territory_id=None, keywords=None, offset=0, size=10, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", ) def test_should_foward_keywords_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) keywords = ["foo", "bar", "zpto"] list( self.gazette_access.get_gazettes(filters=GazetteRequest(keywords=keywords))[ 1 ] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( until=None, since=None, territory_id=None, keywords=keywords, offset=0, size=10, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", ) def test_should_foward_page_fields_filter_to_gateway(self): gazette_access = GazetteAccess( self.mock_data_gateway, self.mock_database_gateway ) list( self.gazette_access.get_gazettes( filters=GazetteRequest(offset=10, size=100) )[1] ) self.mock_data_gateway.get_gazettes.assert_called_once_with( until=None, since=None, territory_id=None, keywords=None, offset=10, size=100, fragment_size=150, number_of_fragments=1, pre_tags=[""], post_tags=[""], sort_by="descending_date", )