Esempio n. 1
0
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

        # Create a user who has permissions for companies and company reps
        self.user = self.user_model.objects.create_user(
            username='******',
            email='*****@*****.**',
            password='******',
            first_name='John',
            last_name='Superuser')

        company_content_type = ContentType.objects.get_for_model(Company)
        rep_content_type = ContentType.objects.get_for_model(CompanyRep)
        company_add_permission = Permission.objects.get(
            content_type=company_content_type, codename='add_company')
        company_change_permission = Permission.objects.get(
            content_type=company_content_type, codename='change_company')
        companies_view_permission = Permission.objects.get(
            content_type=company_content_type, codename='view_companies')
        rep_add_permission = Permission.objects.get(
            content_type=rep_content_type, codename='add_companyrep')
        rep_change_permission = Permission.objects.get(
            content_type=rep_content_type, codename='change_companyrep')
        rep_delete_permission = Permission.objects.get(
            content_type=rep_content_type, codename='delete_companyrep')
        self.user.user_permissions.add(company_add_permission,
                                       company_change_permission,
                                       companies_view_permission,
                                       rep_add_permission,
                                       rep_change_permission,
                                       rep_delete_permission)
Esempio n. 2
0
    def test_company_is_expired(self):
        """Make sure is_expired properly indicates expiration."""
        today = datetime.date.today()

        # Make the company's expiration in the past:
        expiration = today - datetime.timedelta(days=1)
        company = Company(name='Test Company', expiration_date=expiration)
        company.save()
        self.assertTrue(company.is_expired())
Esempio n. 3
0
    def test_company_is_expired(self):
        """Make sure is_expired properly indicates expiration."""
        today = datetime.date.today()

        # Make the company's expiration in the past:
        expiration = today - datetime.timedelta(days=1)
        company = Company(name='Test Company', expiration_date=expiration)
        company.save()
        self.assertTrue(company.is_expired())
Esempio n. 4
0
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

        # Create a user who has permissions for companies and company reps
        self.user = self.user_model.objects.create_user(
            username='******',
            email='*****@*****.**',
            password='******',
            first_name='John',
            last_name='Superuser')

        company_content_type = ContentType.objects.get_for_model(Company)
        rep_content_type = ContentType.objects.get_for_model(CompanyRep)
        company_add_permission = Permission.objects.get(
            content_type=company_content_type, codename='add_company')
        company_change_permission = Permission.objects.get(
            content_type=company_content_type, codename='change_company')
        rep_add_permission = Permission.objects.get(
            content_type=rep_content_type, codename='add_companyrep')
        rep_change_permission = Permission.objects.get(
            content_type=rep_content_type, codename='change_companyrep')
        self.user.user_permissions.add(
            company_add_permission, company_change_permission,
            rep_add_permission, rep_change_permission)
Esempio n. 5
0
class CompanyFormsTest(TestCase):
    fixtures = ['groups.yaml']

    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

    def test_company_rep_creation_form(self):
        """Ensure that the company rep creation form works to create a new user
        and the corresponding CompanyRep object.
        """
        # The email addresses don't match
        rep_attrs = {
            'username': '******',
            'email': '*****@*****.**',
            'confirm_email': '*****@*****.**',
            'first_name': 'Jane',
            'last_name': 'Doe',
            'company': self.company.pk
        }
        form = CompanyRepCreationForm(rep_attrs)
        self.assertFalse(form.is_valid())

        # Fix the email address mismatch
        rep_attrs['confirm_email'] = '*****@*****.**'
        form = CompanyRepCreationForm(rep_attrs)
        self.assertTrue(form.is_valid())
        form.save()

        # Check that a user was created with the appropriate attributes and
        # an unusable password:
        rep_user = self.user_model.objects.get(username=rep_attrs['username'])
        self.assertEquals(rep_user.get_username(), rep_attrs['username'])
        self.assertEquals(rep_user.email, rep_attrs['email'])
        self.assertEquals(rep_user.first_name, rep_attrs['first_name'])
        self.assertEquals(rep_user.last_name, rep_attrs['last_name'])
        self.assertEquals(rep_user.last_name, rep_attrs['last_name'])
        self.assertFalse(rep_user.has_usable_password())

        # Check that the CompanyRep was created correctly:
        rep = rep_user.companyrep
        self.assertEquals(rep.company, self.company)
