quyuan 1 anno fa
parent
commit
f5939a51b6
1 ha cambiato i file con 4 aggiunte e 13 eliminazioni
  1. 4 13
      tests/test_cli/lib/common.py

+ 4 - 13
tests/test_cli/lib/common.py

@@ -4,20 +4,11 @@ def check_shell(cmd):
     res = os.system(cmd)
     res = os.system(cmd)
     assert res == 0
     assert res == 0
 
 
-def count_folders_and_check_contents(directory):
+def count_folders_and_check_contents(file_path):
     # 获取目录下的所有文件和文件夹
     # 获取目录下的所有文件和文件夹
-    contents = os.listdir(directory)
-    folder_count = 0
-    for item in contents:
-        # 检查是否为文件夹
-        if os.path.isdir(os.path.join(directory, item)):
-            # 检查文件夹是否为空
-            folder_path = os.path.join(directory, item)
-            for folder in os.listdir(folder_path):
-                folder_count = folder_count + 1
-                assert os.listdir(folder_path) is not None
-    print (folder_count)
-    assert folder_count > 0
+    if os.path.exists(file_path):
+        folder_count = os.path.getsize(file_path)
+        assert folder_count > 0
 
 
 
 
 if __name__ == "__main__":
 if __name__ == "__main__":