Merge pull request #72 from ungleich/develop
Update develop from master because of the hotfix for blog posts
This commit is contained in:
		
				commit
				
					
						ac51ce9290
					
				
			
		
					 15 changed files with 528 additions and 39 deletions
				
			
		|  | @ -1,6 +1,5 @@ | |||
| from django.contrib import admin | ||||
| from .models import RailsBetaUser, VirtualMachineType | ||||
| from .models import VirtualMachineType | ||||
| 
 | ||||
| 
 | ||||
| admin.site.register(RailsBetaUser) | ||||
| admin.site.register(VirtualMachineType) | ||||
|  |  | |||
|  | @ -10,8 +10,8 @@ class Command(BaseCommand): | |||
|         hetzner = { | ||||
|             'base_price': 10, | ||||
|             'core_price': 10, | ||||
|             'memory_price': 10, | ||||
|             'disk_size_price': 10, | ||||
|             'memory_price': 5, | ||||
|             'disk_size_price': 1, | ||||
|             'description': 'VM auf einzelner HW, Raid1, kein HA' | ||||
|         } | ||||
| 
 | ||||
|  |  | |||
							
								
								
									
										20
									
								
								hosting/migrations/0015_auto_20160512_0448.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								hosting/migrations/0015_auto_20160512_0448.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,20 @@ | |||
| # -*- coding: utf-8 -*- | ||||
| # Generated by Django 1.9.4 on 2016-05-12 04:48 | ||||
| from __future__ import unicode_literals | ||||
| 
 | ||||
| from django.db import migrations | ||||
| 
 | ||||
| 
 | ||||
| class Migration(migrations.Migration): | ||||
| 
 | ||||
|     dependencies = [ | ||||
|         ('hosting', '0014_auto_20160505_0541'), | ||||
|     ] | ||||
| 
 | ||||
|     operations = [ | ||||
|         migrations.RenameField( | ||||
|             model_name='hostingorder', | ||||
|             old_name='VMPlan', | ||||
|             new_name='vm_plan', | ||||
|         ), | ||||
|     ] | ||||
							
								
								
									
										18
									
								
								hosting/migrations/0016_delete_railsbetauser.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								hosting/migrations/0016_delete_railsbetauser.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,18 @@ | |||
| # -*- coding: utf-8 -*- | ||||
| # Generated by Django 1.9.4 on 2016-05-14 06:50 | ||||
| from __future__ import unicode_literals | ||||
| 
 | ||||
| from django.db import migrations | ||||
| 
 | ||||
| 
 | ||||
| class Migration(migrations.Migration): | ||||
| 
 | ||||
|     dependencies = [ | ||||
|         ('hosting', '0015_auto_20160512_0448'), | ||||
|     ] | ||||
| 
 | ||||
|     operations = [ | ||||
|         migrations.DeleteModel( | ||||
|             name='RailsBetaUser', | ||||
|         ), | ||||
|     ] | ||||
|  | @ -10,14 +10,6 @@ from utils.models import BillingAddress | |||
| from .managers import VMPlansManager | ||||
| 
 | ||||
| 
 | ||||
| class RailsBetaUser(models.Model): | ||||
|     email = models.EmailField(unique=True) | ||||
|     received_date = models.DateTimeField('date received') | ||||
| 
 | ||||
|     def __str__(self): | ||||
|         return "%s - %s" % (self.email, self.received_date) | ||||
| 
 | ||||
| 
 | ||||
| class VirtualMachineType(models.Model): | ||||
| 
 | ||||
|     HETZNER_NUG = 'hetzner_nug' | ||||
|  | @ -86,6 +78,9 @@ class VirtualMachinePlan(models.Model): | |||
| 
 | ||||
|     objects = VMPlansManager() | ||||
| 
 | ||||
|     def __str__(self): | ||||
|         return "%s" % (self.id) | ||||
| 
 | ||||
|     @cached_property | ||||
|     def hosting_company_name(self): | ||||
|         return self.vm_type.get_hosting_company_display() | ||||
|  | @ -106,7 +101,7 @@ class HostingOrder(models.Model): | |||
|     ORDER_APPROVED_STATUS = 'Approved' | ||||
|     ORDER_DECLINED_STATUS = 'Declined' | ||||
| 
 | ||||