Esempio n. 6
0
    def test_company_user_auth_succeeds_for_valid_account(self):
        """Verify that a company rep user can log in if their company's
        subscription is not expired.
        """
        # Create a company rep for the user, with a company that is active
        # (is_expired returns False)
        company = Company(name='Test Company',
                          expiration_date=datetime.date.today())
        company.save()
        with patch.object(Company, 'is_expired',
                          return_value=False) as mock_is_expired:
            rep = CompanyRep(user=self.user, company=company)
            rep.save()

            # Ensure that the user can be authenticated
            form = AuthenticationForm(None, self.form_data)
            self.assertTrue(form.is_valid())
        self.assertEquals(mock_is_expired.call_count, 1)
Esempio n. 7
0
    def test_company_user_auth_succeeds_for_valid_account(self):
        """Verify that a company rep user can log in if their company's
        subscription is not expired.
        """
        # Create a company rep for the user, with a company that is active
        # (is_expired returns False)
        company = Company(name='Test Company',
                          expiration_date=datetime.date.today())
        company.save()
        with patch.object(Company, 'is_expired',
                          return_value=False) as mock_is_expired:
            rep = CompanyRep(user=self.user, company=company)
            rep.save()

            # Ensure that the user can be authenticated
            form = AuthenticationForm(None, self.form_data)
            self.assertTrue(form.is_valid())
        self.assertEquals(mock_is_expired.call_count, 1)
Esempio n. 8
0
    def test_company_is_not_expired(self):
        """Make sure is_expired properly indicates non-expiration."""
        today = datetime.date.today()

        # Make the company's expiration date weeks in the future:
        expiration = today + datetime.timedelta(weeks=5)
        company = Company(name='Test Company', expiration_date=expiration)
        company.save()
        self.assertFalse(company.is_expired())

        # Move the expiration date to today, which should be the final
        # non-expired day:
        expiration = today
        company.expiration_date = expiration
        company.save()
        self.assertFalse(company.is_expired())
Esempio n. 9
0
    def test_companyrep_delete_view(self):
        """Ensure that representatives can be deleted and that deleted reps
        have their accounts disabled.
        """
        company = Company(expiration_date='3000-01-01')
        company.save()
        companyrep_user = self.user_model.objects.create_user(
            username='******', password='******')
        companyrep_user_pk = companyrep_user.pk
        companyrep = CompanyRep(company=company, user=companyrep_user)
        companyrep.save()

        self.assertTrue(
            self.client.login(username=self.user.username,
                              password='******'))
        self.assertTrue(companyrep_user.is_active)
        self.assertTrue(
            self.client.login(username='******', password='******'))

        # Use the view
        self.assertTrue(
            self.client.login(username=self.user.username,
                              password='******'))
        response = self.client.get(reverse('companies:rep-delete',
                                           args=(companyrep.pk, )),
                                   follow=True)
        self.assertContains(response, 'Are you sure')
        self.client.post(reverse('companies:rep-delete',
                                 args=(companyrep.pk, )),
                         follow=True)

        # Check that everything has been deleted
        self.assertFalse(CompanyRep.objects.exists())

        # Check that the rep can't log in
        companyrep_user = User.objects.get(pk=companyrep_user_pk)
        self.assertFalse(companyrep_user.is_active)
        self.assertFalse(
            self.client.login(username='******', password='******'))
Esempio n. 10
0
    def test_company_user_auth_fails_for_expired_account(self):
        """Verify that a company rep user cannot log in if their company's
        subscription is expired.
        """
        # Create a company rep for the user, with a company that has an expired
        # subscription (is_expired returns True)
        company = Company(name='Test Company',
                          expiration_date=datetime.date.today())
        company.save()
        with patch.object(Company, 'is_expired',
                          return_value=True) as mock_is_expired:
            rep = CompanyRep(user=self.user, company=company)
            rep.save()

            # Ensure that the user cannot be authenticated
            form = AuthenticationForm(None, self.form_data)
            self.assertFalse(form.is_valid())
            expected_error_msg = (
                '{}\'s subscription to this website has expired'.format(
                    company.name))
            self.assertIn(expected_error_msg, form.non_field_errors()[0])
        self.assertEquals(mock_is_expired.call_count, 1)
Esempio n. 11
0
    def test_company_user_auth_fails_for_expired_account(self):
        """Verify that a company rep user cannot log in if their company's
        subscription is expired.
        """
        # Create a company rep for the user, with a company that has an expired
        # subscription (is_expired returns True)
        company = Company(name='Test Company',
                          expiration_date=datetime.date.today())
        company.save()
        with patch.object(Company, 'is_expired',
                          return_value=True) as mock_is_expired:
            rep = CompanyRep(user=self.user, company=company)
            rep.save()

            # Ensure that the user cannot be authenticated
            form = AuthenticationForm(None, self.form_data)
            self.assertFalse(form.is_valid())
            expected_error_msg = (
                '{}\'s subscription to this website has expired'.format(
                    company.name))
            self.assertIn(expected_error_msg, form.non_field_errors()[0])
        self.assertEquals(mock_is_expired.call_count, 1)
