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)