|     VMPlan = models.ForeignKey(VirtualMachinePlan, related_name='hosting_orders') | ||||
|     vm_plan = models.ForeignKey(VirtualMachinePlan, related_name='hosting_orders') | ||||
|     customer = models.ForeignKey(StripeCustomer) | ||||
|     billing_address = models.ForeignKey(BillingAddress) | ||||
|     created_at = models.DateTimeField(auto_now_add=True) | ||||
|  | @ -115,13 +110,16 @@ class HostingOrder(models.Model): | |||
|     cc_brand = models.CharField(max_length=10) | ||||
|     stripe_charge_id = models.CharField(max_length=100, null=True) | ||||
| 
 | ||||
|     def __str__(self): | ||||
|         return "%s" % (self.id) | ||||
| 
 | ||||
|     @cached_property | ||||
|     def status(self): | ||||
|         return self.ORDER_APPROVED_STATUS if self.approved else self.ORDER_DECLINED_STATUS | ||||
| 
 | ||||
|     @classmethod | ||||
|     def create(cls, VMPlan=None, customer=None, billing_address=None): | ||||
|         instance = cls.objects.create(VMPlan=VMPlan, customer=customer, | ||||
|     def create(cls, vm_plan=None, customer=None, billing_address=None): | ||||
|         instance = cls.objects.create(vm_plan=vm_plan, customer=customer, | ||||
|                                       billing_address=billing_address) | ||||
|         return instance | ||||
| 
 | ||||
|  |  | |||
|  | @ -6,7 +6,7 @@ | |||
|     <div class="row"> | ||||
|         <div class="col-xs-8 col-xs-offset-2"> | ||||
|     		<div class="invoice-title"> | ||||
|     			<h2>Invoice</h2><h3 class="pull-right">Order # {{object.id}}</h3> | ||||
|     			<h2>Invoice</h2><h3 class="pull-right">Order # {{order.id}}</h3> | ||||
|     		</div> | ||||
|     		<hr> | ||||
|     		<div class="row"> | ||||
|  | @ -14,18 +14,18 @@ | |||
|     				<address> | ||||
|                     <h3><b>Billed To:</b></h3> | ||||
|     					{{user.name}}<br> | ||||
|                         {{object.billing_address.street_address}},{{order.billing_address.postal_code}}<br> | ||||
|                         {{object.billing_address.city}}, {{object.billing_address.country}}. | ||||
|                         {{order.billing_address.street_address}},{{order.billing_address.postal_code}}<br> | ||||
|                         {{order.billing_address.city}}, {{order.billing_address.country}}. | ||||
|     				</address> | ||||
|     			</div> | ||||
|                 <div class="col-xs-6 text-right"> | ||||
|                     <address> | ||||
|                         <strong>Order Date:</strong><br> | ||||
|                         {{object.created_at}}<br><br> | ||||
|                         {{order.created_at}}<br><br> | ||||
|                         <strong>Status:</strong><br> | ||||
|                         <strong class="{% if object.status == 'Approved' %}text-success | ||||
|                         <strong class="{% if order.status == 'Approved' %}text-success | ||||
|                                        {%else%} text-danger | ||||
|                                        {% endif %}">{{object.status}}</strong> | ||||
|                                        {% endif %}">{{order.status}}</strong> | ||||
|                         <br><br> | ||||
|                     </address> | ||||
| 
 | ||||
|  | @ -35,7 +35,7 @@ | |||
|     			<div class="col-xs-6"> | ||||
|     				<address> | ||||
|     					<strong>Payment Method:</strong><br> | ||||
|     					{{object.cc_brand}} ending **** {{object.last4}}<br> | ||||
|     					{{order.cc_brand}} ending **** {{order.last4}}<br> | ||||
|     					{{user.email}} | ||||
|     				</address> | ||||
|     			</div> | ||||
|  | @ -48,15 +48,15 @@ | |||
|             <h3><b>Order summary</b></h3> | ||||
|             <hr> | ||||
|             <div class="content"> | ||||
|                 <p><b>Type</b> <span class="pull-right">{{object.VMPlan.hosting_company_name}}</span></p> | ||||
|                 <p><b>Type</b> <span class="pull-right">{{order.vm_plan.hosting_company_name}}</span></p> | ||||
|                 <hr> | ||||
|                 <p><b>Cores</b> <span class="pull-right">{{object.VMPlan.cores}}</span></p> | ||||
|                 <p><b>Cores</b> <span class="pull-right">{{order.vm_plan.cores}}</span></p> | ||||
|                 <hr> | ||||
|                 <p><b>Memory</b> <span class="pull-right">{{object.VMPlan.memory}} GiB</span></p> | ||||
|                 <p><b>Memory</b> <span class="pull-right">{{order.vm_plan.memory}} GiB</span></p> | ||||
|                 <hr> | ||||
|                 <p><b>Disk space</b> <span class="pull-right">{{object.VMPlan.disk_size}} GiB</span></p> | ||||
|                 <p><b>Disk space</b> <span class="pull-right">{{order.vm_plan.disk_size}} GiB</span></p> | ||||
|                 <hr> | ||||
|                 <h4>Total<p class="pull-right"><b>{{object.VMPlan.price}} CHF</b></p></h4> | ||||
|                 <h4>Total<p class="pull-right"><b>{{order.vm_plan.price}} CHF</b></p></h4> | ||||
|             </div> | ||||
|         </div> | ||||
|     </div> | ||||
|  |  | |||
|  | @ -23,7 +23,7 @@ | |||
| 					<tr>  | ||||
| 						<td scope="row">{{order.id}}</td>  | ||||
| 						<td>{{order.created_at}}</td>  | ||||
| 						<td>{{order.VMPlan.price}} CHF</td>  | ||||
| 						<td>{{order.vm_plan.price}} CHF</td>  | ||||
| 						<td>{% if order.approved %} | ||||
| 								<span class="text-success strong">Approved</span> | ||||
| 							{% else%}  | ||||
|  |  | |||
|  | @ -104,7 +104,7 @@ | |||
| 											<tr>  | ||||
| 												<td scope="row">{{order.id}}</td>  | ||||
| 												<td>{{order.created_at}}</td>  | ||||
| 												<td>{{order.VMPlan.price}} CHF</td>  | ||||
| 												<td>{{order.vm_plan.price}} CHF</td>  | ||||
| 												<td>{% if order.approved %} | ||||
| 														<span class="text-success strong">Approved</span> | ||||
| 													{% else%}  | ||||
|  |  | |||
							
								
								
									
										75
									
								
								hosting/test_models.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										75
									
								
								hosting/test_models.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,75 @@ | |||
| from django.test import TestCase | ||||
| 
 | ||||
| from django.core.management import call_command | ||||
| 
 | ||||
| 
 | ||||
| from .models import VirtualMachineType | ||||
| 
 | ||||
| 
 | ||||
| class VirtualMachineTypeModelTest(TestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         self.HETZNER_NUG_NAME = 'hetzner_nug' | ||||
|         self.HETZNER_NAME = 'hetzner' | ||||
|         self.HETZNER_RAID6_NAME = 'hetzner_raid6' | ||||
|         self.HETZNER_GLUSTERFS_NAME = 'hetzner_glusterfs' | ||||
|         self.BERN_NAME = 'bern' | ||||
|         self.HETZNER_NUG_EXPECTED_PRICE = 79 | ||||
|         self.HETZNER_EXPECTED_PRICE = 180 | ||||
|         self.HETZNER_RAID6_EXPECTED_PRICE = 216 | ||||
|         self.HETZNER_GLUSTERFS_EXPECTED_PRICE = 252 | ||||
|         self.BERN_EXPECTED_PRICE = 202 | ||||
| 
 | ||||
|         call_command('create_vm_types') | ||||
| 
 | ||||
|     def test_calculate_price(self): | ||||
| 
 | ||||
|         # hetzner_nug | ||||
|         specifications = { | ||||
|             'cores': 2, | ||||
|             'memory': 10, | ||||
|             'disk_size': 100 | ||||
|         } | ||||
|         vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_NUG_NAME) | ||||
|         calculated_price = vm_type.calculate_price(specifications) | ||||
|         self.assertEqual(calculated_price, self.HETZNER_NUG_EXPECTED_PRICE) | ||||
| 
 | ||||
|         # hetzner | ||||
|         specifications = { | ||||
|             'cores': 2, | ||||
|             'memory': 10, | ||||
|             'disk_size': 100 | ||||
|         } | ||||
|         vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_NAME) | ||||
|         calculated_price = vm_type.calculate_price(specifications) | ||||
|         self.assertEqual(calculated_price, self.HETZNER_EXPECTED_PRICE) | ||||
| 
 | ||||
