def setUp(self): super(TestGoogleDriveConfigViews, self).setUp() self.user = AuthUserFactory() self.user.add_addon('googledrive') self.user_settings = self.user.get_addon('googledrive') self.project = ProjectFactory(creator=self.user) self.project.add_addon('googledrive', Auth(self.user)) self.node_settings = self.project.get_addon('googledrive') self.node_settings.user_settings = self.user_settings oauth_settings = GoogleDriveOAuthSettingsFactory() oauth_settings.save() self.user_settings.oauth_settings = oauth_settings self.node_settings.save() # Log user in self.app.authenticate(*self.user.auth)
def setUp(self): super(TestGoogleDriveNodeSettingsModel, self).setUp() self.user = UserFactory() self.user.add_addon('googledrive') self.user.save() self.user_settings = self.user.get_addon('googledrive') oauth_settings = GoogleDriveOAuthSettingsFactory() oauth_settings.save() self.user_settings.oauth_settings = oauth_settings self.user_settings.save() self.project = ProjectFactory() self.node_settings = GoogleDriveNodeSettingsFactory( user_settings=self.user_settings, owner=self.project, )
def test_get_targets(self): GoogleDriveUserSettings.remove() addons = [ GoogleDriveUserSettings(), GoogleDriveUserSettings( oauth_settings=GoogleDriveOAuthSettingsFactory()), ] for addon in addons: addon.save() targets = get_targets() assert_equal(targets.count(), 1) assert_equal(targets[0]._id, addons[-1]._id)
def setUp(self): super(TestGoogleDriveAuthViews, self).setUp() self.user = AuthUserFactory() self.user.add_addon('googledrive') self.project = ProjectFactory(creator=self.user) self.project.add_addon('googledrive', Auth(self.user)) self.node_settings = self.project.get_addon('googledrive') self.user_settings = self.user.get_addon('googledrive') oauth_settings = GoogleDriveOAuthSettingsFactory() self.user_settings.oauth_settings = oauth_settings self.node_settings.user_settings = self.user_settings # Log user in self.app.authenticate(*self.user.auth) self.flow = mock.Mock() self.credentials = mock.Mock()
def test_migration_no_project(self): user = UserFactory() user.add_addon('googledrive') user_addon = user.get_addon('googledrive') user_addon.oauth_settings = GoogleDriveOAuthSettingsFactory() user_addon.save() do_migration([user_addon]) user_addon.reload() assert_is_none(user_addon.oauth_settings) assert_equal(len(user.external_accounts), 1) account = user.external_accounts[0] assert_equal(account.provider, 'googledrive') assert_equal(account.oauth_key, 'abcdef1')
def setUp(self): super(TestGoogleDriveUtils, self).setUp() self.user = AuthUserFactory() self.user.add_addon('googledrive') self.project = ProjectFactory(creator=self.user) self.project.add_addon('googledrive', Auth(self.user)) self.node_settings = self.project.get_addon('googledrive') self.user_settings = self.user.get_addon('googledrive') oauth_settings = GoogleDriveOAuthSettingsFactory() self.user_settings.oauth_settings = oauth_settings self.node_settings.user_settings = self.user_settings self.node_settings.folder_id = '09120912' self.node_settings.folder_path = 'foo/bar' self.user_settings.save() self.node_settings.save() # Log user in self.app.authenticate(*self.user.auth)
def test_migration_removes_targets(self): GoogleDriveUserSettings.remove() user = UserFactory() project = ProjectFactory(creator=user) user.add_addon('googledrive', auth=Auth(user)) user_addon = user.get_addon('googledrive') user_addon.oauth_settings = GoogleDriveOAuthSettingsFactory() user_addon.save() project.add_addon('googledrive', auth=Auth(user)) node_addon = project.get_addon('googledrive') node_addon.foreign_user_settings = user_addon node_addon.save() assert_equal(get_targets().count(), 1) do_migration([user_addon]) user_addon.reload() assert_equal(get_targets().count(), 0)
def test_migration_multiple_users(self): user1 = UserFactory() user2 = UserFactory() oauth_settings = GoogleDriveOAuthSettingsFactory() user1.add_addon('googledrive') user1_addon = user1.get_addon('googledrive') user1_addon.oauth_settings = oauth_settings user1_addon.save() user2.add_addon('googledrive') user2_addon = user2.get_addon('googledrive') user2_addon.oauth_settings = oauth_settings user2_addon.save() do_migration([user1_addon, user2_addon]) user1_addon.reload() user2_addon.reload() assert_equal( user1.external_accounts[0], user2.external_accounts[0], )
def setUp(self): super(TestGoogleDriveUtils, self).setUp() self.user = AuthUserFactory() self.user.add_addon('googledrive') self.project = ProjectFactory(creator=self.user) self.project.add_addon('googledrive', Auth(self.user)) self.node_settings = self.project.get_addon('googledrive') self.user_settings = self.user.get_addon('googledrive') oauth_settings = GoogleDriveOAuthSettingsFactory() self.user_settings.oauth_settings = oauth_settings self.node_settings.user_settings = self.user_settings self.node_settings.folder_id = '09120912' self.node_settings.folder_path = 'foo/bar' self.user_settings.save() self.node_settings.save() # Log user in self.app.authenticate(*self.user.auth) self.patcher = mock.patch( 'website.addons.googledrive.model.GoogleDriveUserSettings.fetch_access_token' ) self.patcher.return_value = 'fakeaccesstoken' self.patcher.start()