appenv/tests/test_init.py at 018038f020866c5c7fd5cfa1eb3d55080010cdc7 · flyingcircusio/appenv · GitHub

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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import io
import os

import appenv


def test_init(workdir, monkeypatch):
    monkeypatch.setattr("sys.stdin", io.StringIO("ducker\n\n\n"))

    assert not os.path.exists(os.path.join(workdir, "ducker"))

    env = appenv.AppEnv(os.path.join(workdir, 'ducker'), os.getcwd())
    env.init()

    assert os.readlink(os.path.join(workdir, "ducker", "ducker")) == 'appenv'

    with open(os.path.join(workdir, "ducker", "appenv")) as f:
        ducker_appenv = f.read()
    with open(appenv.__file__) as f:
        original_appenv = f.read()
    assert ducker_appenv == original_appenv

    with open(os.path.join(workdir, "ducker", "requirements.in")) as f:
        requirements = f.read()

    assert requirements == "ducker\n"

    # calling it again doesn't break:
    monkeypatch.setattr("sys.stdin", io.StringIO("ducker\n\n\n"))
    env.init()

    assert os.readlink(os.path.join(workdir, "ducker", "ducker")) == 'appenv'

    with open(os.path.join(workdir, "ducker", "appenv")) as f:
        ducker_appenv = f.read()
    with open(appenv.__file__) as f:
        original_appenv = f.read()
    assert ducker_appenv == original_appenv

    with open(os.path.join(workdir, "ducker", "requirements.in")) as f:
        requirements = f.read()

    assert requirements == "ducker\n"


def test_init_explicit_target(workdir, monkeypatch):
    monkeypatch.setattr("sys.stdin", io.StringIO("ducker\n\nbaz\n"))

    env = appenv.AppEnv(os.path.join(workdir, 'ducker'), os.getcwd())
    env.init()

    assert os.path.exists(os.path.join(workdir, "baz"))
    with open(os.path.join(workdir, "baz", "ducker")) as f:
        ducker_appenv = f.read()
    with open(appenv.__file__) as f:
        original_appenv = f.read()

    with open(os.path.join(workdir, "baz", "requirements.in")) as f:
        requirements = f.read()

    assert requirements == "ducker\n"
    assert ducker_appenv == original_appenv


def test_init_explicit_package_and_target(workdir, monkeypatch):
    monkeypatch.setattr("sys.stdin", io.StringIO("foo\nbar\nbaz\n"))

    env = appenv.AppEnv(os.path.join(workdir, 'ducker'), os.getcwd())
    env.init()

    assert os.path.exists(os.path.join(workdir, "baz"))
    with open(os.path.join(workdir, "baz", "foo")) as f:
        ducker_appenv = f.read()
    with open(appenv.__file__) as f:
        original_appenv = f.read()

    with open(os.path.join(workdir, "baz", "requirements.in")) as f:
        requirements = f.read()

    assert requirements == "bar\n"
    assert ducker_appenv == original_appenv