def test_add_media_with_consumed_state_argument(self):
user = User('testname', 'P@ssw0rd')
db.session.add(user)
db.session.commit()
media = add_media(user.id, 'testmedianame', consumed_state='finished')
self.assertIsNotNone(media)
self.assertIn(media, db.session)
self.assertEqual(media.medianame, 'testmedianame')
self.assertEqual(media.medium, 'other')
self.assertEqual(media.consumed_state, 'finished')
def test_add_media_with_medium_argument(self):
user = User('testname', 'P@ssw0rd')
db.session.add(user)
db.session.commit()
media = add_media(user.id, 'testmedianame', medium='audio')
self.assertIsNotNone(media)
self.assertIn(media, db.session)
self.assertEqual(media.medianame, 'testmedianame')
self.assertEqual(media.medium, 'audio')
self.assertEqual(media.consumed_state, 'not started')
def test_add_media(self):
user = User('testname', 'P@ssw0rd')
db.session.add(user)
db.session.commit()
media = add_media(user.id, 'testmedianame')
self.assertIsNotNone(media)
self.assertIn(media, db.session)
self.assertEqual(media.id, 1)
self.assertEqual(media.user, 1)
self.assertEqual(media.medianame, 'testmedianame')
self.assertEqual(media.medium, 'other')
self.assertEqual(media.consumed_state, 'not started')
self.assertEqual(media.description, '')
self.assertEqual(media.order, 0)