test_models.py 2.41 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
class RegisteredUserTest(TestCase):
12
13
14
15
16
17
18
19
    """
        Test cases for the RegisteredUser Model

        - check approval and registration status flipping
        - check blocking
        - check organization field
        - check full name field
    """
David Haynes's avatar
David Haynes committed
20

21
    def setUp(self):
22
23
24
25
26
        """
            Set up any variables such as dummy objects that will be utilised in
            testing methods
        """

27
        User.objects.create(username='dhaynes', password='password')
28
29
30
31
32
33

    def test_registereduser_creation(self):
        """
            check if RegisteredUsers are actually made
        """

34
35
36
        getUser = User.objects.get(username='dhaynes')
        getRegisteredUser = RegisteredUser.objects.get(user=getUser)
        self.assertTrue(getRegisteredUser)
37
38
39
40
41
42

    def test_check_str(self):
        """
            check printing
        """

43
44
45
46
47
        getUser = User.objects.get(username='dhaynes')
        getRegisteredUser = RegisteredUser.objects.get(user=getUser)
        expected = '<Registered User: dhaynes - Approval Status: False>'
        actual = str(getRegisteredUser)
        self.assertEqual(expected, actual)
David Haynes's avatar
David Haynes committed
48

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
    def test_description_blank(self):
        """
            - add in description (blank)
        """

        getUser = User.objects.get(username='dhaynes')
        getRegisteredUser = RegisteredUser.objects.get(user=getUser)
        self.assertEqual(getRegisteredUser.description, "")

    def test_description_text(self):
        """
            - add in description
        """

        getUser = User.objects.get(username='dhaynes')
        getRegisteredUser = RegisteredUser.objects.get(user=getUser)
        getRegisteredUser.description = "We're going to build a big beautiful testcase"
        self.assertEqual(getRegisteredUser.description, "We're going to build a big beautiful testcase")


69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
"""
    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!")