class TestUtilsFunctions(TestCase):
def setUp(self):
self.user1 = self.make_user(username="******")
self.user2 = self.make_user(username="******")
def test_get_dialogs_with_user(self):
self.dialog = Dialog()
self.dialog.owner = self.user2
self.dialog.opponent = self.user1
self.dialog.save()
dialog = get_dialogs_with_user(self.user1, self.user2)[0]
self.assertEqual(dialog, self.dialog)
class MessageMethodTest(TestCase):
def setUp(self):
self.dialog = Dialog()
self.dialog.owner = self.make_user(username="******")
self.dialog.opponent = self.make_user(username="******")
self.dialog.save()
self.message = Message()
self.message.dialog = self.dialog
self.message.sender = self.dialog.owner
self.message.text = "text about something interesting"
self.message.save()
def test_soft_delete(self):
msg = self.message
self.message.delete()
self.assertNotIn(msg, Message.objects.all())
self.assertIn(msg, Message.all_objects.filter(is_removed=True))
class TestDialogListView(TestCase):
def setUp(self):
self.factory = RequestFactory()
self.owner_user = self.make_user(username="******")
self.oponet_user = self.make_user(username="******")
self.dialog = Dialog()
self.dialog.owner = self.owner_user
self.dialog.opponent = self.oponet_user
self.dialog.save()
self.right_dialog = self.dialog
self.dialog = Dialog()
self.dialog.owner = self.make_user(username="******")
self.dialog.opponent = self.make_user(username="******")
self.dialog.save()
def test_get_queryset(self):
request = self.factory.get(
reverse('dialogs_detail', kwargs={'username': '******'}))
request.user = self.owner_user
test_view = DialogListView()
test_view.request = request
queryset = list(test_view.get_queryset())
required_queryset = [self.right_dialog]
self.assertEqual(queryset, required_queryset)