diff --git a/audio/tests/test_episode_model.py b/audio/tests/test_episode_model.py new file mode 100644 index 0000000..52bcd71 --- /dev/null +++ b/audio/tests/test_episode_model.py @@ -0,0 +1,24 @@ +from django.test import TestCase +from audio.models import Feed, Episode +from django.contrib.auth.models import User +from uuid import UUID +from datetime import datetime, timedelta + + +class EpisodeModelTestCase(TestCase): + + def setUp(self): + user_a = User.objects.create(username='user_a') + feed_a = Feed.objects.create(user=user_a, title='Civil War') + p_date = datetime.today() - timedelta(days=8) + Episode.objects.create( + title='Chapter I', feed=feed_a, user=user_a, + pub_date=p_date + ) + + def test_episode(self): + qs = Episode.objects.all() + self.assertEquals(len(qs), 1) + episode_a = Episode.objects.get(title='Chapter I') + self.assertTrue(isinstance(episode_a.pk, UUID)) + self.assertEquals(str(episode_a), episode_a.title) diff --git a/audio/tests/test_feed_model.py b/audio/tests/test_feed_model.py new file mode 100644 index 0000000..2fce3e7 --- /dev/null +++ b/audio/tests/test_feed_model.py @@ -0,0 +1,18 @@ +from django.test import TestCase +from audio.models import Feed +from django.contrib.auth.models import User +from uuid import UUID + + +class FeedModelTestCase(TestCase): + + def setUp(self): + user_a = User.objects.create(username='user_a') + Feed.objects.create(user=user_a, title='Civil War') + + def test_feed(self): + qs = Feed.objects.all() + self.assertEquals(len(qs), 1) + feed_a = Feed.objects.get(title='Civil War') + self.assertTrue(isinstance(feed_a.pk, UUID)) + self.assertEquals(str(feed_a), feed_a.title)