def test_account_return(self): account_data = {"id": 1, "name": "TestName"} mock_data_interface = Mock() mock_data_interface.get.return_value = account_data account = Account(mock_data_interface) self.assertDictEqual(account_data, account.get_account(1))
def test_account_return_exception(self): mock_data_interface = Mock() mock_data_interface.get.side_effect = ConnectionError() account = Account(mock_data_interface) self.assertEqual('Connection error occurred. Try Again.', account.get_account(1))