def test_get_by_user_id(self):
"""Test for getting connections by a specific user id."""
user_2 = create_user()
users = [create_user() for i in range(10)]
for user in users:
UserConnection.objects.create(created_user=user_2, with_user=user)
connections = UserConnection.objects.get_by_user_id(user_id=user_2.id)
self.assertEqual(len(connections), 10)
def test_get_by_user_id(self):
"""Test for getting connections by a specific user id."""
user_2 = create_user()
users = [create_user() for i in range(10)]
for user in users:
UserConnection.objects.create(created_user=user_2, with_user=user)
connections = UserConnection.objects.get_by_user_id(user_id=user_2.id)
self.assertEqual(len(connections), 10)
def test_get_connected_user(self):
"""Test getting the user the connection is with. Could be the connected
user or the with_user.
"""
user_2 = create_user()
c = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
self.assertEqual(c.get_connected_user(self.user), user_2)
self.assertEqual(c.get_connected_user(user_2), self.user)
user_3 = create_user()
self.assertIsNone(c.get_connected_user(user_3))
def test_get_connected_user(self):
"""Test getting the user the connection is with. Could be the connected
user or the with_user.
"""
user_2 = create_user()
c = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
self.assertEqual(c.get_connected_user(self.user), user_2)
self.assertEqual(c.get_connected_user(user_2), self.user)
user_3 = create_user()
self.assertIsNone(c.get_connected_user(user_3))
def test_get_by_shared_objects(self):
"""Get shares for a shared objects."""
obj_1 = TestSharedObjectModel.objects.create()
obj_2 = TestSharedObjectModel2.objects.create()
user_2 = create_user()
share_user_1_obj_1 = Share.objects.create_for_user(
created_user=self.user, for_user=self.user, shared_object=obj_1)
share_user_1_obj_2 = Share.objects.create_for_user(
created_user=self.user, for_user=self.user, shared_object=obj_2)
share_user_2_obj_1 = Share.objects.create_for_user(created_user=user_2,
for_user=user_2,
shared_object=obj_1)
share_user_2_obj_2 = Share.objects.create_for_user(created_user=user_2,
for_user=user_2,
shared_object=obj_2)
shares = list(Share.objects.get_by_shared_objects(objs=[obj_1, obj_2]))
self.assertEqual(len(shares), 4)
self.assertTrue(share_user_1_obj_1 in shares)
self.assertTrue(share_user_1_obj_2 in shares)
self.assertTrue(share_user_2_obj_1 in shares)
self.assertTrue(share_user_2_obj_2 in shares)
shares = Share.objects.get_by_shared_objects(objs=[obj_1, obj_2],
for_user=user_2)
self.assertEqual(len(shares), 2)
self.assertTrue(share_user_2_obj_1 in shares)
self.assertTrue(share_user_2_obj_2 in shares)
def test_get_by_user(self):
"""Test model manager get by user."""
user = create_user()
obj = TestManagerModel.objects.create(created_user=self.user,
user=user)
obj_db = TestManagerModel.objects.get_by_user(user=user)
self.assertEqual(obj, obj_db[0])
def test_get_full_name_for_user(self):
"""Test get full name for a share for existing user."""
first_name = 'John'
last_name = 'Doe'
user_2 = create_user(first_name=first_name, last_name=last_name)
share = Share.objects.create_for_user(created_user=user_2,
for_user=user_2,
shared_object=self.shared_user)
self.assertEqual(share.get_full_name(),
'{0} {1}'.format(first_name, last_name))
def test_get_for_user_id(self):
"""Get shares for a user id."""
user = create_user()
share = Share.objects.create_for_user(created_user=self.user,
for_user=user,
shared_object=self.shared_user)
shares = Share.objects.get_for_user_id(user_id=user.id)
self.assertEqual(len(shares), 1)
self.assertEqual(shares[0], share)
def test_get_email(self):
"""Get shares by email."""
user = create_user()
share = Share.objects.create_for_user(created_user=self.user,
for_user=user,
shared_object=self.shared_user)
shares = Share.objects.get_by_email(email=user.email)
self.assertEqual(len(shares), 1)
self.assertEqual(shares[0], share)
def test_get_by_shared_object(self):
"""Get shares for a shared object."""
shared_object = create_user()
share = Share.objects.create_for_user(created_user=self.user,
for_user=self.user,
shared_object=shared_object)
shares = Share.objects.get_by_shared_object(obj=shared_object)
self.assertEqual(len(shares), 1)
self.assertEqual(shares[0], share)
def test_decline_connection(self):
"""Test for declining a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
self.assertEqual(conn.status, Status.PENDING)
conn.decline()
self.assertEqual(conn.status, Status.DECLINED)
def test_accept_connection(self):
"""Test for accepting a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
self.assertEqual(conn.status, Status.PENDING)
conn.accept()
self.assertEqual(conn.status, Status.ACCEPTED)
def test_delete_connection(self):
"""Testing deleting a connection."""
user_2 = create_user()
c = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
c.delete()
conn_1 = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertIsNone(conn_1)
def test_get_by_token(self):
"""Test for getting a connection by a token."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
conn_2 = UserConnection.objects.get_by_token(token=conn.token)
self.assertEqual(conn, conn_2)
def test_accept_connection(self):
"""Test for accepting a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
self.assertEqual(conn.status, Status.PENDING)
conn.accept()
self.assertEqual(conn.status, Status.ACCEPTED)
def test_delete_connection(self):
"""Testing deleting a connection."""
user_2 = create_user()
c = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
c.delete()
conn_1 = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertIsNone(conn_1)
def test_decline_connection(self):
"""Test for declining a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
self.assertEqual(conn.status, Status.PENDING)
conn.decline()
self.assertEqual(conn.status, Status.DECLINED)
def test_get_by_token(self):
"""Test for getting a connection by a token."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
conn_2 = UserConnection.objects.get_by_token(token=conn.token)
self.assertEqual(conn, conn_2)
def test_incremement_activity_count(self):
"""Test for incrementing the total activity count for a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2)
self.assertEqual(conn.activity_count, 1)
conn.increment_activity_count()
conn = UserConnection.objects.get(id=conn.id)
self.assertEqual(conn.activity_count, 2)
def test_incremement_activity_count(self):
"""Test for incrementing the total activity count for a connection."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2)
self.assertEqual(conn.activity_count, 1)
conn.increment_activity_count()
conn = UserConnection.objects.get(id=conn.id)
self.assertEqual(conn.activity_count, 2)
def test_get_connection_by_user_ids(self):
"""Test for getting a connection between 2 users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
conn_1 = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertEqual(conn, conn_1)
conn_2 = UserConnection.objects.get_for_users(user_1=user_2,
user_2=self.user)
self.assertEqual(conn, conn_2)
def test_get_connection_by_user_ids(self):
"""Test for getting a connection between 2 users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.PENDING)
conn_1 = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertEqual(conn, conn_1)
conn_2 = UserConnection.objects.get_for_users(user_1=user_2,
user_2=self.user)
self.assertEqual(conn, conn_2)
def test_get_for_users(self):
"""Test getting connection between two users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2)
conn_db = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertEqual(conn, conn_db)
conn_db_2 = UserConnection.objects.get_for_users(user_1=user_2,
user_2=self.user)
self.assertEqual(conn, conn_db_2)
def test_get_for_users(self):
"""Test getting connection between two users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2)
conn_db = UserConnection.objects.get_for_users(user_1=self.user,
user_2=user_2)
self.assertEqual(conn, conn_db)
conn_db_2 = UserConnection.objects.get_for_users(user_1=user_2,
user_2=self.user)
self.assertEqual(conn, conn_db_2)
def test_umanage_activate_account_view(self):
"""Test the activate account url to ensure it renders correctly."""
test_user = create_user()
test_user.is_active = False
test_user.save()
self.assertFalse(test_user.is_active)
authorization = AccountActivationAuthorization.objects.create(
created_user=test_user)
self.response_test_get(url=authorization.get_absolute_url(),
expected_status_code=302)
test_user = get_user_model().objects.get(id=test_user.id)
self.assertTrue(test_user.is_active)
def test_create_connection_accepted(self):
"""Test for adding an accepted connection between 2 users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
self.assertEqual(conn.activity_count, 1)
self.assertEqual(conn.status, Status.ACCEPTED)
self.assertEqual(conn.created_user, self.user)
users = conn.users
self.assertEqual(len(users), 2)
self.assertTrue(self.user in users)
self.assertTrue(user_2 in users)
def test_create_connection_accepted(self):
"""Test for adding an accepted connection between 2 users."""
user_2 = create_user()
conn = UserConnection.objects.create(created_user=self.user,
with_user=user_2,
status=Status.ACCEPTED)
self.assertEqual(conn.activity_count, 1)
self.assertEqual(conn.status, Status.ACCEPTED)
self.assertEqual(conn.created_user, self.user)
users = conn.users
self.assertEqual(len(users), 2)
self.assertTrue(self.user in users)
self.assertTrue(user_2 in users)
def test_umanage_activate_account_view(self):
"""Test the activate account url to ensure it renders correctly."""
test_user = create_user()
test_user.is_active = False
test_user.save()
self.assertFalse(test_user.is_active)
authorization = AccountActivationAuthorization.objects.create(
created_user=test_user
)
self.response_test_get(
url=authorization.get_absolute_url(),
expected_status_code=302
)
test_user = get_user_model().objects.get(id=test_user.id)
self.assertTrue(test_user.is_active)
def test_create_many(self):
"""Test for creating many objects at once. This is different from
bulk_create. See ``create_many`` doc.
"""
user = create_user()
obj_1 = TestSharedObjectModel.objects.create()
obj_2 = TestSharedObjectModel.objects.create()
obj_3 = TestSharedObjectModel.objects.create()
objs = [obj_1, obj_2, obj_3]
# There shouldn't be any shares here.
self.assertEqual(obj_1.shares.count(), 0)
self.assertEqual(obj_2.shares.count(), 0)
self.assertEqual(obj_3.shares.count(), 0)
ShareClass = TestSharedObjectModel.get_share_class()
shares = ShareClass.objects.create_many(objs=objs,
for_user=user,
created_user=user,
status=Status.ACCEPTED)
self.assertEqual(obj_1.shares.count(), 1)
self.assertEqual(obj_2.shares.count(), 1)
self.assertEqual(obj_3.shares.count(), 1)
def test_create_many_prevent_duplicate_share(self):
"""Test the ``create_many`` method that ensure no duplicate shares are
created for a single user.
"""
user = create_user()
obj_1 = TestSharedObjectModel.objects.create()
obj_1.shares.create_for_user(for_user=user,
created_user=user,
status=Status.ACCEPTED)
self.assertEqual(obj_1.shares.count(), 1)
obj_2 = TestSharedObjectModel.objects.create()
obj_3 = TestSharedObjectModel.objects.create()
objs = [obj_1, obj_2, obj_3]
ShareClass = TestSharedObjectModel.get_share_class()
shares = ShareClass.objects.create_many(objs=objs,
for_user=user,
created_user=user,
status=Status.ACCEPTED)
self.assertEqual(obj_1.shares.count(), 1)
self.assertEqual(obj_2.shares.count(), 1)
self.assertEqual(obj_3.shares.count(), 1)
def setUp(self):
super(GenericObjectModelTests, self).setUp()
self.user = create_user()
def test_umanage_forgot_password_change_password_view(self):
"""Test the forgot password change password url renders."""
authorization = ForgotPasswordAuthorization.objects.create(
created_user=create_user()
)
self.response_test_get(url=authorization.get_absolute_url())
def setUpClass(cls):
super(UmanageUnauthenticatedUrlTests, cls).setUpClass()
cls.user = create_user()
def setUpClass(cls):
super(ConnectionManagerTestCase, cls).setUpClass()
cls.user = create_user()
def setUpClass(cls):
super(SingleUserTestCase, cls).setUpClass()
cls.user = create_user()
def test_forgot_password_form_email_valid(self):
"""Test the forgot password user's email"""
user = create_user()
form = ForgotPasswordForm(data={'username_or_email': user.email})
self.assertTrue(form.is_valid())
self.assertEqual(user.email, form.clean_username_or_email())
def test_forgot_password_form_send_email(self):
"""Test the forgot password email is sent."""
user = create_user()
form = ForgotPasswordForm(data={'username_or_email': user.email})
form.is_valid()
form.send_email()
def test_forgot_username_form_email_valid(self):
"""Test the forgot username valid email"""
user = create_user()
form = ForgotUsernameForm(data={'email': user.email})
self.assertTrue(form.is_valid())
self.assertEqual(user.email, form.clean_email())
def test_forgot_username_form_email_valid(self):
"""Test the forgot username valid email"""
user = create_user()
form = ForgotUsernameForm(data={'email': user.email})
self.assertTrue(form.is_valid())
self.assertEqual(user.email, form.clean_email())
def test_forgot_username_form_send_email(self):
"""Test the forgot username valid email"""
user = create_user()
form = ForgotUsernameForm(data={'email': user.email})
form.is_valid()
form.send_email()
def setUp(self):
"""Run once per test."""
super(ShareTests, self).setUp()
self.shared_user = create_user()
def setUpClass(cls):
super(UmanageUnauthenticatedUrlTests, cls).setUpClass()
cls.user = create_user()
def test_umanage_forgot_password_change_password_view(self):
"""Test the forgot password change password url renders."""
authorization = ForgotPasswordAuthorization.objects.create(
created_user=create_user())
self.response_test_get(url=authorization.get_absolute_url())
def test_forgot_password_form_email_valid(self):
"""Test the forgot password user's email"""
user = create_user()
form = ForgotPasswordForm(data={'username_or_email': user.email})
self.assertTrue(form.is_valid())
self.assertEqual(user.email, form.clean_username_or_email())
def test_forgot_password_form_send_email(self):
"""Test the forgot password email is sent."""
user = create_user()
form = ForgotPasswordForm(data={'username_or_email': user.email})
form.is_valid()
form.send_email()
def setUp(self):
super(GenericObjectModelTests, self).setUp()
self.user = create_user()
def test_forgot_username_form_send_email(self):
"""Test the forgot username valid email"""
user = create_user()
form = ForgotUsernameForm(data={'email': user.email})
form.is_valid()
form.send_email()
def setUpClass(cls):
super(ConnectionManagerTestCase, cls).setUpClass()
cls.user = create_user()
