293 lines
11 KiB
Python
293 lines
11 KiB
Python
from model_mommy import mommy
|
|
from unittest import mock
|
|
|
|
from django.test import TestCase
|
|
from django.conf import settings
|
|
from django.core.urlresolvers import reverse
|
|
from django.core.urlresolvers import resolve
|
|
from cms.test_utils.testcases import CMSTestCase
|
|
from django.contrib.auth.tokens import default_token_generator
|
|
from django.utils.http import urlsafe_base64_encode
|
|
from django.utils.encoding import force_bytes
|
|
from cms.api import create_page
|
|
|
|
from membership.models import CustomUser, StripeCustomer
|
|
from utils.tests import BaseTestCase
|
|
|
|
|
|
from .views import LoginView, SignupView, PasswordResetView, PasswordResetConfirmView,\
|
|
MembershipPricingView, MembershipPaymentView
|
|
from .models import MembershipType, MembershipOrder
|
|
|
|
|
|
class ContactViewTest(TestCase):
|
|
def setUp(self):
|
|
self.url = reverse('digitalglarus:contact')
|
|
self.data = {
|
|
'name': 'test',
|
|
'email': 'test@gmail.com',
|
|
'phone_number': '32123123123123',
|
|
'message': 'This is a message',
|
|
}
|
|
|
|
def url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_any_user_should_contact_us(self):
|
|
response = self.client.post(self.url, self.data, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
|
|
class ViewsTest(CMSTestCase):
|
|
def setUp(self):
|
|
self.page1 = create_page('home', 'home_digitalglarus.html', published=True, language='en-us')
|
|
self.page2 = create_page('about', 'about.html', published=True, language='en-us', slug='about')
|
|
|
|
def test_digitalglarus_templates(self):
|
|
res1 = self.client.get('/en-us/')
|
|
self.assertContains(res1, 'Digital Glarus', status_code=200)
|
|
res2 = self.client.get('/en-us/about/')
|
|
self.assertEqual(res2.status_code, 200)
|
|
|
|
|
|
class MembershipPricingViewTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(MembershipPricingViewTest, self).setUp()
|
|
|
|
self.membership_type = mommy.make(MembershipType)
|
|
self.url = reverse('digitalglarus:membership_pricing')
|
|
self.view = MembershipPricingView
|
|
self.expected_template = 'digitalglarus/membership_pricing.html'
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
# Anonymous user should get data
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.context['membership_type'], self.membership_type)
|
|
self.assertTemplateUsed(response, self.expected_template)
|
|
|
|
|
|
class MembershipPaymentViewTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(MembershipPaymentViewTest, self).setUp()
|
|
|
|
self.membership_type = mommy.make(MembershipType)
|
|
self.url = reverse('digitalglarus:membership_payment')
|
|
self.view = MembershipPaymentView
|
|
self.expected_template = 'digitalglarus/membership_payment.html'
|
|
|
|
# post data
|
|
self.billing_address = {
|
|
'street_address': 'street name',
|
|
'city': 'MyCity',
|
|
'postal_code': '32123123123123',
|
|
'country': 'VE',
|
|
'token': 'a23kfmslwxhkwis',
|
|
'membership_type': self.membership_type.id
|
|
}
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
|
|
# Anonymous user should get redirect to login
|
|
response = self.client.get(self.url)
|
|
expected_url = "%s?next=%s" % (reverse('digitalglarus:signup'),
|
|
reverse('digitalglarus:membership_payment'))
|
|
self.assertRedirects(response, expected_url=expected_url,
|
|
status_code=302, target_status_code=200)
|
|
|
|
# Logged user should get the page
|
|
response = self.customer_client.get(self.url, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.context['stripe_key'],
|
|
settings.STRIPE_API_PUBLIC_KEY)
|
|
self.assertEqual(response.context['membership_type'],
|
|
self.membership_type)
|
|
|
|
@mock.patch('utils.stripe_utils.StripeUtils.create_customer')
|
|
def test_post(self, stripe_mocked_call):
|
|
|
|
# Anonymous user should get redirect to login
|
|
# response = self.client.post(self.url)
|
|
# expected_url = "%s?next=%s" % (reverse('digitalglarus:signup'),
|
|
# reverse('digitalglarus:membership_payment'))
|
|
# self.assertRedirects(response, expected_url=expected_url,
|
|
# status_code=302, target_status_code=200)
|
|
|
|
# Customer user should be able to pay
|
|
stripe_mocked_call.return_value = {
|
|
'paid': True,
|
|
'response_object': self.stripe_mocked_customer,
|
|
'error': None
|
|
}
|
|
response = self.customer_client.post(self.url, self.billing_address)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(StripeCustomer.objects.filter(user__email=self.customer.email).exists())
|
|
stripe_customer = StripeCustomer.objects.get(user__email=self.customer.email)
|
|
self.assertEqual(stripe_customer.user, self.customer)
|
|
self.assertTrue(MembershipOrder.objects.filter(customer=stripe_customer).exists())
|
|
membership_order = MembershipOrder.objects.filter(customer=stripe_customer).first()
|
|
session_data = {
|
|
'membership_price': membership_order.membership.type.first_month_price,
|
|
'membership_dates': membership_order.membership.type.first_month_formated_range
|
|
}
|
|
self.assertEqual(session_data.get('membership_price'),
|
|
self.session_data.get('membership_price'))
|
|
self.assertEqual(session_data.get('membership_dates'),
|
|
self.session_data.get('membership_dates'))
|
|
|
|
# self.assertTrue(HostingOrder.objects.filter(customer=stripe_customer).exists())
|
|
# hosting_order = HostingOrder.objects.filter(customer=stripe_customer)[0]
|
|
# vm_plan = {
|
|
# 'cores': hosting_order.vm_plan.cores,
|
|
# 'memory': hosting_order.vm_plan.memory,
|
|
# 'disk_size': hosting_order.vm_plan.disk_size,
|
|
# 'price': hosting_order.vm_plan.price,
|
|
# 'hosting_company': hosting_order.vm_plan.vm_type.hosting_company,
|
|
# 'configuration': hosting_order.vm_plan.configuration
|
|
# }
|
|
# self.assertEqual(vm_plan, self.session_data.get('vm_specs'))
|
|
|
|
|
|
class LoginViewTest(TestCase):
|
|
|
|
def setUp(self):
|
|
self.url = reverse('digitalglarus:login')
|
|
self.view = LoginView
|
|
self.expected_template = 'digitalglarus/login.html'
|
|
self.user = mommy.make('membership.CustomUser')
|
|
self.password = 'fake_password'
|
|
self.user.set_password(self.password)
|
|
self.user.save()
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTemplateUsed(response, self.expected_template)
|
|
|
|
def test_anonymous_user_can_login(self):
|
|
data = {
|
|
'email': self.user.email,
|
|
'password': self.password
|
|
}
|
|
response = self.client.post(self.url, data=data, follow=True)
|
|
self.assertEqual(response.context['user'], self.user)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
|
|
class SignupViewTest(TestCase):
|
|
|
|
def setUp(self):
|
|
self.url = reverse('digitalglarus:signup')
|
|
self.expected_template = 'digitalglarus/signup.html'
|
|
self.view = SignupView
|
|
self.signup_data = {
|
|
'name': 'ungleich',
|
|
'email': 'test@ungleich.com',
|
|
'password': 'fake_password',
|
|
'confirm_password': 'fake_password',
|
|
}
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTemplateUsed(response, self.expected_template)
|
|
|
|
def test_anonymous_user_can_signup(self):
|
|
response = self.client.post(self.url, data=self.signup_data, follow=True)
|
|
self.user = CustomUser.objects.get(email=self.signup_data.get('email'))
|
|
self.assertEqual(response.context['user'], self.user)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
|
|
class PasswordResetViewTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(PasswordResetViewTest, self).setUp()
|
|
|
|
self.url = reverse('digitalglarus:reset_password')
|
|
self.view = PasswordResetView
|
|
self.expected_template = 'digitalglarus/reset_password.html'
|
|
self.user = mommy.make('membership.CustomUser')
|
|
self.password = 'fake_password'
|
|
self.user.set_password(self.password)
|
|
self.user.save()
|
|
|
|
self.post_data = {
|
|
'email': self.user.email
|
|
}
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTemplateUsed(response, self.expected_template)
|
|
|
|
def test_post(self):
|
|
response = self.client.post(self.url, data=self.post_data, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_test_generate_email_context(self):
|
|
context = self.setup_view(self.view()).\
|
|
test_generate_email_context(self.user)
|
|
self.assertEqual(context.get('user'), self.user)
|
|
self.assertEqual(context.get('site_name'), 'ungleich')
|
|
self.assertEqual(len(context.get('token')), 24)
|
|
|
|
|
|
class PasswordResetConfirmViewTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(PasswordResetConfirmViewTest, self).setUp()
|
|
|
|
self.view = PasswordResetConfirmView
|
|
self.expected_template = 'digitalglarus/confirm_reset_password.html'
|
|
self.user = mommy.make('membership.CustomUser')
|
|
self.password = 'fake_password'
|
|
self.user.set_password(self.password)
|
|
self.user.save()
|
|
|
|
self.token = default_token_generator.make_token(self.user)
|
|
self.uid = urlsafe_base64_encode(force_bytes(self.user.pk))
|
|
self.url = reverse('digitalglarus:reset_password_confirm',
|
|
kwargs={'token': self.token, 'uidb64': self.uid})
|
|
|
|
self.post_data = {
|
|
'new_password1': 'new_password',
|
|
'new_password2': 'new_password'
|
|
}
|
|
|
|
def test_url_resolve_to_view_correctly(self):
|
|
found = resolve(self.url)
|
|
self.assertEqual(found.func.__name__, self.view.__name__)
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTemplateUsed(response, self.expected_template)
|
|
|
|
def test_post(self):
|
|
response = self.client.post(self.url, data=self.post_data, follow=True)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertTrue(not response.context['form'].errors)
|