xref: /openbmc/openbmc-build-scripts/scripts/unit-test.py (revision 780ec095f037b7dcbd84c37770f35adf4de704b5)
1#!/usr/bin/env python
2
3"""
4This script determines the given package's openbmc dependencies from its
5configure.ac file where it downloads, configures, builds, and installs each of
6these dependencies. Then the given package is configured, built, and installed
7prior to executing its unit tests.
8"""
9
10from git import Repo
11from urlparse import urljoin
12from subprocess import check_call, call, CalledProcessError
13import os
14import sys
15import argparse
16import multiprocessing
17import re
18import platform
19
20
21class DepTree():
22    """
23    Represents package dependency tree, where each node is a DepTree with a
24    name and DepTree children.
25    """
26
27    def __init__(self, name):
28        """
29        Create new DepTree.
30
31        Parameter descriptions:
32        name               Name of new tree node.
33        """
34        self.name = name
35        self.children = list()
36
37    def AddChild(self, name):
38        """
39        Add new child node to current node.
40
41        Parameter descriptions:
42        name               Name of new child
43        """
44        new_child = DepTree(name)
45        self.children.append(new_child)
46        return new_child
47
48    def AddChildNode(self, node):
49        """
50        Add existing child node to current node.
51
52        Parameter descriptions:
53        node               Tree node to add
54        """
55        self.children.append(node)
56
57    def RemoveChild(self, name):
58        """
59        Remove child node.
60
61        Parameter descriptions:
62        name               Name of child to remove
63        """
64        for child in self.children:
65            if child.name == name:
66                self.children.remove(child)
67                return
68
69    def GetNode(self, name):
70        """
71        Return node with matching name. Return None if not found.
72
73        Parameter descriptions:
74        name               Name of node to return
75        """
76        if self.name == name:
77            return self
78        for child in self.children:
79            node = child.GetNode(name)
80            if node:
81                return node
82        return None
83
84    def GetParentNode(self, name, parent_node=None):
85        """
86        Return parent of node with matching name. Return none if not found.
87
88        Parameter descriptions:
89        name               Name of node to get parent of
90        parent_node        Parent of current node
91        """
92        if self.name == name:
93            return parent_node
94        for child in self.children:
95            found_node = child.GetParentNode(name, self)
96            if found_node:
97                return found_node
98        return None
99
100    def GetPath(self, name, path=None):
101        """
102        Return list of node names from head to matching name.
103        Return None if not found.
104
105        Parameter descriptions:
106        name               Name of node
107        path               List of node names from head to current node
108        """
109        if not path:
110            path = []
111        if self.name == name:
112            path.append(self.name)
113            return path
114        for child in self.children:
115            match = child.GetPath(name, path + [self.name])
116            if match:
117                return match
118        return None
119
120    def GetPathRegex(self, name, regex_str, path=None):
121        """
122        Return list of node paths that end in name, or match regex_str.
123        Return empty list if not found.
124
125        Parameter descriptions:
126        name               Name of node to search for
127        regex_str          Regex string to match node names
128        path               Path of node names from head to current node
129        """
130        new_paths = []
131        if not path:
132            path = []
133        match = re.match(regex_str, self.name)
134        if (self.name == name) or (match):
135            new_paths.append(path + [self.name])
136        for child in self.children:
137            return_paths = None
138            full_path = path + [self.name]
139            return_paths = child.GetPathRegex(name, regex_str, full_path)
140            for i in return_paths:
141                new_paths.append(i)
142        return new_paths
143
144    def MoveNode(self, from_name, to_name):
145        """
146        Mode existing from_name node to become child of to_name node.
147
148        Parameter descriptions:
149        from_name          Name of node to make a child of to_name
150        to_name            Name of node to make parent of from_name
151        """
152        parent_from_node = self.GetParentNode(from_name)
153        from_node = self.GetNode(from_name)
154        parent_from_node.RemoveChild(from_name)
155        to_node = self.GetNode(to_name)
156        to_node.AddChildNode(from_node)
157
158    def ReorderDeps(self, name, regex_str):
159        """
160        Reorder dependency tree.  If tree contains nodes with names that
161        match 'name' and 'regex_str', move 'regex_str' nodes that are
162        to the right of 'name' node, so that they become children of the
163        'name' node.
164
165        Parameter descriptions:
166        name               Name of node to look for
167        regex_str          Regex string to match names to
168        """
169        name_path = self.GetPath(name)
170        if not name_path:
171            return
172        paths = self.GetPathRegex(name, regex_str)
173        is_name_in_paths = False
174        name_index = 0
175        for i in range(len(paths)):
176            path = paths[i]
177            if path[-1] == name:
178                is_name_in_paths = True
179                name_index = i
180                break
181        if not is_name_in_paths:
182            return
183        for i in range(name_index + 1, len(paths)):
184            path = paths[i]
185            if name in path:
186                continue
187            from_name = path[-1]
188            self.MoveNode(from_name, name)
189
190    def GetInstallList(self):
191        """
192        Return post-order list of node names.
193
194        Parameter descriptions:
195        """
196        install_list = []
197        for child in self.children:
198            child_install_list = child.GetInstallList()
199            install_list.extend(child_install_list)
200        install_list.append(self.name)
201        return install_list
202
203    def PrintTree(self, level=0):
204        """
205        Print pre-order node names with indentation denoting node depth level.
206
207        Parameter descriptions:
208        level              Current depth level
209        """
210        INDENT_PER_LEVEL = 4
211        print ' ' * (level * INDENT_PER_LEVEL) + self.name
212        for child in self.children:
213            child.PrintTree(level + 1)
214
215
216def check_call_cmd(dir, *cmd):
217    """
218    Verbose prints the directory location the given command is called from and
219    the command, then executes the command using check_call.
220
221    Parameter descriptions:
222    dir                 Directory location command is to be called from
223    cmd                 List of parameters constructing the complete command
224    """
225    printline(dir, ">", " ".join(cmd))
226    check_call(cmd)
227
228
229def clone_pkg(pkg):
230    """
231    Clone the given openbmc package's git repository from gerrit into
232    the WORKSPACE location
233
234    Parameter descriptions:
235    pkg                 Name of the package to clone
236    """
237    pkg_dir = os.path.join(WORKSPACE, pkg)
238    if os.path.exists(os.path.join(pkg_dir, '.git')):
239        return pkg_dir
240    pkg_repo = urljoin('https://gerrit.openbmc-project.xyz/openbmc/', pkg)
241    os.mkdir(pkg_dir)
242    printline(pkg_dir, "> git clone", pkg_repo, "./")
243    return Repo.clone_from(pkg_repo, pkg_dir).working_dir
244
245
246def get_deps(configure_ac):
247    """
248    Parse the given 'configure.ac' file for package dependencies and return
249    a list of the dependencies found.
250
251    Parameter descriptions:
252    configure_ac        Opened 'configure.ac' file object
253    """
254    line = ""
255    dep_pkgs = set()
256    for cfg_line in configure_ac:
257        # Remove whitespace & newline
258        cfg_line = cfg_line.rstrip()
259        # Check for line breaks
260        if cfg_line.endswith('\\'):
261            line += str(cfg_line[:-1])
262            continue
263        line = line+cfg_line
264
265        # Find any defined dependency
266        line_has = lambda x: x if x in line else None
267        macros = set(filter(line_has, DEPENDENCIES.iterkeys()))
268        if len(macros) == 1:
269            macro = ''.join(macros)
270            deps = filter(line_has, DEPENDENCIES[macro].iterkeys())
271            dep_pkgs.update(map(lambda x: DEPENDENCIES[macro][x], deps))
272
273        line = ""
274    deps = list(dep_pkgs)
275
276    return deps
277
278
279make_parallel = [
280    'make',
281    # Run enough jobs to saturate all the cpus
282    '-j', str(multiprocessing.cpu_count()),
283    # Don't start more jobs if the load avg is too high
284    '-l', str(multiprocessing.cpu_count()),
285    # Synchronize the output so logs aren't intermixed in stdout / stderr
286    '-O',
287]
288
289def build_and_install(pkg):
290    """
291    Builds and installs the package in the environment. Optionally
292    builds the examples and test cases for package.
293
294    Parameter description:
295    pkg                 The package we are building
296    """
297    pkgdir = os.path.join(WORKSPACE, pkg)
298    # Build & install this package
299    conf_flags = [
300        '--disable-silent-rules',
301        '--enable-tests',
302        '--enable-code-coverage',
303        '--enable-valgrind'
304    ]
305    os.chdir(pkgdir)
306    # Add any necessary configure flags for package
307    if CONFIGURE_FLAGS.get(pkg) is not None:
308        conf_flags.extend(CONFIGURE_FLAGS.get(pkg))
309    check_call_cmd(pkgdir, './bootstrap.sh')
310    check_call_cmd(pkgdir, './configure', *conf_flags)
311    check_call_cmd(pkgdir, *make_parallel)
312    check_call_cmd(pkgdir, *(make_parallel + [ 'install' ]))
313
314def install_deps(dep_list):
315    """
316    Install each package in the ordered dep_list.
317
318    Parameter descriptions:
319    dep_list            Ordered list of dependencies
320    """
321    for pkg in dep_list:
322        build_and_install(pkg)
323
324def build_dep_tree(pkg, pkgdir, dep_added, head, dep_tree=None):
325    """
326    For each package(pkg), starting with the package to be unit tested,
327    parse its 'configure.ac' file from within the package's directory(pkgdir)
328    for each package dependency defined recursively doing the same thing
329    on each package found as a dependency.
330
331    Parameter descriptions:
332    pkg                 Name of the package
333    pkgdir              Directory where package source is located
334    dep_added           Current list of dependencies and added status
335    head                Head node of the dependency tree
336    dep_tree            Current dependency tree node
337    """
338    if not dep_tree:
339        dep_tree = head
340    os.chdir(pkgdir)
341    # Open package's configure.ac
342    with open("/root/.depcache", "r") as depcache:
343        cached = depcache.readline()
344    with open("configure.ac", "rt") as configure_ac:
345        # Retrieve dependency list from package's configure.ac
346        configure_ac_deps = get_deps(configure_ac)
347        for dep_pkg in configure_ac_deps:
348            if dep_pkg in cached:
349                continue
350            # Dependency package not already known
351            if dep_added.get(dep_pkg) is None:
352                # Dependency package not added
353                new_child = dep_tree.AddChild(dep_pkg)
354                dep_added[dep_pkg] = False
355                dep_pkgdir = clone_pkg(dep_pkg)
356                # Determine this dependency package's
357                # dependencies and add them before
358                # returning to add this package
359                dep_added = build_dep_tree(dep_pkg,
360                                           dep_pkgdir,
361                                           dep_added,
362                                           head,
363                                           new_child)
364            else:
365                # Dependency package known and added
366                if dep_added[dep_pkg]:
367                    continue
368                else:
369                    # Cyclic dependency failure
370                    raise Exception("Cyclic dependencies found in "+pkg)
371
372    if not dep_added[pkg]:
373        dep_added[pkg] = True
374
375    return dep_added
376
377def make_target_exists(target):
378    """
379    Runs a check against the makefile in the current directory to determine
380    if the target exists so that it can be built.
381
382    Parameter descriptions:
383    target              The make target we are checking
384    """
385    try:
386        cmd = [ 'make', '-n', target ]
387        with open(os.devnull, 'w') as devnull:
388            check_call(cmd, stdout=devnull, stderr=devnull)
389        return True
390    except CalledProcessError:
391        return False
392
393def run_unit_tests(top_dir):
394    """
395    Runs the unit tests for the package via `make check`
396
397    Parameter descriptions:
398    top_dir             The root directory of our project
399    """
400    try:
401        cmd = make_parallel + [ 'check' ]
402        for i in range(0, args.repeat):
403            check_call_cmd(top_dir,  *cmd)
404    except CalledProcessError:
405        for root, _, files in os.walk(top_dir):
406            if 'test-suite.log' not in files:
407                continue
408            check_call_cmd(root, 'cat', os.path.join(root, 'test-suite.log'))
409        raise Exception('Unit tests failed')
410
411def run_cppcheck(top_dir):
412    try:
413        # http://cppcheck.sourceforge.net/manual.pdf
414        ignore_list = ['-i%s' % path for path in os.listdir(top_dir) \
415                       if path.endswith('-src') or path.endswith('-build')]
416        ignore_list.extend(('-itest', '-iscripts'))
417        params = ['cppcheck', '-j', str(multiprocessing.cpu_count()),
418                  '--enable=all']
419        params.extend(ignore_list)
420        params.append('.')
421
422        check_call_cmd(top_dir, *params)
423    except CalledProcessError:
424        raise Exception('Cppcheck failed')
425
426def maybe_run_valgrind(top_dir):
427    """
428    Potentially runs the unit tests through valgrind for the package
429    via `make check-valgrind`. If the package does not have valgrind testing
430    then it just skips over this.
431
432    Parameter descriptions:
433    top_dir             The root directory of our project
434    """
435    # Valgrind testing is currently broken by an aggressive strcmp optimization
436    # that is inlined into optimized code for POWER by gcc 7+. Until we find
437    # a workaround, just don't run valgrind tests on POWER.
438    # https://github.com/openbmc/openbmc/issues/3315
439    if re.match('ppc64', platform.machine()) is not None:
440        return
441    if not make_target_exists('check-valgrind'):
442        return
443
444    try:
445        cmd = make_parallel + [ 'check-valgrind' ]
446        check_call_cmd(top_dir,  *cmd)
447    except CalledProcessError:
448        for root, _, files in os.walk(top_dir):
449            for f in files:
450                if re.search('test-suite-[a-z]+.log', f) is None:
451                    continue
452                check_call_cmd(root, 'cat', os.path.join(root, f))
453        raise Exception('Valgrind tests failed')
454
455def maybe_run_coverage(top_dir):
456    """
457    Potentially runs the unit tests through code coverage for the package
458    via `make check-code-coverage`. If the package does not have code coverage
459    testing then it just skips over this.
460
461    Parameter descriptions:
462    top_dir             The root directory of our project
463    """
464    if not make_target_exists('check-code-coverage'):
465        return
466
467    # Actually run code coverage
468    try:
469        cmd = make_parallel + [ 'check-code-coverage' ]
470        check_call_cmd(top_dir,  *cmd)
471    except CalledProcessError:
472        raise Exception('Code coverage failed')
473
474if __name__ == '__main__':
475    # CONFIGURE_FLAGS = [GIT REPO]:[CONFIGURE FLAGS]
476    CONFIGURE_FLAGS = {
477        'phosphor-objmgr': ['--enable-unpatched-systemd'],
478        'sdbusplus': ['--enable-transaction'],
479        'phosphor-logging':
480        ['--enable-metadata-processing',
481         'YAML_DIR=/usr/local/share/phosphor-dbus-yaml/yaml']
482    }
483
484    # DEPENDENCIES = [MACRO]:[library/header]:[GIT REPO]
485    DEPENDENCIES = {
486        'AC_CHECK_LIB': {'mapper': 'phosphor-objmgr'},
487        'AC_CHECK_HEADER': {
488            'host-ipmid': 'phosphor-host-ipmid',
489            'blobs-ipmid': 'phosphor-ipmi-blobs',
490            'sdbusplus': 'sdbusplus',
491            'sdeventplus': 'sdeventplus',
492            'gpioplus': 'gpioplus',
493            'phosphor-logging/log.hpp': 'phosphor-logging',
494        },
495        'AC_PATH_PROG': {'sdbus++': 'sdbusplus'},
496        'PKG_CHECK_MODULES': {
497            'phosphor-dbus-interfaces': 'phosphor-dbus-interfaces',
498            'openpower-dbus-interfaces': 'openpower-dbus-interfaces',
499            'ibm-dbus-interfaces': 'ibm-dbus-interfaces',
500            'sdbusplus': 'sdbusplus',
501            'sdeventplus': 'sdeventplus',
502            'gpioplus': 'gpioplus',
503            'phosphor-logging': 'phosphor-logging',
504            'phosphor-snmp': 'phosphor-snmp',
505        },
506    }
507
508    # DEPENDENCIES_REGEX = [GIT REPO]:[REGEX STRING]
509    DEPENDENCIES_REGEX = {
510        'phosphor-logging': r'\S+-dbus-interfaces$'
511    }
512
513    # Set command line arguments
514    parser = argparse.ArgumentParser()
515    parser.add_argument("-w", "--workspace", dest="WORKSPACE", required=True,
516                        help="Workspace directory location(i.e. /home)")
517    parser.add_argument("-p", "--package", dest="PACKAGE", required=True,
518                        help="OpenBMC package to be unit tested")
519    parser.add_argument("-v", "--verbose", action="store_true",
520                        help="Print additional package status messages")
521    parser.add_argument("-r", "--repeat", help="Repeat tests N times",
522                        type=int, default=1)
523    args = parser.parse_args(sys.argv[1:])
524    WORKSPACE = args.WORKSPACE
525    UNIT_TEST_PKG = args.PACKAGE
526    if args.verbose:
527        def printline(*line):
528            for arg in line:
529                print arg,
530            print
531    else:
532        printline = lambda *l: None
533
534    # First validate code formatting if repo has style formatting files.
535    # The format-code.sh checks for these files.
536    CODE_SCAN_DIR = WORKSPACE + "/" + UNIT_TEST_PKG
537    check_call_cmd(WORKSPACE, "./format-code.sh", CODE_SCAN_DIR)
538
539    # Automake
540    if os.path.isfile(CODE_SCAN_DIR + "/configure.ac"):
541        prev_umask = os.umask(000)
542        # Determine dependencies and add them
543        dep_added = dict()
544        dep_added[UNIT_TEST_PKG] = False
545        # Create dependency tree
546        dep_tree = DepTree(UNIT_TEST_PKG)
547        build_dep_tree(UNIT_TEST_PKG,
548                       os.path.join(WORKSPACE, UNIT_TEST_PKG),
549                       dep_added,
550                       dep_tree)
551
552        # Reorder Dependency Tree
553        for pkg_name, regex_str in DEPENDENCIES_REGEX.iteritems():
554            dep_tree.ReorderDeps(pkg_name, regex_str)
555        if args.verbose:
556            dep_tree.PrintTree()
557        install_list = dep_tree.GetInstallList()
558        # install reordered dependencies
559        install_deps(install_list)
560        top_dir = os.path.join(WORKSPACE, UNIT_TEST_PKG)
561        os.chdir(top_dir)
562        # Refresh dynamic linker run time bindings for dependencies
563        check_call_cmd(top_dir, 'ldconfig')
564        # Run package unit tests
565        run_unit_tests(top_dir)
566        maybe_run_valgrind(top_dir)
567        maybe_run_coverage(top_dir)
568        run_cppcheck(top_dir)
569
570        os.umask(prev_umask)
571
572    # Cmake
573    elif os.path.isfile(CODE_SCAN_DIR + "/CMakeLists.txt"):
574        top_dir = os.path.join(WORKSPACE, UNIT_TEST_PKG)
575        os.chdir(top_dir)
576        check_call_cmd(top_dir, 'cmake', '.')
577        check_call_cmd(top_dir, 'cmake', '--build', '.', '--', '-j',
578                       str(multiprocessing.cpu_count()))
579        if make_target_exists('test'):
580            check_call_cmd(top_dir, 'ctest', '.')
581        maybe_run_valgrind(top_dir)
582        maybe_run_coverage(top_dir)
583        run_cppcheck(top_dir)
584
585    else:
586        print "Not a supported repo for CI Tests, exit"
587        quit()
588