|         # hetzner_raid6 | ||||
|         specifications = { | ||||
|             'cores': 2, | ||||
|             'memory': 10, | ||||
|             'disk_size': 100 | ||||
|         } | ||||
|         vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_RAID6_NAME) | ||||
|         calculated_price = vm_type.calculate_price(specifications) | ||||
|         self.assertEqual(calculated_price, self.HETZNER_RAID6_EXPECTED_PRICE) | ||||
| 
 | ||||
|         # hetzner_glusterfs | ||||
|         specifications = { | ||||
|             'cores': 2, | ||||
|             'memory': 10, | ||||
|             'disk_size': 100 | ||||
|         } | ||||
|         vm_type = VirtualMachineType.objects.get(hosting_company=self.HETZNER_GLUSTERFS_NAME) | ||||
|         calculated_price = vm_type.calculate_price(specifications) | ||||
|         self.assertEqual(calculated_price, self.HETZNER_GLUSTERFS_EXPECTED_PRICE) | ||||
| 
 | ||||
|         # bern | ||||
|         specifications = { | ||||
|             'cores': 2, | ||||
|             'memory': 10, | ||||
|             'disk_size': 100 | ||||
|         } | ||||
|         vm_type = VirtualMachineType.objects.get(hosting_company=self.BERN_NAME) | ||||
|         calculated_price = vm_type.calculate_price(specifications) | ||||
|         self.assertEqual(calculated_price, self.BERN_EXPECTED_PRICE) | ||||
|  | @ -1,8 +1,18 @@ | |||
| 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, VirtualMachinePlan | ||||
| from .views import DjangoHostingView, RailsHostingView, NodeJSHostingView, LoginView, SignupView, \ | ||||
|     PaymentVMView, OrdersHostingDetailView, OrdersHostingListView, VirtualMachineDetailView, \ | ||||
|     VirtualMachinesPlanListView | ||||
| from utils.tests import BaseTestCase | ||||
| 
 | ||||
