universalsoftware/uchat
2
1from django.test import TestCase2 3# Create your tests here.4 5 6from catalog.models import Language7from django.urls import reverse8 9 10class LanguageViewTest(TestCase):11 12 @classmethod13 def setUpTestData(cls):14 langs = [ 'Hrvatski', 'Slovenački', 'Srpski', 'Bosanski', 'Makedonski', 'Bugarski']15 for lang in langs:16 l = Language.objects.create(name=lang) ; l.save()17 18import datetime19from django.utils import timezone20 21from bot.models import Language22 23# Get user model from settings24from django.contrib.auth import get_user_model25User = get_user_model()26 27class UserViewTest(TestCase):28 29 def setUp(self):30 # Create two users31 test_user1 = User.objects.create_user(32 username='testuser1', password='testuser1')33 test_user2 = User.objects.create_user(34 username='testuser2', password='testuser2')35 36 test_user1.save()37 test_user2.save()38 39 