test_models.py 1.38 KB
Newer Older
1
2
3
# Future Imports
from __future__ import unicode_literals, absolute_import, print_function, division

David Haynes's avatar
David Haynes committed
4
# Django Imports
David Haynes's avatar
David Haynes committed
5
from django.test import TestCase
6
from django.contrib.auth.models import User
David Haynes's avatar
David Haynes committed
7

David Haynes's avatar
David Haynes committed
8
9
# App Imports
from go.models import URL, RegisteredUser
David Haynes's avatar
David Haynes committed
10

David Haynes's avatar
David Haynes committed
11
12
"""
    Test cases for the RegisteredUser Model
13
14
15
16
17
18
19
20
21

    - check if RegisteredUsers are actually made
    - check approval and registration status flipping
    - check blocking
    - check printing
    - add in description
    - check organization field
    - check full name field
    - check print(RegisteredUser)
David Haynes's avatar
David Haynes committed
22
23
24
"""
class RegisteredUserTest(TestCase):

25
26
27
28
29
30
31
    def setUp(self):
        User.objects.create(username='dhaynes', password='password')

    def test_RegisteredUserCreation(self):
        getUser = User.objects.get(username='dhaynes')
        getRegisteredUser = RegisteredUser.objects.get(user=getUser)
        self.assertTrue(getRegisteredUser)
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

    def test_checkPrint(self):
        # expected = '<Registered User: %s - Approval Status: %s>' % (self.user, self.approved)

"""
    Test cases for the URL Model

    - check if URL's are actually created
    - modify clicks (social, qr, normal)
    - check expiration date creation
    - check print function
"""
class URLTest(TestCase):

    """
        Default test case, does not actually test anything
    """
    def test_Django_Test(self):
        self.assertEqual("Hello World!", "Hello World!")