summaryrefslogtreecommitdiff
path: root/testenv/conf/local_files.py
blob: 5f9c8fad761873e28808c6baa6edcf872ba29e2a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from conf import hook

""" Pre-Test Hook: LocalFiles
This is a pre-test hook used to generate the specific environment before a test
is run. The LocalFiles hook creates the files which should exist on disk before
invoking Wget.
"""


@hook()
class LocalFiles:
    def __init__(self, local_files):
        self.local_files = local_files

    def __call__(self, _):
        for f in self.local_files:
            with open(f.name, 'w') as fp:
                fp.write(f.content)