| 
 | ||||
| class ProcessVMSelectionTestMixin(object): | ||||
|  | @ -70,3 +80,280 @@ 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 VirtualMachineDetailViewTest(BaseTestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         super(VirtualMachineDetailViewTest, self).setUp() | ||||
| 
 | ||||
|         self.stripe_customer = mommy.make(StripeCustomer, user=self.customer) | ||||
|         self.vm = mommy.make(VirtualMachinePlan) | ||||
|         self.order = mommy.make(HostingOrder, customer=self.stripe_customer, vm_plan=self.vm) | ||||
|         self.url = reverse('hosting:virtual_machines', kwargs={'pk': self.vm.id}) | ||||
|         self.view = VirtualMachineDetailView() | ||||
|         self.expected_template = 'hosting/virtual_machine_detail.html' | ||||
| 
 | ||||
|     def 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('hosting:login'), | ||||
|                                        reverse('hosting:virtual_machines', | ||||
|                                        kwargs={'pk': self.vm.id})) | ||||
|         self.assertRedirects(response, expected_url=expected_url, | ||||
|                              status_code=302, target_status_code=200) | ||||
| 
 | ||||
|         # Customer should be able to get data | ||||
|         response = self.customer_client.get(self.url, follow=True) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         self.assertEqual(response.context['virtual_machine'], self.vm) | ||||
|         self.assertTemplateUsed(response, self.expected_template) | ||||
| 
 | ||||
| 
 | ||||
