# Copyright 2008-2015 Nokia Networks
# Copyright 2016- Robot Framework Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import Iterator
from .stats import SuiteStat
[docs]
class SuiteStatistics:
"""Container for suite statistics."""
def __init__(self, suite):
self.stat = SuiteStat(suite)
self.suites: list[SuiteStatistics] = []
[docs]
def visit(self, visitor):
visitor.visit_suite_statistics(self)
def __iter__(self) -> Iterator[SuiteStat]:
yield self.stat
for child in self.suites:
yield from child
[docs]
class SuiteStatisticsBuilder:
def __init__(self, suite_stat_level):
self._suite_stat_level = suite_stat_level
self._stats_stack: list[SuiteStatistics] = []
self.stats: SuiteStatistics | None = None
@property
def current(self) -> "SuiteStatistics|None":
return self._stats_stack[-1] if self._stats_stack else None
[docs]
def start_suite(self, suite):
self._stats_stack.append(SuiteStatistics(suite))
if self.stats is None:
self.stats = self.current
[docs]
def add_test(self, test):
self.current.stat.add_test(test)
[docs]
def end_suite(self):
stats = self._stats_stack.pop()
if self.current:
self.current.stat.add_stat(stats.stat)
if self._is_child_included():
self.current.suites.append(stats)
def _is_child_included(self):
return self._include_all_levels() or self._below_threshold()
def _include_all_levels(self):
return self._suite_stat_level == -1
def _below_threshold(self):
return len(self._stats_stack) < self._suite_stat_level