Python Account.get_account Examples

Example #1

File: mock_test.py Project: justupgrade/python

    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))

Example #2

File: mock_test.py Project: justupgrade/python

 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))