def setUp(self): self.handlers = mock.MagicMock() self.settings = mock.MagicMock() self.database = {"database": DATABASE_NAME} self.base_handler = BaseHandler(self.handlers, self.settings, **self.database) self.database = mock.MagicMock() self.result = EMPLOYEE_DATA
class TestBaseHandler(unittest.TestCase): """ test case for BaseHandler """ def setUp(self): self.handlers = mock.MagicMock() self.settings = mock.MagicMock() self.database = {"database": DATABASE_NAME} self.base_handler = BaseHandler(self.handlers, self.settings, **self.database) self.database = mock.MagicMock() self.result = EMPLOYEE_DATA @mock.patch("sqlite3.connect") def test_initialize(self, mock_connect): """ test case for method initialize :param mock_connect: :return: """ self.base_handler.initialize(self.database) mock_connect.assert_called_once_with(self.database) def test_dict_builder(self): """ test case for method dict_builder :return: """ result = self.base_handler.dict_builder(self.result) self.assertEqual(result[0]['emp_id'], self.result[0][0]) self.assertEqual(result[0]['first_name'], self.result[0][1]) self.assertEqual(result[0]['zip'], self.result[0][-1]) self.assertEqual(len(result), 1) self.assertIsInstance(result, list) def test_employee_obj(self): """ test case for method employee_obj :return: """ result = self.base_handler.employee_obj(self.result) self.assertEqual(result[0].emp_id, self.result[0][0]) self.assertEqual(result[0].first_name, self.result[0][1]) self.assertEqual(result[0].zip, self.result[0][-1]) self.assertEqual(len(result), 1) self.assertIsInstance(result, list)