| class VirtualMachinesPlanListViewTest(BaseTestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         super(VirtualMachinesPlanListViewTest, self).setUp() | ||||
| 
 | ||||
|         self.stripe_customer = mommy.make(StripeCustomer, user=self.customer) | ||||
|         mommy.make(HostingOrder, customer=self.stripe_customer, approved=True, _quantity=20) | ||||
|         _vms = VirtualMachinePlan.objects.all() | ||||
|         self.vms = sorted(_vms, key=lambda vm: vm.id, reverse=True) | ||||
|         self.url = reverse('hosting:virtual_machines') | ||||
|         self.view = VirtualMachinesPlanListView() | ||||
|         self.expected_template = 'hosting/virtual_machines.html' | ||||
| 
 | ||||
|     def 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('hosting:login'), | ||||
|                                        reverse('hosting:virtual_machines')) | ||||
|         self.assertRedirects(response, expected_url=expected_url, | ||||
|                              status_code=302, target_status_code=200) | ||||
| 
 | ||||
|         # Customer should be able to get his orders | ||||
|         response = self.customer_client.get(self.url, follow=True) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         self.assertEqual(list(response.context['vms']), self.vms[:10]) | ||||
|         self.assertTemplateUsed(response, self.expected_template) | ||||
| 
 | ||||
| 
 | ||||
| class OrderHostingDetailViewTest(BaseTestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         super(OrderHostingDetailViewTest, self).setUp() | ||||
| 
 | ||||
|         self.stripe_customer = mommy.make(StripeCustomer, user=self.customer) | ||||
|         self.order = mommy.make(HostingOrder, customer=self.stripe_customer) | ||||
|         self.url = reverse('hosting:orders', kwargs={'pk': self.order.id}) | ||||
|         self.view = OrdersHostingDetailView() | ||||
|         self.expected_template = 'hosting/order_detail.html' | ||||
| 
 | ||||
|     def 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('hosting:login'), | ||||
|                                        reverse('hosting:orders', kwargs={'pk': self.order.id})) | ||||
|         self.assertRedirects(response, expected_url=expected_url, | ||||
|                              status_code=302, target_status_code=200) | ||||
| 
 | ||||
|         # Customer should be able to get data | ||||
|         response = self.customer_client.get(self.url, follow=True) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         self.assertEqual(response.context['order'], self.order) | ||||
|         self.assertTemplateUsed(response, self.expected_template) | ||||
| 
 | ||||
| 
 | ||||
| class OrdersHostingListViewTest(BaseTestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         super(OrdersHostingListViewTest, self).setUp() | ||||
| 
 | ||||
|         self.stripe_customer = mommy.make(StripeCustomer, user=self.customer) | ||||
|         _orders = mommy.make(HostingOrder, customer=self.stripe_customer, _quantity=20) | ||||
|         self.orders = sorted(_orders, key=lambda order: order.id, reverse=True) | ||||
|         self.url = reverse('hosting:orders') | ||||
|         self.view = OrdersHostingListView() | ||||
|         self.expected_template = 'hosting/orders.html' | ||||
| 
 | ||||
|     def 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('hosting:login'), reverse('hosting:orders')) | ||||
|         self.assertRedirects(response, expected_url=expected_url, | ||||
|                              status_code=302, target_status_code=200) | ||||
| 
 | ||||
|         # Customer should be able to get his orders | ||||
|         response = self.customer_client.get(self.url, follow=True) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         self.assertEqual(list(response.context['orders']), self.orders[:10]) | ||||
|         self.assertTemplateUsed(response, self.expected_template) | ||||
| 
 | ||||
| 
 | ||||
| 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) | ||||
|  |  | |||
|  | @ -3,7 +3,7 @@ from django.conf.urls import url | |||
| from .views import DjangoHostingView, RailsHostingView, PaymentVMView, \ | ||||
|     NodeJSHostingView, LoginView, SignupView, IndexView, \ | ||||
|     OrdersHostingListView, OrdersHostingDetailView, VirtualMachinesPlanListView,\ | ||||
|     VirtualMachineDetailListView | ||||
|     VirtualMachineDetailView | ||||
| 
 | ||||
