class MyTestCase(unittest.TestCase):
""" Simple Test """
def setUp(self):
""" Hook method for setting up the test fixture before exercising it. """
self.account_data = {"id": "1", "name": "test"}
mock_data_interface = Mock()
mock_data_interface.get_id.return_value = '1'
mock_data_interface.get_name.return_value = 'test'
mock_data_interface.get.side_effect = AccountException()
self.account = Account(mock_data_interface)
def test_account_returns_data_for_id(self):
""" Simple Test for ID """
self.assertEqual(self.account_data["id"],
self.account.get_account_id())
def test_account_returns_data_for_name(self):
""" Simple Test for NAME """
self.assertEqual(self.account_data["name"],
self.account.get_account_name())
def test_account_when_connect_exception_raised(self):
""" Raises Exception """
self.assertEqual("Connection error occurred",
self.account.get_account())
def test_create_account_object_with_params(self):
initID = "001"
initBalance = 500
account = Account(initID, initBalance)
self.assertEqual(initID, account.get_account_id())
self.assertEqual(initBalance, account.get_account_balance())
