django-gallery/account/tests.py

32 lines
1.2 KiB
Python
Raw Normal View History

2018-03-09 04:54:14 +00:00
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.test import TestCase
2018-03-12 02:59:51 +00:00
from account.models import Account
from django.contrib.auth.models import User
2018-03-09 04:54:14 +00:00
# Create your tests here.
2018-03-12 02:59:51 +00:00
class AccountTestCase(TestCase):
def setUp(self):
usr1 = User.objects.create(username='usr1',
first_name='test',
last_name='1',
email='test1@example.com')
usr2 = User.objects.create(username='usr2',
first_name='test',
last_name='2',
email='test2@example.com')
Account.objects.create(user=usr1)
Account.objects.create(user=usr2)
def test_user_meta_data(self):
""" Should return some info about these user accounts """
usr1 = User.objects.get(username='usr1')
usr_acct1 = Account.objects.get(user=usr1)
usr2 = User.objects.get(username='usr2')
usr_acct2 = Account.objects.get(user=usr2)
self.assertEqual(usr_acct1.__str__(), {'user': 'usr1', 'first_login': True, 'accepted_terms': None})
self.assertEqual(usr_acct2.__str__(), {'user': 'usr2', 'first_login': True, 'accepted_terms': None})