Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import factory
- from .models import *
- class UserFacTory(factory.DjangoModelFactory):
- class Meta:
- model = User
- username = factory.Sequence(lambda n: 'user{0}'.format(n))
- class ChatFactory(factory.DjangoModelFactory):
- class Meta:
- model = BaseChanel
- name_chanel = factory.Sequence(lambda n: 'chanel{0}'.format(n))
- # creat_time = factory.LazyAttribute(lambda o: datetime.datetime.now)
- creat_time = factory.LazyFunction(timezone.now)
- author = factory.SubFactory(UserFacTory)
- # users_chanel = factory.SubFactory(User)
- class MessageChatFactory(factory.DjangoModelFactory):
- class Meta:
- model = MessageChat
- chat = factory.SubFactory(ChatFactory)
- author_message = factory.SubFactory(UserFacTory)
- message = 'message'
- created_message = factory.LazyFunction(timezone.now)
- from django.test import TestCase
- import pytest
- from chat.factories import ChatFactory, MessageChatFactory, timezone, datetime
- @pytest.mark.django_db
- class TestChatFactory(TestCase):
- # def setUp(self):
- # self.chat = ChatFactory.create(username='user0')
- def test_basechanel_name(self):
- chat1 = ChatFactory()
- self.assertEqual(chat1.name_chanel, 'chanel0')
- self.assertEqual(chat1.author.username, 'user0')
- # self.assertEqual(chat1.creat_time, timezone.now()) #какой то непонятный рассинхрон времени
- # ChatFactory.create()
- # self.assertEqual(BaseChanel.objects.count(),1)
- # self.assertFalse(BaseChanel.objects.exists())
- # assert ChatFactory.exists()
- class Test_Message_Factory(TestCase):
- def test_message_chat_factory(self):
- messag1 = MessageChatFactory()
- self.assertEqual(messag1.chat.name_chanel, 'chanel1')
- self.assertEqual(messag1.message, 'message')
- self.assertEqual(messag1.author_message.username, 'user2')
- self.assertEqual(messag1.created_message, timezone.now()) # этот работает правильно
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement