def test_directory_contents(self): expected_dict = { 'path': TEST_DIR, 'children': [ {'is_dir': False, 'filename': 'file1.txt'}, {'is_dir': False, 'filename': 'file2.txt'}, {'is_dir': False, 'filename': 'file3.txt'}, {'is_dir': True, 'filename': 'folder 1'}, {'is_dir': True, 'filename': 'folder 2'}, {'is_dir': True, 'filename': 'folder 3'} ] } dir_contents = get_directory_contents(TEST_DIR) self.assertEqual(expected_dict, dir_contents)
def directory_contents(): path = request.args.get('path', "") return jsonify({'files': get_directory_contents(path)})
def reset_test_files(): rebuild_test_tree() return jsonify({'files': get_directory_contents(TEST_DIR)})