Esempio n. 12
0
    def test_company_is_not_expired(self):
        """Make sure is_expired properly indicates non-expiration."""
        today = datetime.date.today()

        # Make the company's expiration date weeks in the future:
        expiration = today + datetime.timedelta(weeks=5)
        company = Company(name='Test Company', expiration_date=expiration)
        company.save()
        self.assertFalse(company.is_expired())

        # Move the expiration date to today, which should be the final
        # non-expired day:
        expiration = today
        company.expiration_date = expiration
        company.save()
        self.assertFalse(company.is_expired())
Esempio n. 13
0
class CompanyFormsTest(TestCase):
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

    def test_company_rep_creation_form(self):
        """Ensure that the company rep creation form works to create a new user
        and the corresponding CompanyRep object.
        """
        rep_attrs = {
            'username': '******',
            'email': '*****@*****.**',
            'first_name': 'Jane',
            'last_name': 'Doe',
            'company': self.company.pk
        }
        form = CompanyRepCreationForm(rep_attrs)
        self.assertTrue(form.is_valid())
        form.save()

        # Check that a user was created with the appropriate attributes and
        # an unusable password:
        rep_user = self.user_model.objects.get(username=rep_attrs['username'])
        self.assertEquals(rep_user.get_username(), rep_attrs['username'])
        self.assertEquals(rep_user.email, rep_attrs['email'])
        self.assertEquals(rep_user.first_name, rep_attrs['first_name'])
        self.assertEquals(rep_user.last_name, rep_attrs['last_name'])
        self.assertEquals(rep_user.last_name, rep_attrs['last_name'])
        self.assertFalse(rep_user.has_usable_password())

        # Check that the CompanyRep was created correctly:
        rep = rep_user.companyrep
        self.assertEquals(rep.company, self.company)
Esempio n. 14
0
class CompanyRepViewTest(TestCase):
    """Test the CompanyRep views."""

    fixtures = ['groups.yaml']

    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

        # Create a user who has permissions for companies and company reps
        self.user = self.user_model.objects.create_user(
            username='******',
            email='*****@*****.**',
            password='******',
            first_name='John',
            last_name='Superuser')

        company_content_type = ContentType.objects.get_for_model(Company)
        rep_content_type = ContentType.objects.get_for_model(CompanyRep)
        company_add_permission = Permission.objects.get(
            content_type=company_content_type, codename='add_company')
        company_change_permission = Permission.objects.get(
            content_type=company_content_type, codename='change_company')
        companies_view_permission = Permission.objects.get(
            content_type=company_content_type, codename='view_companies')
        rep_add_permission = Permission.objects.get(
            content_type=rep_content_type, codename='add_companyrep')
        rep_change_permission = Permission.objects.get(
            content_type=rep_content_type, codename='change_companyrep')
        rep_delete_permission = Permission.objects.get(
            content_type=rep_content_type, codename='delete_companyrep')
        self.user.user_permissions.add(company_add_permission,
                                       company_change_permission,
                                       companies_view_permission,
                                       rep_add_permission,
                                       rep_change_permission,
                                       rep_delete_permission)

    def test_company_rep_create_view(self):
        """Ensure that the user can create a company rep successfully, and that
        the company rep receives a password reset email.
        """
        self.assertTrue(
            self.client.login(username=self.user.username,
                              password='******'))
        create_rep_url = reverse('companies:rep-create')
        rep_data = {
            'username': '******',
            'email': '*****@*****.**',
            'confirm_email': '*****@*****.**',
            'first_name': 'Jane',
            'last_name': 'Doe',
            'company': self.company.pk
        }
        # Make sure there are currently no company reps and that no emails have
        # been sent
        self.assertFalse(CompanyRep.objects.exists())
        self.assertEqual(len(mail.outbox), 0)

        # Create the rep using the view
        response = self.client.post(create_rep_url, rep_data, follow=True)
        self.assertEqual(response.status_code, 200)
        self.assertContains(response,
                            'Successfully created a new company rep account')
        self.assertContains(response, rep_data['username'])
        self.assertContains(response, self.company.name)

        # Check that the rep account was successfully created
        rep_user = self.user_model.objects.get(username=rep_data['username'])
        self.assertEquals(rep_user.companyrep, CompanyRep.objects.get())
        self.assertEquals(rep_user.companyrep.company, self.company)

        # Check the sent email to ensure the company rep received an email for
        # setting their password
        self.assertEqual(len(mail.outbox), 1)
        email = mail.outbox[0]
        self.assertEqual(email.to, [rep_user.email])

        # Make sure the email contains a password reset link:
        self.assertIn('accounts/password/reset', email.body)

    def test_companyrep_delete_view(self):
        """Ensure that representatives can be deleted and that deleted reps
        have their accounts disabled.
        """
        company = Company(expiration_date='3000-01-01')
        company.save()
        companyrep_user = self.user_model.objects.create_user(
            username='******', password='******')
        companyrep_user_pk = companyrep_user.pk
        companyrep = CompanyRep(company=company, user=companyrep_user)
        companyrep.save()

        self.assertTrue(
            self.client.login(username=self.user.username,
                              password='******'))
        self.assertTrue(companyrep_user.is_active)
        self.assertTrue(
            self.client.login(username='******', password='******'))

        # Use the view
        self.assertTrue(
            self.client.login(username=self.user.username,
                              password='******'))
        response = self.client.get(reverse('companies:rep-delete',
                                           args=(companyrep.pk, )),
                                   follow=True)
        self.assertContains(response, 'Are you sure')
        self.client.post(reverse('companies:rep-delete',
                                 args=(companyrep.pk, )),
                         follow=True)

        # Check that everything has been deleted
        self.assertFalse(CompanyRep.objects.exists())

        # Check that the rep can't log in
        companyrep_user = User.objects.get(pk=companyrep_user_pk)
        self.assertFalse(companyrep_user.is_active)
        self.assertFalse(
            self.client.login(username='******', password='******'))
