class TestUserSettings(OsfTestCase):
ADDONS_UNDER_TEST = {
MockOAuth2Provider.short_name: {
'user_settings': MockUserSettings,
'node_settings': MockNodeSettings,
}
}
@classmethod
def setUpClass(cls):
super(TestUserSettings, cls).setUpClass()
@classmethod
def tearDownClass(cls):
super(TestUserSettings, cls).tearDownClass()
def setUp(self):
super(TestUserSettings, self).setUp()
self.user = AuthUserFactory()
self.user_settings = self.user.get_or_add_addon(
MockUserSettings.oauth_provider.short_name)
self.external_account = ExternalAccountFactory()
self.user.external_accounts.append(self.external_account)
self.user.save()
self.project = ProjectFactory(creator=self.user)
def tearDown(self):
super(TestUserSettings, self).tearDown()
def test_connected_accounts_empty(self):
self.user.external_accounts = []
assert_equal(self.user_settings.external_accounts, [])
def test_connected_accounts(self):
assert_equal(self.user_settings.external_accounts,
[self.external_account])
def test_verify_false_no_grants(self):
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_false_with_grants(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=ProjectFactory())
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_false_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'baz': 'qiz'}))
def test_verify_true(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_true_with_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}))
def test_grant(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_equal(self.user_settings.oauth_grants,
{self.project._id: {
self.external_account._id: {}
}})
def test_grant_not_owned(self):
self.user.external_accounts = []
with assert_raises(PermissionsError):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_equal(self.user_settings.oauth_grants, {})
def test_grant_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_equal(
self.user_settings.oauth_grants,
{self.project._id: {
self.external_account._id: {
'foo': 'bar'
}
}})
@mock.patch(
'tests.test_addons_oauth.MockUserSettings.revoke_remote_oauth_access')
@mock.patch('framework.auth.core._get_current_user')
def test_revoke_remote_access_called(self, mock_decorator, mock_revoke):
mock_decorator.return_value = self.user
self.user_settings.delete()
assert_equal(mock_revoke.call_count, 1)
@mock.patch(
'tests.test_addons_oauth.MockUserSettings.revoke_remote_oauth_access')
@mock.patch('framework.auth.core._get_current_user')
def test_revoke_remote_access_not_called(self, mock_decorator,
mock_revoke):
mock_decorator.return_value = self.user
user2 = AuthUserFactory()
user2.external_accounts.append(self.external_account)
user2.save()
self.user_settings.delete()
assert_equal(mock_revoke.call_count, 0)
def test_on_delete(self):
node_settings = self.project.get_or_add_addon(
MockUserSettings.oauth_provider.short_name,
auth=Auth(user=self.user))
node_settings.set_auth(external_account=self.external_account,
user=self.user)
self.user.delete_addon(MockUserSettings.oauth_provider.short_name)
node_settings.reload()
assert_is_none(node_settings.external_account)
assert_is_none(node_settings.user_settings)
class TestUserSettings(OsfTestCase):
ADDONS_UNDER_TEST = {
MockOAuth2Provider.short_name: {
'user_settings': MockUserSettings,
'node_settings': MockNodeSettings,
}
}
@classmethod
def setUpClass(cls):
super(TestUserSettings, cls).setUpClass()
@classmethod
def tearDownClass(cls):
super(TestUserSettings, cls).tearDownClass()
def setUp(self):
super(TestUserSettings, self).setUp()
self.user = AuthUserFactory()
self.user_settings = self.user.get_or_add_addon(
MockUserSettings.oauth_provider.short_name
)
self.external_account = ExternalAccountFactory()
self.user.external_accounts.append(self.external_account)
self.user.save()
self.project = ProjectFactory(creator=self.user)
def tearDown(self):
super(TestUserSettings, self).tearDown()
def test_connected_accounts_empty(self):
self.user.external_accounts = []
assert_equal(
self.user_settings.external_accounts,
[]
)
def test_connected_accounts(self):
assert_equal(
self.user_settings.external_accounts,
[self.external_account]
)
def test_verify_false_no_grants(self):
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project
)
)
def test_verify_false_with_grants(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=ProjectFactory()
)
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project
)
)
def test_verify_false_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}
)
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'baz': 'qiz'}
)
)
def test_verify_true(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project
)
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project
)
)
def test_verify_true_with_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}
)
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}
)
)
def test_grant(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project
)
assert_equal(
self.user_settings.oauth_grants,
{
self.project._id: {
self.external_account._id: {}
}
}
)
def test_grant_not_owned(self):
self.user.external_accounts = []
with assert_raises(PermissionsError):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project
)
assert_equal(
self.user_settings.oauth_grants,
{}
)
def test_grant_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}
)
assert_equal(
self.user_settings.oauth_grants,
{
self.project._id: {
self.external_account._id: {'foo': 'bar'}
}
}
)
def test_on_delete(self):
node_settings = self.project.get_or_add_addon(
MockUserSettings.oauth_provider.short_name,
auth=Auth(user=self.user)
)
node_settings.set_auth(
external_account=self.external_account,
user=self.user
)
self.user.delete_addon(
MockUserSettings.oauth_provider.short_name
)
node_settings.reload()
assert_is_none(node_settings.external_account)
assert_is_none(node_settings.user_settings)
class TestUserSettings(OsfTestCase):
@classmethod
def setUpClass(cls):
init_mock_addon()
super(TestUserSettings, cls).setUpClass()
@classmethod
def tearDownClass(cls):
remove_mock_addon()
super(TestUserSettings, cls).tearDownClass()
def setUp(self):
super(TestUserSettings, self).setUp()
self.user = AuthUserFactory()
self.user_settings = self.user.get_or_add_addon(
MockUserSettings.oauth_provider.short_name)
self.external_account = ExternalAccountFactory()
self.user.external_accounts.append(self.external_account)
self.user.save()
self.project = ProjectFactory(creator=self.user)
def tearDown(self):
super(TestUserSettings, self).tearDown()
def test_connected_accounts_empty(self):
self.user.external_accounts = []
assert_equal(self.user_settings.external_accounts, [])
def test_connected_accounts(self):
assert_equal(self.user_settings.external_accounts,
[self.external_account])
def test_verify_false_no_grants(self):
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_false_with_grants(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=ProjectFactory())
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_false_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_false(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'baz': 'qiz'}))
def test_verify_true(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account, node=self.project))
def test_verify_true_with_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_true(
self.user_settings.verify_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'}))
def test_grant(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_equal(self.user_settings.oauth_grants,
{self.project._id: {
self.external_account._id: {}
}})
def test_grant_not_owned(self):
self.user.external_accounts = []
with assert_raises(PermissionsError):
self.user_settings.grant_oauth_access(
external_account=self.external_account, node=self.project)
assert_equal(self.user_settings.oauth_grants, {})
def test_grant_metadata(self):
self.user_settings.grant_oauth_access(
external_account=self.external_account,
node=self.project,
metadata={'foo': 'bar'})
assert_equal(
self.user_settings.oauth_grants,
{self.project._id: {
self.external_account._id: {
'foo': 'bar'
}
}})
def test_on_delete(self):
node_settings = self.project.get_or_add_addon(
MockUserSettings.oauth_provider.short_name,
auth=Auth(user=self.user))
node_settings.set_auth(external_account=self.external_account,
user=self.user)
self.user.delete_addon(MockUserSettings.oauth_provider.short_name)
node_settings.reload()
assert_is_none(node_settings.external_account)
assert_is_none(node_settings.user_settings)
