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)