New Model tests
This commit is contained in:
		@@ -0,0 +1,55 @@
 | 
			
		||||
from django.test import TestCase
 | 
			
		||||
from django.db import models
 | 
			
		||||
from django.contrib.auth.models import User
 | 
			
		||||
 | 
			
		||||
from lostplaces_app.models import Explorer
 | 
			
		||||
from lostplaces_app.tests import mock_user
 | 
			
		||||
 | 
			
		||||
class ExplorerTestCase(TestCase):
 | 
			
		||||
    
 | 
			
		||||
    def setUp(self):
 | 
			
		||||
        mock_user()
 | 
			
		||||
        
 | 
			
		||||
    def test_epxlorer_creation(self):
 | 
			
		||||
        '''
 | 
			
		||||
        Tests if the explorer profile will be automticly
 | 
			
		||||
        created when a user is created
 | 
			
		||||
        '''
 | 
			
		||||
        
 | 
			
		||||
        user = User.objects.get(id=1)
 | 
			
		||||
        explorer_list = Explorer.objects.all()
 | 
			
		||||
        self.assertTrue(len(explorer_list) > 0,
 | 
			
		||||
            msg='Expecting at least one Exlorer object, none found'
 | 
			
		||||
        )
 | 
			
		||||
        self.assertTrue(hasattr(user, 'explorer'),
 | 
			
		||||
            msg='''Expecting the User instance to have an \'explorer\' attribute. 
 | 
			
		||||
            Check the Explorer model and the related name.'''
 | 
			
		||||
        )
 | 
			
		||||
        
 | 
			
		||||
        explorer = Explorer.objects.get(id=1)
 | 
			
		||||
        self.assertEqual(explorer, user.explorer,
 | 
			
		||||
            msg='''The Explorer object of the User did not match.
 | 
			
		||||
            Expecting User with id 1 to have Explorer with id 1'''
 | 
			
		||||
        )
 | 
			
		||||
        
 | 
			
		||||
        explorer = Explorer.objects.get(id=1)
 | 
			
		||||
        self.assertEqual(explorer.user, user,
 | 
			
		||||
            msg='''The User object of the Explorer did not match.
 | 
			
		||||
            Expecting Explorer with id 1 to have User with id 1'''
 | 
			
		||||
        )
 | 
			
		||||
        
 | 
			
		||||
    def test_explorer_deletion(self):
 | 
			
		||||
        '''
 | 
			
		||||
        Tests if the Explorer objects get's deleted when the User instance is deleted
 | 
			
		||||
        '''
 | 
			
		||||
        
 | 
			
		||||
        user = User.objects.get(id=1)
 | 
			
		||||
        explorer_id = user.explorer.id
 | 
			
		||||
        user.delete()
 | 
			
		||||
        with self.assertRaises(models.ObjectDoesNotExist,
 | 
			
		||||
            msg='Expecting explorer objec to be deleted when the corresponding User object is deleted'
 | 
			
		||||
            ):
 | 
			
		||||
            Explorer.objects.get(id=explorer_id)
 | 
			
		||||
        
 | 
			
		||||
        
 | 
			
		||||
    
 | 
			
		||||
							
								
								
									
										51
									
								
								lostplaces/lostplaces_app/tests/models/test_voucher_model.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								lostplaces/lostplaces_app/tests/models/test_voucher_model.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,51 @@
 | 
			
		||||
import datetime
 | 
			
		||||
 | 
			
		||||
from django.test import TestCase
 | 
			
		||||
from django.db import models
 | 
			
		||||
 | 
			
		||||
from lostplaces_app.models import Voucher
 | 
			
		||||
from lostplaces_app.tests.models import ModelTestCase
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def mock_voucher():
 | 
			
		||||
    return Voucher.objects.create(
 | 
			
		||||
        code='ayDraJCCwfhcFiYmSR5GrcjcchDfcahv',
 | 
			
		||||
        expires=datetime.datetime.now() + datetime.timedelta(days=1)
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class VoucheTestCase(ModelTestCase, TestCase):
 | 
			
		||||
    model_name = 'Voucher'
 | 
			
		||||
 | 
			
		||||
    def setUp(self):
 | 
			
		||||
        self.object = mock_voucher()
 | 
			
		||||
 | 
			
		||||
    def test_voucher_code(self):
 | 
			
		||||
        self._test_char_field(
 | 
			
		||||
            'code',
 | 
			
		||||
            10,
 | 
			
		||||
            100,
 | 
			
		||||
            must_have={'unique': True}
 | 
			
		||||
        )
 | 
			
		||||
        
 | 
			
		||||
    def test_voucher_created(self):
 | 
			
		||||
        self._test_field(
 | 
			
		||||
            'created_when',
 | 
			
		||||
            models.DateTimeField,
 | 
			
		||||
            must_have={'auto_now_add': True}
 | 
			
		||||
        )
 | 
			
		||||
        
 | 
			
		||||
    def test_voucher_expires(self):
 | 
			
		||||
        self._test_field(
 | 
			
		||||
            'expires_when',
 | 
			
		||||
            models.DateTimeField,
 | 
			
		||||
            must_not_have={'auto_now_add': True}
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def test_str(self):
 | 
			
		||||
        voucher = self.object
 | 
			
		||||
        self.assertTrue(voucher.code.lower() in str(voucher).lower(),
 | 
			
		||||
            msg='Expecting %s.__str__ to contain the voucher code' % (
 | 
			
		||||
                self.model_name
 | 
			
		||||
            )
 | 
			
		||||
        )
 | 
			
		||||
		Reference in New Issue
	
	Block a user