Added based view test, Added billing address form test, payment view test , Added test for hetzner pricing calculation, Added test for bern pricing calculation, Fixed script to load initial pricing data

hosting/test_views.py

@@ -1,8 +1,17 @@
-from django.test import TestCase
+from unittest import mock
+from django.conf import settings
+from django.test import TestCase, RequestFactory
 from django.core.urlresolvers import reverse
 from django.core.urlresolvers import resolve
-from .models import VirtualMachineType
-from .views import DjangoHostingView, RailsHostingView, NodeJSHostingView
+
+from model_mommy import mommy
+
+
+from membership.models import CustomUser, StripeCustomer
+from .models import VirtualMachineType, HostingOrder
+from .views import DjangoHostingView, RailsHostingView, NodeJSHostingView, LoginView, SignupView,\
+    PaymentVMView
+from utils.tests import BaseTestCase
 
 
 class ProcessVMSelectionTestMixin(object):
@@ -70,3 +79,152 @@ class NodeJSHostingViewTest(TestCase, ProcessVMSelectionTestMixin):
             'email': "info@node-hosting.ch",
             'vm_types': VirtualMachineType.get_serialized_vm_types(),
         }
+
+
+class PaymentVMViewTest(BaseTestCase):
+
+    def setUp(self):
+        super(PaymentVMViewTest, self).setUp()
+
+        self.view = PaymentVMView
+
+        # VM
+        self.vm = mommy.make(VirtualMachineType, base_price=10000,
+                             memory_price=100,
+                             core_price=1000,
+                             disk_size_price=1)
+
+        # post data
+        self.billing_address = {
+            'street_address': 'street name',
+            'city': 'MyCity',
+            'postal_code': '32123123123123',
+            'country': 'VE',
+            'token': 'a23kfmslwxhkwis'
+        }
+
+        # urls
+        self.url = reverse('hosting:payment')
+
+        # Session data
+        self.session_data = {
+            'vm_specs': {
+                'hosting_company': self.vm.hosting_company,
+                'cores': 1,
+                'memory': 10,
+                'disk_size': 10000,
+                'price': 22000,
+            }
+        }
+
+        session = self.customer_client.session
+        session.update(self.session_data)
+        session.save()
+
+    def test_url_resolve_to_view_correctly(self):
+        found = resolve(self.url)
+        self.assertEqual(found.func.__name__, self.view.__name__)
+
+    @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('hosting:login'), reverse('hosting: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(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
+        }
+        self.assertEqual(vm_plan, self.session_data.get('vm_specs'))
+
+    def test_get(self):
+
+        # Anonymous user should get redirect to login
+        response = self.client.get(self.url)
+        expected_url = "%s?next=%s" % (reverse('hosting:login'), reverse('hosting: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)
+
+
+class LoginViewTest(TestCase):
+
+    def setUp(self):
+        self.url = reverse('hosting:login')
+        self.view = LoginView
+        self.expected_template = 'hosting/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('hosting:signup')
+        self.expected_template = 'hosting/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)