def test_fetch_access_token_with_token_expired(self, mock_refresh): external_account = GoogleDriveAccountFactory() self.user.external_accounts.append(external_account) external_account.expires_at = datetime.utcnow() + relativedelta.relativedelta(minutes=4) external_account.oauth_key = 'fake-token' external_account.refresh_token = 'refresh-fake-token' external_account.save() fake_token = { 'access_token': 'new-access-token', 'refresh_token': 'new-refresh-token', 'expires_at': 1234.5 } mock_refresh.return_value = fake_token self.node_settings.set_auth(external_account, self.user) self.node_settings.fetch_access_token() mock_refresh.assert_called_once() assert_equal(external_account.oauth_key, 'new-access-token') assert_equal(external_account.refresh_token, 'new-refresh-token') assert_equal(external_account.expires_at, datetime.utcfromtimestamp(1234.5))
def test_fetch_access_token_with_token_expired(self, mock_refresh): external_account = GoogleDriveAccountFactory() self.user.external_accounts.append(external_account) external_account.expires_at = datetime.utcnow( ) + relativedelta.relativedelta(minutes=4) external_account.oauth_key = 'fake-token' external_account.refresh_token = 'refresh-fake-token' external_account.save() fake_token = { 'access_token': 'new-access-token', 'refresh_token': 'new-refresh-token', 'expires_at': 1234.5 } mock_refresh.return_value = fake_token self.node_settings.set_auth(external_account, self.user) self.node_settings.fetch_access_token() mock_refresh.assert_called_once() assert_equal(external_account.oauth_key, 'new-access-token') assert_equal(external_account.refresh_token, 'new-refresh-token') assert_equal(external_account.expires_at, datetime.utcfromtimestamp(1234.5))