def test_make_new_node_settings(self):
node_settings_document = self.node_settings_documents[0]
node = Node.load(node_settings_document['owner'])
user_settings_document = database['addons3usersettings'].find_one({
'_id': node_settings_document['user_settings']
})
external_account, user, new = migration.migrate_to_external_account(
user_settings_document
)
user_settings = migration.make_new_user_settings(user)
node_settings = migration.make_new_node_settings(
node,
node_settings_document,
external_account,
user_settings
)
assert(
'addons3nodesettings' not in node._backrefs['addons']
)
assert_equal(
len(node._backrefs['addons']['s3nodesettings']['owner']),
1
)
assert_equal(
node._backrefs['addons']['s3nodesettings']['owner'][0],
node_settings._id
)
def test_migrate_two_users_one_account(self):
self.mock_user_info.stop()
self.mock_user_info.return_value = fake_user_info(id='1234567890', display_name='s3.user')
self.mock_user_info.start()
self.linked_user_settings[1]["_id"] = self.linked_user_settings[0]["_id"]
self.linked_user_settings[1]["_version"] = self.linked_user_settings[0]["_version"]
self.linked_user_settings[1]["access_key"] = self.linked_user_settings[0]["access_key"]
self.linked_user_settings[1]["deleted"] = self.linked_user_settings[0]["deleted"]
external_account_1, user_1, new_1 = migration.migrate_to_external_account(self.linked_user_settings[0])
external_account_2, user_2, new_2 = migration.migrate_to_external_account(self.linked_user_settings[1])
assert_equal(external_account_1._id, external_account_2._id)
assert_not_equal(user_1, user_2)
assert_true(new_1)
assert_false(new_2)
def test_migrate_to_external_account(self):
assert_equal(ExternalAccount.find().count(), 0)
user_settings_document = self.unlinked_user_settings[0]
external_account, user, new = migration.migrate_to_external_account(
user_settings_document)
assert_true(new)
assert_equal(ExternalAccount.find().count(), 1)
assert_is_not_none(external_account)
assert_equal(user_settings_document['owner'], user._id)
assert_equal(external_account.provider, 's3')
assert_equal(external_account.provider_name, 'Amazon S3')
assert_equal(external_account.oauth_key,
user_settings_document['access_key'])
assert_is_not_none(external_account.display_name)
def test_migrate_to_external_account(self):
assert_equal(ExternalAccount.find().count(), 0)
user_settings_document = self.unlinked_user_settings[0]
external_account, user, new = migration.migrate_to_external_account(user_settings_document)
assert_true(new)
assert_equal(ExternalAccount.find().count(), 1)
assert_is_not_none(external_account)
assert_equal(user_settings_document['owner'], user._id)
assert_equal(external_account.provider, 's3')
assert_equal(external_account.provider_name, 'Amazon S3')
assert_equal(
external_account.oauth_key,
user_settings_document['access_key']
)
assert_is_not_none(external_account.display_name)
