test_models.py 875 Bytes
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
13
"""
    Test cases for the URL Model
"""
David Haynes's avatar
David Haynes committed
14
class URLTest(TestCase):
David Haynes's avatar
David Haynes committed
15

David Haynes's avatar
David Haynes committed
16
17
18
    """
        Default test case, does not actually test anything
    """
David Haynes's avatar
David Haynes committed
19
    def test_Django_Test(self):
David Haynes's avatar
David Haynes committed
20
        self.assertEqual("Hello World!", "Hello World!")
David Haynes's avatar
David Haynes committed
21
22
23
24
25
26

"""
    Test cases for the RegisteredUser Model
"""
class RegisteredUserTest(TestCase):

27
28
29
30
31
32
33
    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)