def test_get_local_files(self):
# check no files to start with
count = File.query.count()
self.assertEqual(count, 0)
# now scan for files and check we get some
tasks.get_local_files()
count = File.query.count()
self.assertTrue(count > 0)
def test_api_all_files(self):
# check error response works
resp = self.app.get('/api/v1/files/')
data = json.loads(resp.data)
self.assertTrue(data.has_key('error'))
# now scan for files and ensure response contains some
tasks.get_local_files()
resp = self.app.get('/api/v1/files/')
data = json.loads(resp.data)
self.assertTrue(len(data) > 0)
def test_api_single_file(self):
tasks.get_local_files()
# check error response works
resp = self.app.get('/api/v1/files/9876543210')
data = json.loads(resp.data)
self.assertTrue(data.has_key('error'))
# now check for good response
resp = self.app.get('/api/v1/files/1')
data = json.loads(resp.data)
self.assertIn('name', data)
self.assertIn('dir_path', data)
