100600bfbSopenharmony_ci#!/usr/bin/env python3 200600bfbSopenharmony_ci# -*- coding: utf-8 -*- 300600bfbSopenharmony_ci# Copyright (C) 2024 Huawei Device Co., Ltd. 400600bfbSopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License"); 500600bfbSopenharmony_ci# you may not use this file except in compliance with the License. 600600bfbSopenharmony_ci# You may obtain a copy of the License at 700600bfbSopenharmony_ci# 800600bfbSopenharmony_ci# http://www.apache.org/licenses/LICENSE-2.0 900600bfbSopenharmony_ci# 1000600bfbSopenharmony_ci# Unless required by applicable law or agreed to in writing, software 1100600bfbSopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS, 1200600bfbSopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1300600bfbSopenharmony_ci# See the License for the specific language governing permissions and 1400600bfbSopenharmony_ci# limitations under the License. 1500600bfbSopenharmony_ci 1600600bfbSopenharmony_ciimport pytest 1700600bfbSopenharmony_ciimport re 1800600bfbSopenharmony_cifrom utils import * 1900600bfbSopenharmony_ci 2000600bfbSopenharmony_ci@print_check_result 2100600bfbSopenharmony_cidef check_df_k(output): 2200600bfbSopenharmony_ci result = re.search("cmd is: df -k\n\n([^\n]+\n){4,}", output) 2300600bfbSopenharmony_ci return result is not None 2400600bfbSopenharmony_ci 2500600bfbSopenharmony_ci@print_check_result 2600600bfbSopenharmony_cidef check_lsof(output): 2700600bfbSopenharmony_ci result = re.search("cmd is: lsof\n\n([^\n]+\n){4,}", output) 2800600bfbSopenharmony_ci return result is not None 2900600bfbSopenharmony_ci 3000600bfbSopenharmony_ci@print_check_result 3100600bfbSopenharmony_cidef check_iotop(output): 3200600bfbSopenharmony_ci result = re.search("cmd is: iotop -n 1 -m 100\n([^\n]+\n){4,}", output) 3300600bfbSopenharmony_ci return result is not None 3400600bfbSopenharmony_ci 3500600bfbSopenharmony_ci@print_check_result 3600600bfbSopenharmony_cidef check_proc_mount(output): 3700600bfbSopenharmony_ci result = re.search("/proc/mounts\n\n([^\n]+\n){4,}", output) 3800600bfbSopenharmony_ci return result is not None 3900600bfbSopenharmony_ci 4000600bfbSopenharmony_cidef CheckStorageWithoutPid(output): 4100600bfbSopenharmony_ci ret = all(check(output) for check in [check_df_k, check_lsof, check_iotop, check_proc_mount]) 4200600bfbSopenharmony_ci return ret 4300600bfbSopenharmony_ci 4400600bfbSopenharmony_cidef CheckStorageWithPid(output): 4500600bfbSopenharmony_ci result = re.search("storage io", output) 4600600bfbSopenharmony_ci return result is not None 4700600bfbSopenharmony_ci 4800600bfbSopenharmony_ciclass TestHidumperStorage: 4900600bfbSopenharmony_ci @pytest.mark.L0 5000600bfbSopenharmony_ci def test_storage_all(self): 5100600bfbSopenharmony_ci command = "hidumper --storage" 5200600bfbSopenharmony_ci # 校验命令行输出 5300600bfbSopenharmony_ci CheckCmd(command, CheckStorageWithoutPid) 5400600bfbSopenharmony_ci # 校验命令行重定向输出 5500600bfbSopenharmony_ci CheckCmdRedirect(command, CheckStorageWithoutPid) 5600600bfbSopenharmony_ci # 校验命令行输出到zip文件 5700600bfbSopenharmony_ci CheckCmdZip(command, CheckStorageWithoutPid) 5800600bfbSopenharmony_ci 5900600bfbSopenharmony_ci @pytest.mark.L0 6000600bfbSopenharmony_ci def test_storage_pid(self): 6100600bfbSopenharmony_ci command = f"hidumper --storage 1" 6200600bfbSopenharmony_ci # 校验命令行输出 6300600bfbSopenharmony_ci CheckCmd(command, CheckStorageWithPid) 6400600bfbSopenharmony_ci # 校验命令行重定向输出 6500600bfbSopenharmony_ci CheckCmdRedirect(command, CheckStorageWithPid) 6600600bfbSopenharmony_ci # 校验命令行输出到zip文件 6700600bfbSopenharmony_ci CheckCmdZip(command, CheckStorageWithPid)