| urlpatterns = [ | ||||
|     # url(r'pricing/?$', VMPricingView.as_view(), name='pricing'), | ||||
|  | @ -15,7 +15,7 @@ urlpatterns = [ | |||
|     url(r'orders/?$', OrdersHostingListView.as_view(), name='orders'), | ||||
|     url(r'orders/(?P<pk>\d+)/?$', OrdersHostingDetailView.as_view(), name='orders'), | ||||
|     url(r'my-virtual-machines/?$', VirtualMachinesPlanListView.as_view(), name='virtual_machines'), | ||||
|     url(r'my-virtual-machines/(?P<pk>\d+)/?$', VirtualMachineDetailListView.as_view(), | ||||
|     url(r'my-virtual-machines/(?P<pk>\d+)/?$', VirtualMachineDetailView.as_view(), | ||||
|         name='virtual_machines'), | ||||
|     url(r'login/?$', LoginView.as_view(), name='login'), | ||||
|     url(r'signup/?$', SignupView.as_view(), name='signup'), | ||||
|  |  | |||
|  | @ -143,8 +143,9 @@ class SignupView(CreateView): | |||
|         return HttpResponseRedirect(self.get_success_url()) | ||||
| 
 | ||||
| 
 | ||||
| class PaymentVMView(FormView): | ||||
| class PaymentVMView(LoginRequiredMixin, FormView): | ||||
|     template_name = 'hosting/payment.html' | ||||
|     login_url = reverse_lazy('hosting:login') | ||||
|     form_class = BillingAddressForm | ||||
| 
 | ||||
|     def get_context_data(self, **kwargs): | ||||
|  | @ -183,7 +184,7 @@ class PaymentVMView(FormView): | |||
|             billing_address = form.save() | ||||
| 
 | ||||
|             # Create a Hosting Order | ||||
|             order = HostingOrder.create(VMPlan=plan, customer=customer, | ||||
|             order = HostingOrder.create(vm_plan=plan, customer=customer, | ||||
|                                         billing_address=billing_address) | ||||
| 
 | ||||
|             # Make stripe charge to a customer | ||||
|  | @ -219,6 +220,7 @@ class PaymentVMView(FormView): | |||
| 
 | ||||
| class OrdersHostingDetailView(LoginRequiredMixin, DetailView): | ||||
|     template_name = "hosting/order_detail.html" | ||||
|     context_object_name = "order" | ||||
|     login_url = reverse_lazy('hosting:login') | ||||
|     model = HostingOrder | ||||
| 
 | ||||
|  | @ -229,6 +231,7 @@ class OrdersHostingListView(LoginRequiredMixin, ListView): | |||
|     context_object_name = "orders" | ||||
|     model = HostingOrder | ||||
|     paginate_by = 10 | ||||
|     ordering = '-id' | ||||
| 
 | ||||
|     def get_queryset(self): | ||||
|         user = self.request.user | ||||
|  | @ -242,6 +245,7 @@ class VirtualMachinesPlanListView(LoginRequiredMixin, ListView): | |||
|     context_object_name = "vms" | ||||
|     model = VirtualMachinePlan | ||||
|     paginate_by = 10 | ||||
|     ordering = '-id' | ||||
| 
 | ||||
|     def get_queryset(self): | ||||
|         user = self.request.user | ||||
|  | @ -249,7 +253,7 @@ class VirtualMachinesPlanListView(LoginRequiredMixin, ListView): | |||
|         return super(VirtualMachinesPlanListView, self).get_queryset() | ||||
| 
 | ||||
| 
 | ||||
| class VirtualMachineDetailListView(LoginRequiredMixin, DetailView): | ||||
| class VirtualMachineDetailView(LoginRequiredMixin, DetailView): | ||||
|     template_name = "hosting/virtual_machine_detail.html" | ||||
|     login_url = reverse_lazy('hosting:login') | ||||
|     model = VirtualMachinePlan | ||||
|  |  | |||
|  | @ -16,6 +16,7 @@ psycopg2 | |||
| django-mptt | ||||
| easy_thumbnails | ||||
| django-polymorphic | ||||
| model-mommy | ||||
| 
 | ||||
| #PLUGINS | ||||
| djangocms_flash | ||||
|  |  | |||
|  | @ -1,5 +1,5 @@ | |||
| from django.test import TestCase | ||||
| from .forms import ContactUsForm | ||||
| from .forms import ContactUsForm, BillingAddressForm | ||||
| 
 | ||||
| 
 | ||||
| class ContactUsFormTest(TestCase): | ||||
|  | @ -23,3 +23,27 @@ class ContactUsFormTest(TestCase): | |||
|     def test_invalid_form(self): | ||||
|         form = ContactUsForm(data=self.incompleted_data) | ||||
|         self.assertFalse(form.is_valid()) | ||||
| 
 | ||||
| 
 | ||||
| class BillingAddressFormTest(TestCase): | ||||
| 
 | ||||
|     def setUp(self): | ||||
|         self.completed_data = { | ||||
|             'street_address': 'street name', | ||||
|             'city': 'MyCity', | ||||
|             'postal_code': '32123123123123', | ||||
|             'country': 'VE', | ||||
|             'token': 'a23kfmslwxhkwis' | ||||
|         } | ||||
| 
 | ||||
|         self.incompleted_data = { | ||||
|             'street_address': 'test', | ||||
|         } | ||||
| 
 | ||||
|     def test_valid_form(self): | ||||
|         form = BillingAddressForm(data=self.completed_data) | ||||
|         self.assertTrue(form.is_valid()) | ||||
| 
 | ||||
|     def test_invalid_form(self): | ||||
|         form = BillingAddressForm(data=self.incompleted_data) | ||||
|         self.assertFalse(form.is_valid()) | ||||
|  |  | |||
|  | @ -1,3 +1,66 @@ | |||
| from django.test import TestCase | ||||
| from django.test import Client | ||||
| from model_mommy import mommy | ||||
| 
 | ||||
| # Create your tests here. | ||||
| 
 | ||||
| class BaseTestCase(TestCase): | ||||
|     """ | ||||
|     Base class to initialize the test cases | ||||
|     """ | ||||
| 
 | ||||
|     def setUp(self): | ||||
| 
 | ||||
|         # Password | ||||
|         self.dummy_password = 'test_password' | ||||
| 
 | ||||
|         # Users | ||||
|         self.customer, self.another_customer = mommy.make('membership.CustomUser', | ||||
|                                                           _quantity=2) | ||||
|         self.customer.set_password(self.dummy_password) | ||||
|         self.customer.save() | ||||
|         self.another_customer.set_password(self.dummy_password) | ||||
|         self.another_customer.save() | ||||
| 
 | ||||
|         # Stripe mocked data | ||||
|         self.stripe_mocked_customer = self.customer_stripe_mocked_data() | ||||
| 
 | ||||
|         #  Clients | ||||
|         self.customer_client = self.get_client(self.customer) | ||||
|         self.another_customer_client = self.get_client(self.another_customer) | ||||
| 
 | ||||
|     def get_client(self, user): | ||||
|         """ | ||||
|         Authenticate a user and return the client | ||||
|         """ | ||||
|         client = Client() | ||||
|         client.login(email=user.email, password=self.dummy_password) | ||||
|         return client | ||||
| 
 | ||||
|     def customer_stripe_mocked_data(self): | ||||
|         return { | ||||
|             "id": "cus_8R1y9UWaIIjZqr", | ||||
|             "object": "customer", | ||||
|             "currency": "usd", | ||||
|             "default_source": "card_18A9up2eZvKYlo2Cq2RJMGeF", | ||||
|             "email": "vmedixtodd+1@gmail.com", | ||||
|             "livemode": False, | ||||
|             "metadata": { | ||||
|             }, | ||||
|             "shipping": None, | ||||
|             "sources": { | ||||
|                 "object": "list", | ||||
|                 "data": [{ | ||||
|                     "id": "card_18A9up2eZvKYlo2Cq2RJMGeF", | ||||
|                     "object": "card", | ||||
|                     "brand": "Visa", | ||||
|                     "country": "US", | ||||
|                     "customer": "cus_8R1y9UWaIIjZqr", | ||||
|                     "cvc_check": "pass", | ||||
|                     "dynamic_last4": None, | ||||
|                     "exp_month": 12, | ||||
|                     "exp_year": 2018, | ||||
|                     "funding": "credit", | ||||
|                     "last4": "4242", | ||||
|                 }] | ||||
|             } | ||||
|         } | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue