def test_get_api_data(self): """Test if data is retrived correctly""" test_site = "http://www.usa.gov/api/USAGovAPI/contacts.json" test_site += "/contact/48005" test_cache = "tests/fixtures/cassettes/test_cache" data = usa_layer.get_api_data(url=test_site, cache=test_cache) self.assertTrue("Census" in data.keys())
def test_patch_yamls(self): """ Test if yaml files are correctly updated """ test_site = "http://www.usa.gov/api/USAGovAPI/contacts.json" test_site += "/contact/48005" test_cache = "tests/fixtures/cassettes/test_cache" data = usa_layer.get_api_data(url=test_site, cache=test_cache) data.update({'Commerce': {'usa_id': '1111'}}) patcher = usa_layer.patch_yamls( data=data, directory="tests/fixtures/cassettes/test_yaml.yaml") for patched_yaml in patcher: self.assertTrue('usa_id' in patched_yaml[0]) self.assertTrue('usa_id' in patched_yaml[0].get('departments')[0]) self.assertTrue( 'description' in patched_yaml[0].get('departments')[0])
def test_patch_yamls(self): """ Test if yaml files are correctly updated """ test_site = "http://www.usa.gov/api/USAGovAPI/contacts.json" test_site += "/contact/48005" test_cache = "tests/fixtures/cassettes/test_cache" data = usa_layer.get_api_data(url=test_site, cache=test_cache) data.update({'Commerce': {'usa_id': '1111'}}) patcher = usa_layer.patch_yamls( data=data, directory="tests/fixtures/cassettes/test_yaml.yaml") for patched_yaml in patcher: self.assertTrue( 'usa_id' in patched_yaml[0]) self.assertTrue( 'usa_id' in patched_yaml[0].get('departments')[0]) self.assertTrue( 'description' in patched_yaml[0].get('departments')[0])