1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3 4# Copyright (c) 2021-2024 Huawei Device Co., Ltd. 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16# 17 18from functools import cached_property 19from typing import Dict, List 20 21from runner.options.decorator_value import value, _to_str, _to_int 22 23 24class GroupsOptions: 25 __DEFAULT_GROUPS = 1 26 __DEFAULT_GROUP_NUMBER = 1 27 28 def __str__(self) -> str: 29 return _to_str(self, 2) 30 31 def to_dict(self) -> Dict[str, object]: 32 return { 33 "quantity": self.quantity, 34 "number": self.number, 35 "chapters": self.chapters, 36 "chapters-file": self.chapters_file 37 } 38 39 @cached_property 40 @value(yaml_path="test-lists.groups.quantity", cli_name="groups", cast_to_type=_to_int) 41 def quantity(self) -> int: 42 return GroupsOptions.__DEFAULT_GROUPS 43 44 @cached_property 45 @value(yaml_path="test-lists.groups.number", cli_name="group_number", cast_to_type=_to_int) 46 def number(self) -> int: 47 return GroupsOptions.__DEFAULT_GROUP_NUMBER 48 49 def get_command_line(self) -> str: 50 options = [ 51 f'--group={self.quantity}' if self.quantity != GroupsOptions.__DEFAULT_GROUPS else '', 52 f'--group-number={self.number}' if self.number != GroupsOptions.__DEFAULT_GROUP_NUMBER else '' 53 ] 54 return ' '.join(options) 55 56 @cached_property 57 @value(yaml_path="test-lists.groups.chapters", cli_name="chapters") 58 def chapters(self) -> List[str]: 59 return [] 60 61 @cached_property 62 @value(yaml_path="test-lists.groups.chapters-file", cli_name="chapters_file") 63 def chapters_file(self) -> str: 64 return "chapters.yaml" 65