def setUp(self):
self.ae = MockAppEngine()
self.user = self.ae.get_current_user()
self.db = MockAccess(user=self.user, add_owner=self.is_user_owner())
# set up mock Request and Response objects
self.request = mock.MagicMock()
self.response = mock.MagicMock()
# Create a special SpyResource class that has all the same
# functionality as the appropriate Resource for this test, but
# has a fake 'dump' method which stores the last js_object dumped
resource_cls = self.get_resource_cls()
class SpyResource(resource_cls):
def __init__(self, *args, **kwargs):
super(SpyResource, self).__init__(*args, **kwargs)
self.last_response = None
def get_last_response(self):
return self.last_response
def dump(self, js_response):
self.last_response = js_response
# instantiate the SpyResource with fake Request/Response as well as
# AppEngine and DbAccess
self.resource = SpyResource(
request=self.request,
response=self.response,
ae=self.ae,
db=self.db,
)
def setUp(self):
self.ae = MockAppEngine()
self.user = self.ae.get_current_user()
self.db = MockAccess(user=self.user, add_owner=self.is_user_owner())
# set up mock Request and Response objects
self.request = mock.MagicMock()
self.response = mock.MagicMock()
# Create a special SpyResource class that has all the same
# functionality as the appropriate Resource for this test, but
# has a fake 'dump' method which stores the last js_object dumped
resource_cls = self.get_resource_cls()
class SpyResource(resource_cls):
def __init__(self, *args, **kwargs):
super(SpyResource, self).__init__(*args, **kwargs)
self.last_response = None
def get_last_response(self):
return self.last_response
def dump(self, js_response):
self.last_response = js_response
# instantiate the SpyResource with fake Request/Response as well as
# AppEngine and DbAccess
self.resource = SpyResource(
request=self.request,
response=self.response,
ae=self.ae,
db=self.db,
)
def setUp(self):
self.db = MockAccess(User(), add_owner=True)
self.ae = MockAppEngine()
self.rpc = GroupRpcGroup(self.db, self.ae)
def set_user(self, user):
self.user = user
self.db = MockAccess(self.user)
self.ae = MockAppEngine()
self.rpc = UserRpcGroup(self.db, self.ae)
class UserRpcTest(unittest.TestCase):
def setUp(self):
self.set_user(User())
def set_user(self, user):
self.user = user
self.db = MockAccess(self.user)
self.ae = MockAppEngine()
self.rpc = UserRpcGroup(self.db, self.ae)
def Xtest_get_user_none(self):
"""
Verify that calling get_current_user without being logged in
provides you with a login url
"""
# Start with no user logged in
self.set_user(None)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify nothing but login url is returned
self.assertEqual(None, user.logout_url)
self.assertEqual(None, user.email)
self.assertEqual(None, user.user_id)
self.assertEqual(None, user.nickname)
login_url = self.ae.create_login_url(url)
self.assertEqual(login_url, user.login_url)
def Xtest_get_user(self):
"""
Verify that calling get_current_user while logged in provides you
with a logout url and your login info
"""
self.set_user(User(is_admin=True))
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
base = get_base_url(url)
goodbye = '{}/#Goodbye:'.format(base)
self.assertEqual(self.ae.create_logout_url(goodbye), user.logout_url)
self.assertEqual(self.user.nickname(), user.nickname)
self.assertEqual(self.user.email(), user.email)
self.assertEqual(None, user.login_url)
# Verify owner was created correctly
owner = self.db.user_owners.get(self.user)
self.assertEqual(num_owners + 1, len(self.db.owners))
self.assertEqual(self.user, owner.user)
self.assertEqual(owner.key().id(), user.owner_id)
self.assertEqual(self.user.nickname(), owner.nickname)
self.assertEqual(self.user.email(), owner.email)
self.assertEqual(extract_name(self.user.email()), owner.name)
def Xtest_get_user_owner_exists(self):
"""
Verify that a duplicate owner is not created when get_current_user
is called
"""
# Make sure owner exists before call
owner = self.db.add_owner(self.user)
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# verify still the same owner after the call
self.assertEqual(num_owners, len(self.db.owners))
self.assertEqual(owner.key().id(), user.owner_id)
self.assertEqual(owner, self.db.user_owners[self.user])
def Xtest_get_user_add_req(self):
"""
Verify that when non-admin user tries to log in, an Access Request is
created for them.
"""
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
base = get_base_url(url)
goodbye = '{}/#Goodbye:'.format(base)
self.assertEqual(self.ae.create_logout_url(goodbye), user.logout_url)
self.assertEqual(self.user.nickname(), user.nickname)
self.assertEqual(self.user.email(), user.email)
self.assertEqual(None, user.login_url)
self.assertEqual(-1, user.owner_id)
self.assertEqual(num_owners, len(self.db.owners))
# Verify access request was created correctly
self.assertEqual(1, len(self.db.request_ids))
req = self.db.requests.values()[0]
self.assertEqual(self.user, req.user)
def Xtest_get_user_req_exists(self):
"""
Verify that the second time get_current_user is called, a second
AccessRequest is not created.
"""
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
# Create the request that already exists
self.db.add_req(self.user)
self.assertEqual(1, len(self.db.request_ids))
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
self.assertEqual(None, user.login_url)
self.assertEqual(-1, user.owner_id)
self.assertEqual(num_owners, len(self.db.owners))
# Verify no additional request was created correctly
self.assertEqual(1, len(self.db.request_ids))
def Xtest_get_owner(self):
"""
Verify ability to lookup owner by id
"""
# Make sure the owner exists
owner = self.db.add_owner(self.user)
oid = owner.key().id()
lo = self.rpc.get_owner(oid)
self.assertEqual(oid, lo.id)
self.assertEqual(owner.email, lo.email)
self.assertEqual(owner.nickname, lo.nickname)
self.assertEqual(owner.name, lo.name)
def test_update_owner(self):
"""
Make sure user can change name and nickname of list owner
"""
# Make sure the owner exists
self.set_user(User('John Doe', 'Johnny'))
self.db.add_owner(self.user)
owner = self.db.add_owner(self.user)
self.assertFalse(owner.saved())
self.rpc.update_owner(owner.key().id(), 'Joe Smith', 'Joe')
self.assertTrue(owner.saved())
self.assertEqual('Joe Smith', owner.name)
self.assertEqual('Joe', owner.nickname)
def set_user(self, user):
self.user = user
self.db = MockAccess(self.user)
self.ae = MockAppEngine()
self.rpc = UserRpcGroup(self.db, self.ae)
class UserRpcTest(unittest.TestCase):
def setUp(self):
self.set_user(User())
def set_user(self, user):
self.user = user
self.db = MockAccess(self.user)
self.ae = MockAppEngine()
self.rpc = UserRpcGroup(self.db, self.ae)
def Xtest_get_user_none(self):
"""
Verify that calling get_current_user without being logged in
provides you with a login url
"""
# Start with no user logged in
self.set_user(None)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify nothing but login url is returned
self.assertEqual(None, user.logout_url)
self.assertEqual(None, user.email)
self.assertEqual(None, user.user_id)
self.assertEqual(None, user.nickname)
login_url = self.ae.create_login_url(url)
self.assertEqual(login_url, user.login_url)
def Xtest_get_user(self):
"""
Verify that calling get_current_user while logged in provides you
with a logout url and your login info
"""
self.set_user(User(is_admin=True))
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
base = get_base_url(url)
goodbye = '{}/#Goodbye:'.format(base)
self.assertEqual(self.ae.create_logout_url(goodbye), user.logout_url)
self.assertEqual(self.user.nickname(), user.nickname)
self.assertEqual(self.user.email(), user.email)
self.assertEqual(None, user.login_url)
# Verify owner was created correctly
owner = self.db.user_owners.get(self.user)
self.assertEqual(num_owners + 1, len(self.db.owners))
self.assertEqual(self.user, owner.user)
self.assertEqual(owner.key().id(), user.owner_id)
self.assertEqual(self.user.nickname(), owner.nickname)
self.assertEqual(self.user.email(), owner.email)
self.assertEqual(extract_name(self.user.email()), owner.name)
def Xtest_get_user_owner_exists(self):
"""
Verify that a duplicate owner is not created when get_current_user
is called
"""
# Make sure owner exists before call
owner = self.db.add_owner(self.user)
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# verify still the same owner after the call
self.assertEqual(num_owners, len(self.db.owners))
self.assertEqual(owner.key().id(), user.owner_id)
self.assertEqual(owner, self.db.user_owners[self.user])
def Xtest_get_user_add_req(self):
"""
Verify that when non-admin user tries to log in, an Access Request is
created for them.
"""
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
base = get_base_url(url)
goodbye = '{}/#Goodbye:'.format(base)
self.assertEqual(self.ae.create_logout_url(goodbye), user.logout_url)
self.assertEqual(self.user.nickname(), user.nickname)
self.assertEqual(self.user.email(), user.email)
self.assertEqual(None, user.login_url)
self.assertEqual(-1, user.owner_id)
self.assertEqual(num_owners, len(self.db.owners))
# Verify access request was created correctly
self.assertEqual(1, len(self.db.request_ids))
req = self.db.requests.values()[0]
self.assertEqual(self.user, req.user)
def Xtest_get_user_req_exists(self):
"""
Verify that the second time get_current_user is called, a second
AccessRequest is not created.
"""
# Verify no owner exists before call
self.assertEqual(None, self.db.user_owners.get(self.user, None))
num_owners = len(self.db.owners)
# Create the request that already exists
self.db.add_req(self.user)
self.assertEqual(1, len(self.db.request_ids))
url = 'http://www.domain.com/some/extra/path/'
user = self.rpc.get_current_user(url)
# Verify correct user details are returned
self.assertEqual(None, user.login_url)
self.assertEqual(-1, user.owner_id)
self.assertEqual(num_owners, len(self.db.owners))
# Verify no additional request was created correctly
self.assertEqual(1, len(self.db.request_ids))
def Xtest_get_owner(self):
"""
Verify ability to lookup owner by id
"""
# Make sure the owner exists
owner = self.db.add_owner(self.user)
oid = owner.key().id()
lo = self.rpc.get_owner(oid)
self.assertEqual(oid, lo.id)
self.assertEqual(owner.email, lo.email)
self.assertEqual(owner.nickname, lo.nickname)
self.assertEqual(owner.name, lo.name)
def test_update_owner(self):
"""
Make sure user can change name and nickname of list owner
"""
# Make sure the owner exists
self.set_user(User('John Doe', 'Johnny'))
self.db.add_owner(self.user)
owner = self.db.add_owner(self.user)
self.assertFalse(owner.saved())
self.rpc.update_owner(owner.key().id(), 'Joe Smith', 'Joe')
self.assertTrue(owner.saved())
self.assertEqual('Joe Smith', owner.name)
self.assertEqual('Joe', owner.nickname)
