fable.py


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

import os
import sys
import tempfile


class Context:

    def __init__(self):
        self._vars = {}

    def get(self, key, default=None):
        return self._vars.get(key, default)

    def __getitem__(self, key):
        return self._vars[key]

    def __setitem__(self, key, value):
        self._vars[key] = value


class Run:

    def __init__(self):
        self._name = None
        self._context = None
        self._srcdir = None
        self._tmpdir = None

    def get_context(self):
        return self._context

    def start(self, name):
        self._context = Context()
        self._name = name
        self._tmpdir = tempfile.mkdtemp()
        self._srcdir = os.getcwd()
        os.chdir(self._tmpdir)

    def end(self):
        sys.stdout.write('OK: {}\n'.format(self._name))
        os.chdir(self._srcdir)


def assertEqual(a, b):
    if a != b:
        raise Exception(
            'expected {!r} == {!r}, but was disappointed'.format(a, b))