summaryrefslogtreecommitdiffstats
path: root/meta/lib/oeqa/core/utils/test.py
blob: d38cab8a511aa35fa9e6cfe35987c61aa06e3d85 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#
# Copyright (C) 2016 Intel Corporation
#
# SPDX-License-Identifier: MIT
#

import os
import inspect
import unittest

def getSuiteCases(suite):
    """
        Returns individual test from a test suite.
    """
    tests = []

    if isinstance(suite, unittest.TestCase):
        tests.append(suite)
    elif isinstance(suite, unittest.suite.TestSuite):
        for item in suite:
            tests.extend(getSuiteCases(item))

    return tests

def getSuiteModules(suite):
    """
        Returns modules in a test suite.
    """
    modules = set()
    for test in getSuiteCases(suite):
        modules.add(getCaseModule(test))
    return modules

def getSuiteCasesInfo(suite, func):
    """
        Returns test case info from suite. Info is fetched from func.
    """
    tests = []
    for test in getSuiteCases(suite):
        tests.append(func(test))
    return tests

def getSuiteCasesNames(suite):
    """
        Returns test case names from suite.
    """
    return getSuiteCasesInfo(suite, getCaseMethod)

def getSuiteCasesIDs(suite):
    """
        Returns test case ids from suite.
    """
    return getSuiteCasesInfo(suite, getCaseID)

def getSuiteCasesFiles(suite):
    """
        Returns test case files paths from suite.
    """
    return getSuiteCasesInfo(suite, getCaseFile)

def getCaseModule(test_case):
    """
        Returns test case module name.
    """
    return test_case.__module__

def getCaseClass(test_case):
    """
        Returns test case class name.
    """
    return test_case.__class__.__name__

def getCaseID(test_case):
    """
        Returns test case complete id.
    """
    return test_case.id()

def getCaseFile(test_case):
    """
        Returns test case file path.
    """
    return inspect.getsourcefile(test_case.__class__)

def getCaseMethod(test_case):
    """
        Returns test case method name.
    """
    return getCaseID(test_case).split('.')[-1]