def test_access_token_refreshes_timeout_longer(self, mock_refresh):
mock_refresh.return_value = {
'access_token': 'abc',
'refresh_token': '123',
'expires_at': time.time(),
}
user_settings = GoogleDriveUserSettingsFactory()
user_settings.expires_at = datetime.utcnow() + relativedelta.relativedelta(minutes=4)
user_settings.access_token
mock_refresh.assert_called_once()
def test_access_token_refreshes(self, mock_needs_refresh, mock_refresh):
mock_refresh.return_value = {
'access_token': 'abc',
'refresh_token': '123',
'expires_at': time.time(),
}
user_settings = GoogleDriveUserSettingsFactory()
user_settings.expires_at = datetime.now()
user_settings.access_token
mock_refresh.assert_called_once()
def test_access_token_checks(self, mock_needs_refresh):
mock_needs_refresh.return_value = False
user_settings = GoogleDriveUserSettingsFactory()
user_settings.access_token
mock_needs_refresh.assert_called_once()
def test_clear_wo_oauth_settings(self, mock_revoke_access_token):
user_settings = GoogleDriveUserSettingsFactory(access_token='abcde')
user_settings.oauth_settings = None
user_settings.save()
node_settings = GoogleDriveNodeSettingsFactory.build()
node_settings.user_settings = user_settings
node_settings.save()
assert_false(user_settings.oauth_settings)
user_settings.clear()
user_settings.save()
assert_false(mock_revoke_access_token.called)
def test_fields(self):
user_settings = GoogleDriveUserSettingsFactory()
retrieved = GoogleDriveUserSettings.load(user_settings._primary_key)
assert_true(retrieved.owner)
assert_true(retrieved.username)
assert_true(retrieved.expires_at)
assert_true(retrieved.access_token)
def setUp(self):
super(TestNodeSettingsCallbacks, self).setUp()
# Create node settings with auth
self.user_settings = GoogleDriveUserSettingsFactory(access_token='123abc', username='******')
self.node_settings = GoogleDriveNodeSettingsFactory(
user_settings=self.user_settings,
folder='',
)
self.project = self.node_settings.owner
self.user = self.user_settings.owner
def test_delete(self):
user_settings = GoogleDriveUserSettingsFactory()
assert_true(user_settings.has_auth)
user_settings.delete()
user_settings.save()
assert_false(user_settings.access_token)
assert_true(user_settings.deleted)
def test_clear(self):
node_settings = GoogleDriveNodeSettingsFactory.build()
user_settings = GoogleDriveUserSettingsFactory(access_token='abcde')
node_settings.user_settings = user_settings
node_settings.save()
assert_true(user_settings.access_token)
user_settings.clear()
user_settings.save()
assert_false(user_settings.access_token)
def test_clear_clears_associated_node_settings(self):
node_settings = GoogleDriveNodeSettingsFactory.build()
user_settings = GoogleDriveUserSettingsFactory()
node_settings.user_settings = user_settings
node_settings.save()
user_settings.clear()
user_settings.save()
# Node settings no longer associated with user settings
assert_is(node_settings.folder_id, None)
assert_is(node_settings.user_settings, None)
def test_set_user_auth(self):
node_settings = GoogleDriveNodeSettingsFactory()
user_settings = GoogleDriveUserSettingsFactory()
node_settings.set_user_auth(user_settings)
node_settings.save()
assert_true(node_settings.has_auth)
assert_equal(node_settings.user_settings, user_settings)
# A log was saved
last_log = node_settings.owner.logs[-1]
log_params = last_log.params
assert_equal(last_log.user, user_settings.owner)
assert_equal(log_params['folder'], node_settings.folder_path)
assert_equal(last_log.action, 'googledrive_node_authorized')
assert_equal(log_params['node'], node_settings.owner._primary_key)
def test_access_token_doesnt_refresh(self, mock_refresh):
user_settings = GoogleDriveUserSettingsFactory()
user_settings.save()
user_settings.access_token
assert_false(mock_refresh.called)
def test_has_auth(self):
user_settings = GoogleDriveUserSettingsFactory(access_token=None)
assert_false(user_settings.has_auth)
user_settings.access_token = '12345'
user_settings.save()
assert_true(user_settings.has_auth)
