summaryrefslogtreecommitdiffstats
path: root/testing/mozharness/examples/venv.py
blob: ee16162ceae4f0a74bf3952d039ba54761d05dbb (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
#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

"""venv.py

Test virtualenv creation. This installs talos in the local venv; that's it.
"""

import os
import sys

sys.path.insert(1, os.path.dirname(sys.path[0]))

from mozharness.base.errors import PythonErrorList
from mozharness.base.python import virtualenv_config_options, VirtualenvMixin
from mozharness.base.script import BaseScript


# VirtualenvExample {{{1
class VirtualenvExample(VirtualenvMixin, BaseScript):
    config_options = [
        [
            ["--talos-url"],
            {
                "action": "store",
                "dest": "talos_url",
                "default": "https://hg.mozilla.org/build/talos/archive/tip.tar.gz",
                "help": "Specify the talos pip url",
            },
        ]
    ] + virtualenv_config_options

    def __init__(self, require_config_file=False):
        super(VirtualenvExample, self).__init__(
            config_options=self.config_options,
            all_actions=[
                "create-virtualenv",
            ],
            default_actions=[
                "create-virtualenv",
            ],
            require_config_file=require_config_file,
            config={"virtualenv_modules": ["talos"]},
        )


# __main__ {{{1
if __name__ == "__main__":
    venv_example = VirtualenvExample()
    venv_example.run_and_exit()