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)