From b21d5a4a62a2c67c4dd0b1913d7fc7ffc870340b Mon Sep 17 00:00:00 2001 From: Trent Palmer Date: Sun, 11 Apr 2021 23:33:25 -0700 Subject: [PATCH] extend audio/tests/test_new_episode_view.py --- audio/tests/test_new_episode_view.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/audio/tests/test_new_episode_view.py b/audio/tests/test_new_episode_view.py index 8373d55..922985e 100644 --- a/audio/tests/test_new_episode_view.py +++ b/audio/tests/test_new_episode_view.py @@ -1,4 +1,5 @@ from django.test import TestCase +from django.contrib.auth.models import User from audio.tests.test_views import set_up from audio.models import Episode, Feed from django.urls import reverse @@ -50,6 +51,7 @@ class TestNewEpisodeViewTestCase(TestCase): def test_new_episode_view(self): feed_a = Feed.objects.get(title="Short Stories Mark Twain") + user_b = User.objects.get(username='user_b') self.client.login(username='user_b', password='password_user_b') kw_args = {'feed_pk': feed_a.pk, 'feed_title_slug': feed_a.slug} pub_date = str(date.today()) @@ -95,3 +97,5 @@ class TestNewEpisodeViewTestCase(TestCase): self.assertEquals( episode_a.image_license_url, 'https://en.wikipedia.org/wiki/Public_domain' ) + self.assertEquals(episode_a.user, user_b) + self.assertEquals(episode_a.feed, feed_a)