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))