def setUp(self): # Static values self.expected_result = [{"field1": "value1"}] self.raw_result = [{"field1": "value2", "field2": 1}] # Subset result is subset of raw_result with keys in expected result self.subset_result = [{"field1": "value2"}] # Mocks self.mock_schema = MagicMock() ResultInterface.raw_result = MagicMock() ResultInterface.raw_result.return_value = self.raw_result self.config = ResultConfig() self.config.config = { "schema": self.mock_schema, "expected_result": self.expected_result, "match_type": "exact" } self.result = ResultInterface(config=self.config)
def setUp(self): # Static values self.expected_result = [{"field1": "value1"}] self.raw_result = [{"field1": "value2", "field2": 1}] # Subset result is subset of raw_result with keys in expected result self.subset_result = [{"field1": "value2"}] # Mocks self.mock_schema = MagicMock() ResultInterface.raw_result = MagicMock() ResultInterface.raw_result.return_value = self.raw_result self.config = ResultConfig() self.config.config = { "schema": self.mock_schema, "expected_result": self.expected_result, "match_type": "exact", } self.result = ResultInterface(config=self.config)
class ResultInterfaceTestCases(unittest.TestCase): """class for result tests""" def setUp(self): # Static values self.expected_result = [{"field1": "value1"}] self.raw_result = [{"field1": "value2", "field2": 1}] # Subset result is subset of raw_result with keys in expected result self.subset_result = [{"field1": "value2"}] # Mocks self.mock_schema = MagicMock() ResultInterface.raw_result = MagicMock() ResultInterface.raw_result.return_value = self.raw_result self.config = ResultConfig() self.config.config = { "schema": self.mock_schema, "expected_result": self.expected_result, "match_type": "exact" } self.result = ResultInterface(config=self.config) def test_actual_result_with_match_type_exact(self): """stuff""" self.config.config["match_type"] = 'exact' actual_result = self.result.actual_result() self.assertEqual(actual_result, self.raw_result) def test_actual_result_with_match_type_subset(self): """stuff""" self.config.config["match_type"] = 'subset' actual_result = self.result.actual_result() self.assertEqual(actual_result, self.subset_result) def test_actual_result_with_unsupported_match_type_raises_exception(self): """stuff""" self.config.config["match_type"] = 'NO TYPE' self.assertRaises(Exception, self.result.actual_result) def test_set_up_calls_schema_create_if_not_exists(self): """test result interface""" self.result.set_up() self.assertTrue(self.mock_schema.create_if_not_exists.called) def test_actual_and_expected_result(self): """stuff""" self.result.actual_result = MagicMock() stub_actual_result = 'ACTUAL RESULT' self.result.actual_result.return_value = stub_actual_result actual_and_expected_result = self.result.actual_and_expected_result() self.assertEqual(actual_and_expected_result, (stub_actual_result, self.expected_result)) def test_tear_down_calls_delete_if_exists(self): """makes sure teardown removes schema from datastore on teardown""" self.result.tear_down() self.assertTrue(self.mock_schema.delete_if_exists.called)
class ResultInterfaceTestCases(unittest.TestCase): """class for result tests""" def setUp(self): # Static values self.expected_result = [{"field1": "value1"}] self.raw_result = [{"field1": "value2", "field2": 1}] # Subset result is subset of raw_result with keys in expected result self.subset_result = [{"field1": "value2"}] # Mocks self.mock_schema = MagicMock() ResultInterface.raw_result = MagicMock() ResultInterface.raw_result.return_value = self.raw_result self.config = ResultConfig() self.config.config = { "schema": self.mock_schema, "expected_result": self.expected_result, "match_type": "exact", } self.result = ResultInterface(config=self.config) def test_actual_result_with_match_type_exact(self): """stuff""" self.config.config["match_type"] = "exact" actual_result = self.result.actual_result() self.assertEqual(actual_result, self.raw_result) def test_actual_result_with_match_type_subset(self): """stuff""" self.config.config["match_type"] = "subset" actual_result = self.result.actual_result() self.assertEqual(actual_result, self.subset_result) def test_actual_result_with_unsupported_match_type_raises_exception(self): """stuff""" self.config.config["match_type"] = "NO TYPE" self.assertRaises(Exception, self.result.actual_result) def test_set_up_calls_schema_create_if_not_exists(self): """test result interface""" self.result.set_up() self.assertTrue(self.mock_schema.create_if_not_exists.called) def test_actual_and_expected_result(self): """stuff""" self.result.actual_result = MagicMock() stub_actual_result = "ACTUAL RESULT" self.result.actual_result.return_value = stub_actual_result actual_and_expected_result = self.result.actual_and_expected_result() self.assertEqual(actual_and_expected_result, (stub_actual_result, self.expected_result)) def test_tear_down_calls_delete_if_exists(self): """makes sure teardown removes schema from datastore on teardown""" self.result.tear_down() self.assertTrue(self.mock_schema.delete_if_exists.called)