def test_list(self):
with deepstar_path():
video_model = VideoModel()
video_model.insert('test1', 'test2')
video_model.insert('test3', 'test4')
video_model.insert('test5', 'test6')
frame_set_model = FrameSetModel()
frame_set_model.insert(1)
frame_set_model.insert(2)
frame_set_model.insert(3)
args = ['main.py', 'list', 'frame_sets']
opts = {}
route_handler = FrameSetCommandLineRouteHandler()
try:
sys.stdout = StringIO()
route_handler.handle(args, opts)
actual = sys.stdout.getvalue().strip()
finally:
sys.stdout = sys.__stdout__
# stdout
expected = textwrap.dedent('''
3 results
id | fk_videos
-------------
1 | 1
2 | 2
3 | 3''').strip()
self.assertEqual(actual, expected)
def test_list(self):
with deepstar_path():
video_model = VideoModel()
video_model.insert('test1', 'test2')
video_model.insert('test3', 'test4')
video_model.insert('test5', 'test6')
args = ['main.py', 'list', 'videos']
opts = {}
route_handler = VideoCommandLineRouteHandler()
try:
sys.stdout = StringIO()
route_handler.handle(args, opts)
actual = sys.stdout.getvalue().strip()
finally:
sys.stdout = sys.__stdout__
# stdout
expected = textwrap.dedent('''
3 results
id | uri | filename | description
---------------------------------
1 | test1 | test2 | None
2 | test3 | test4 | None
3 | test5 | test6 | None''').strip()
self.assertEqual(actual, expected)
def test_delete(self):
with deepstar_path():
video_model = VideoModel()
video_model.insert('test1', 'test2')
result = video_model.select(1)
self.assertEqual(result, (1, 'test1', 'test2', None))
result = video_model.delete(1)
self.assertTrue(result)
result = video_model.select(1)
self.assertIsNone(result)
def test_list(self):
with deepstar_path():
video_model = VideoModel()
video_model.insert('test1', 'test2')
video_model.insert('test3', 'test4')
video_model.insert('test5', 'test6')
result = video_model.list()
self.assertEqual(len(result), 3)
self.assertEqual(result[0], (1, 'test1', 'test2', None))
self.assertEqual(result[1], (2, 'test3', 'test4', None))
self.assertEqual(result[2], (3, 'test5', 'test6', None))
def test_update(self):
with deepstar_path():
video_model = VideoModel()
video_model.insert('test1', 'test2')
result = video_model.select(1)
self.assertEqual(result, (1, 'test1', 'test2', None))
video_model.update(1, uri='test3')
result = video_model.select(1)
self.assertEqual(result, (1, 'test3', 'test2', None))
def test_insert(self):
with deepstar_path():
video_model = VideoModel()
video_id = video_model.insert('test1', 'test2')
self.assertEqual(video_id, 1)
result = video_model.select(1)
self.assertEqual(result, (1, 'test1', 'test2', None))
