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()