def test_migrate_two_users_one_account(self):
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_token"] = self.linked_user_settings[0]["access_token"]
self.linked_user_settings[1]["deleted"] = self.linked_user_settings[0]["deleted"]
self.linked_user_settings[1]["dropbox_id"] = self.linked_user_settings[0]["dropbox_id"]
self.linked_user_settings[1]["dropbox_info"] = self.linked_user_settings[0]["dropbox_info"]
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_two_users_one_account(self):
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_token"] = self.linked_user_settings[0]["access_token"]
self.linked_user_settings[1]["deleted"] = self.linked_user_settings[0][
"deleted"]
self.linked_user_settings[1]["dropbox_id"] = self.linked_user_settings[
0]["dropbox_id"]
self.linked_user_settings[1][
"dropbox_info"] = self.linked_user_settings[0]["dropbox_info"]
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, 'dropbox')
assert_equal(external_account.provider_name, 'Dropbox')
assert_equal(external_account.oauth_key,
user_settings_document['access_token'])
assert_equal(external_account.display_name,
user_settings_document['dropbox_info']['display_name'])
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['dropboxusersettings'].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_equal(
len(node._backrefs['addons']['dropboxnodesettings']['owner']), 1)
assert_equal(
node._backrefs['addons']['dropboxnodesettings']['owner'][0],
node_settings._id)
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, 'dropbox')
assert_equal(external_account.provider_name, 'Dropbox')
assert_equal(
external_account.oauth_key,
user_settings_document['access_token']
)
assert_equal(
external_account.display_name,
user_settings_document['dropbox_info']['display_name']
)
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['dropboxusersettings'].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_equal(
len(node._backrefs['addons']['dropboxnodesettings']['owner']),
1
)
assert_equal(
node._backrefs['addons']['dropboxnodesettings']['owner'][0],
node_settings._id
)