Esempio n. 15
0
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()
Esempio n. 16
0
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()
Esempio n. 17
0
class CompanyViewsTest(TestCase):
    """Test the companies views."""
    def setUp(self):
        expiration = datetime.date.today() + datetime.timedelta(weeks=5)
        self.company = Company(name='Test Company', expiration_date=expiration)
        self.company.save()

        self.user_model = get_user_model()

        # Create a user who has permissions for companies and company reps
        self.user = self.user_model.objects.create_user(
            username='******',
            email='*****@*****.**',
            password='******',
            first_name='John',
            last_name='Superuser')

        company_content_type = ContentType.objects.get_for_model(Company)
        rep_content_type = ContentType.objects.get_for_model(CompanyRep)
        company_add_permission = Permission.objects.get(
            content_type=company_content_type, codename='add_company')
        company_change_permission = Permission.objects.get(
            content_type=company_content_type, codename='change_company')
        rep_add_permission = Permission.objects.get(
            content_type=rep_content_type, codename='add_companyrep')
        rep_change_permission = Permission.objects.get(
            content_type=rep_content_type, codename='change_companyrep')
        self.user.user_permissions.add(
            company_add_permission, company_change_permission,
            rep_add_permission, rep_change_permission)

    def test_company_rep_create_view(self):
        """Ensure that the user can create a company rep successfully, and that
        the company rep receives a password reset email.
        """
        self.assertTrue(self.client.login(
            username=self.user.username, password='******'))
        create_rep_url = reverse('companies:create-rep')
        rep_data = {
            'username': '******',
            'email': '*****@*****.**',
            'first_name': 'Jane',
            'last_name': 'Doe',
            'company': self.company.pk
        }
        # Make sure there are currently no company reps and that no emails have
        # been sent
        self.assertFalse(CompanyRep.objects.exists())
        self.assertEqual(len(mail.outbox), 0)

        # Create the rep using the view
        response = self.client.post(create_rep_url, rep_data, follow=True)
        self.assertEqual(response.status_code, 200)
        self.assertContains(
            response, 'Successfully created a new company rep account')
        self.assertContains(response, rep_data['username'])
        self.assertContains(response, self.company.name)

        # Check that the rep account was successfully created
        rep_user = self.user_model.objects.get(username=rep_data['username'])
        self.assertEquals(rep_user.companyrep, CompanyRep.objects.get())
        self.assertEquals(rep_user.companyrep.company, self.company)

        # Check the sent email to ensure the company rep received an email for
        # setting their password
        self.assertEqual(len(mail.outbox), 1)
        email = mail.outbox[0]
        self.assertEqual(email.to, [rep_user.email])

        # Make sure the email contains a password reset link:
        self.assertIn('accounts/password/reset', email.body)