def test_add_did_to_followed(self):
""" DATA IDENTIFIERS (CORE): Mark a did as followed """
tmp_scope = InternalScope('mock')
dsn = 'dsn_%s' % generate_uuid()
root = InternalAccount('root')
add_did(scope=tmp_scope, name=dsn, type=DIDType.DATASET, account=root)
add_did_to_followed(scope=tmp_scope, name=dsn, account=root)
users = get_users_following_did(scope=tmp_scope, name=dsn)
rows = 0
for user in users:
rows += 1
assert_equal(rows, 1)
def test_get_users_following_did(self):
""" DATA IDENTIFIERS (CORE): Get the list of users following a did """
tmp_scope = InternalScope('mock', **self.vo)
dsn = 'dsn_%s' % generate_uuid()
root = InternalAccount('root', **self.vo)
add_did(scope=tmp_scope, name=dsn, type=DIDType.DATASET, account=root)
add_did_to_followed(scope=tmp_scope, name=dsn, account=root)
users = get_users_following_did(scope=tmp_scope, name=dsn)
rows = 0
for user in users:
rows += 1
assert rows == 1
def add_did_to_followed(scope, name, account, session=None, vo='def'):
"""
Mark a did as followed by the given account
:param scope: The scope name.
:param name: The data identifier name.
:param account: The account owner.
:param session: The database session in use.
"""
scope = InternalScope(scope, vo=vo)
account = InternalAccount(account, vo=vo)
return did.add_did_to_followed(scope=scope, name=name, account=account, session=session)
