summaryrefslogtreecommitdiff
path: root/tests/sources/fixture.py
blob: c22667ad16de1c025de08242131a305963c580bc (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
import os

from pluginbase import PluginBase
from buildstream import Context, Project
from buildstream._loader import Loader
from buildstream._sourcefactory import SourceFactory


def message_handler(message, context):
    print("{}".format(message.message))


# Hacked hand fashioned fixture, just a helper function
# because its tricky to have a pytest fixture take arguments
#
# datafiles: A project directory datafiles
# target: A target element bst file
#
class Setup():

    def __init__(self, datafiles, target, tmpdir):
        directory = os.path.join(datafiles.dirname, datafiles.basename)

        self.context = Context([])
        self.project = Project(directory, self.context)

        # A message handler is required
        self.context._set_message_handler(message_handler)

        self.context.sourcedir = os.path.join(str(tmpdir), 'sources')
        self.context.builddir = os.path.join(str(tmpdir), 'build')

        if not os.path.exists(self.context.sourcedir):
            os.mkdir(self.context.sourcedir)
        if not os.path.exists(self.context.builddir):
            os.mkdir(self.context.builddir)

        loader = Loader(directory, target, self.project._options)
        element = loader.load()

        # Allow repo aliases to access files in the directories using tmpdir and datafiles
        self.project._aliases['tmpdir'] = "file:///" + str(tmpdir)
        self.project._aliases['datafiles'] = "file:///" + str(datafiles)

        assert(len(element.sources) >= 1)

        base = PluginBase(package='buildstream.plugins')
        self.factory = SourceFactory(base)

        self.sources = [self.factory.create(source.kind,
                                            self.context,
                                            self.project,
                                            source)
                        for source in element.sources]
        self.source = self.sources[0]