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 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 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 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 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)