def test_update_token(self):
client = ApiClient(user_name='test_username', password='******')
client._request = unittest.mock.MagicMock(name='_request', return_value=MockResponse())
client._update_token()
client._request.assert_called_with('zr/token/result/', Method.GET, query=unittest.mock.ANY,
headers=unittest.mock.ANY, default_headers=False)
def test_login_mock(self):
client = ApiClient(user_name='test_username', password='******')
client._session = MockSession()
client._update_token = unittest.mock.MagicMock(name='_update_token')
client._request = unittest.mock.MagicMock(name='_request', return_value=MockResponse())
client.login()
self.assertEqual(2, client._request.call_count)
self.assertIsNotNone(client.state.logged_in_account_data)
