python/mozbuild/mozbuild/test/test_telemetry.py

   1 # Any copyright is dedicated to the Public Domain.
   2 # http://creativecommons.org/publicdomain/zero/1.0/
   3
   4 import os
   5
   6 import buildconfig
   7 import mozunit
   8
   9 from mozbuild.telemetry import filter_args
  10
  11 TELEMETRY_LOAD_ERROR = """
  12 Error loading telemetry. mach output:
  13 =========================================================
  14 %s
  15 =========================================================
  16 """
  17
  18
  19 def test_path_filtering():
  20     srcdir_path = os.path.join(buildconfig.topsrcdir, "a")
  21     srcdir_path_2 = os.path.join(buildconfig.topsrcdir, "a/b/c")
  22     objdir_path = os.path.join(buildconfig.topobjdir, "x")
  23     objdir_path_2 = os.path.join(buildconfig.topobjdir, "x/y/z")
  24     home_path = os.path.join(os.path.expanduser("~"), "something_in_home")
  25     other_path = "/other/path"
  26     args = filter_args(
  27         "pass",
  28         [
  29             "python",
  30             "-c",
  31             "pass",
  32             srcdir_path,
  33             srcdir_path_2,
  34             objdir_path,
  35             objdir_path_2,
  36             home_path,
  37             other_path,
  38         ],
  39         buildconfig.topsrcdir,
  40         buildconfig.topobjdir,
  41         cwd=buildconfig.topsrcdir,
  42     )
  43
  44     expected = [
  45         "a",
  46         "a/b/c",
  47         "$topobjdir/x",
  48         "$topobjdir/x/y/z",
  49         "$HOME/something_in_home",
  50         "<path omitted>",
  51     ]
  52     assert args == expected
  53
  54
  55 def test_path_filtering_in_objdir():
  56     srcdir_path = os.path.join(buildconfig.topsrcdir, "a")
  57     srcdir_path_2 = os.path.join(buildconfig.topsrcdir, "a/b/c")
  58     objdir_path = os.path.join(buildconfig.topobjdir, "x")
  59     objdir_path_2 = os.path.join(buildconfig.topobjdir, "x/y/z")
  60     other_path = "/other/path"
  61     args = filter_args(
  62         "pass",
  63         [
  64             "python",
  65             "-c",
  66             "pass",
  67             srcdir_path,
  68             srcdir_path_2,
  69             objdir_path,
  70             objdir_path_2,
  71             other_path,
  72         ],
  73         buildconfig.topsrcdir,
  74         buildconfig.topobjdir,
  75         cwd=buildconfig.topobjdir,
  76     )
  77     expected = ["$topsrcdir/a", "$topsrcdir/a/b/c", "x", "x/y/z", "<path omitted>"]
  78     assert args == expected
  79
  80
  81 def test_path_filtering_other_cwd(tmpdir):
  82     srcdir_path = os.path.join(buildconfig.topsrcdir, "a")
  83     srcdir_path_2 = os.path.join(buildconfig.topsrcdir, "a/b/c")
  84     other_path = str(tmpdir.join("other"))
  85     args = filter_args(
  86         "pass",
  87         ["python", "-c", "pass", srcdir_path, srcdir_path_2, other_path],
  88         buildconfig.topsrcdir,
  89         buildconfig.topobjdir,
  90         cwd=str(tmpdir),
  91     )
  92     expected = [
  93         "$topsrcdir/a",
  94         "$topsrcdir/a/b/c",
  95         # cwd-relative paths should be relativized
  96         "other",
  97     ]
  98     assert args == expected
  99
 100
 101 if __name__ == "__main__":
 102     mozunit.main()