Unverified Commit e7c9659b by M. Rehan Committed by GitHub

Merge pull request #139 from edx/mrehan/video-existence-api-util

Add video existence api util
parents 2830e2e1 16738b63
...@@ -195,6 +195,16 @@ def update_video_status(edx_video_id, status): ...@@ -195,6 +195,16 @@ def update_video_status(edx_video_id, status):
video.save() video.save()
def is_video_available(edx_video_id):
"""
Returns whether a video exists given a video ID.
Arguments:
edx_video_id: A video ID representing a video record in db.
"""
return Video.objects.filter(edx_video_id=edx_video_id).exists()
def get_transcript_credentials_state_for_org(org, provider=None): def get_transcript_credentials_state_for_org(org, provider=None):
""" """
Returns transcript credentials state for an org Returns transcript credentials state for an org
......
...@@ -261,10 +261,10 @@ class CreateProfileTest(TestCase): ...@@ -261,10 +261,10 @@ class CreateProfileTest(TestCase):
with self.assertRaises(ValCannotCreateError): with self.assertRaises(ValCannotCreateError):
api.create_profile(constants.PROFILE_DESKTOP) api.create_profile(constants.PROFILE_DESKTOP)
@ddt
class GetVideoInfoTest(TestCase): class GetVideoInfoTest(TestCase):
""" """
Tests for our get_video_info function in api.py Tests for our `get_video_info` and `is_video_available` methods in api.py
""" """
def setUp(self): def setUp(self):
...@@ -330,6 +330,17 @@ class GetVideoInfoTest(TestCase): ...@@ -330,6 +330,17 @@ class GetVideoInfoTest(TestCase):
constants.VIDEO_DICT_FISH.get("edx_video_id") constants.VIDEO_DICT_FISH.get("edx_video_id")
) )
@data(
('non-existent-edx-video-id', False),
('super-soaker', True)
)
@unpack
def test_is_video_available(self, edx_video_id, expected_availability):
"""
Tests to see if a video exists for an `edx_video_id`.
"""
self.assertEqual(api.is_video_available(edx_video_id), expected_availability)
class GetUrlsForProfileTest(TestCase): class GetUrlsForProfileTest(TestCase):
""" """
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment