def test_data_messager(self): """Test for data_messager""" from popit.popit_json_importer import data_messager result = data_messager(self.input_data, "links") expected_result = {'birth_date': '1974-01-01', 'founding_date': '2000', 'death_date': '0000-00-00', 'end_date': '0000-00-00', 'url': 'http://www.testurl.com', 'html_url': 'http://www.htmlurl.com', 'area': {'id': '12345'}} self.assertEqual(result, expected_result) for item in ('url', 'html_url'): self.assertIn(item, result) result_no_links = data_messager(self.input_data, "area") for item in ('url', 'html_url'): self.assertNotIn(item, result_no_links) self.assertIn('area', result_no_links)
def test_create_entity(self, mock_data_m): """Test for create_entity""" from popit.popit_json_importer import data_messager, create_entity result = data_messager(self.input_data, "links") mock_data_m.return_value = e_data = {'birth_date': '1974-01-01', 'founding_date': '2000', 'death_date': '0000-00-00', 'end_date': '0000-00-00'} create_entity(self.input_data, "123", "birth_date", "test_api_key") expected_header = {'Apikey': 'test_api_key', 'Content-Type': 'application/json'} self.assertEqual(expected_header, result.headers) self.assertEqual(json.dumps(e_data), result.data)