controller/client/tests/test_sharing.py at 4694844e603583d2b0cee0fa0b4922a9857dc5d9 · drycc/controller · 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
82
83
84
85
86
87
88
89
90
91
92
93
94
"""
Unit tests for the Deis example-[language] projects.

Run these tests with "python -m unittest client.tests.test_sharing"
or with "./manage.py test client.SharingTest".
"""

from __future__ import unicode_literals
from unittest import TestCase
from uuid import uuid4
import os

import pexpect
from .utils import DEIS
from .utils import DEIS_TEST_FLAVOR
from .utils import EXAMPLES
from .utils import clone
from .utils import login
from .utils import purge
from .utils import register


class SharingTest(TestCase):

    @classmethod
    def setUpClass(cls):
        cls.username2, cls.password2 = register()
        cls.username, cls.password = register()
        # create a new formation
        cls.formation = "{}-test-formation-{}".format(
            cls.username, uuid4().hex[:4])
        child = pexpect.spawn("{} formations:create {} --flavor={}".format(
            DEIS, cls.formation, DEIS_TEST_FLAVOR))
        child.expect("created {}.*to scale a basic formation".format(
            cls.formation))
        child.expect(pexpect.EOF)

    @classmethod
    def tearDownClass(cls):
        purge(cls.username2, cls.password2)
        login(cls.username, cls.password)
        child = pexpect.spawn("{} formations:destroy {} --confirm={}".format(
            DEIS, cls.formation, cls.formation))
        child.expect('done in ', timeout=5 * 60)
        child.expect(pexpect.EOF)
        purge(cls.username, cls.password)

    def _test_sharing(self, repo_name):
        # `git clone` the example app repository
        repo_type, repo_url = EXAMPLES[repo_name]
        clone(repo_url, repo_name)
        # create an App
        child = pexpect.spawn("{} create --formation={}".format(
            DEIS, self.formation))
        child.expect('done, created (?P<name>[-_\w]+)', timeout=3 * 60)
        app = child.match.group('name')
        try:
            child.expect('Git remote deis added')
            child.expect(pexpect.EOF)
            home = os.environ['HOME']
            login(self.username2, self.password2)
            os.chdir(os.path.join(home, repo_name))
            child = pexpect.spawn('git push deis master')
            child.expect('access denied')
            child.expect(pexpect.EOF)
            login(self.username, self.password)
            os.chdir(os.path.join(home, repo_name))
            child = pexpect.spawn("{} sharing:add {} --app={}".format(
                DEIS, self.username2, app))
            child.expect('done')
            child.expect(pexpect.EOF)
            login(self.username2, self.password2)
            os.chdir(os.path.join(home, repo_name))
            child = pexpect.spawn('git push deis master')
            # check git output for repo_type, e.g. "Clojure app detected"
            # TODO: for some reason, the next regex times out...
            # child.expect("{} app detected".format(repo_type), timeout=5 * 60)
            child.expect('Launching... ', timeout=10 * 60)
            child.expect('deployed to Deis(?P<url>.+)To learn more', timeout=3 * 60)
            url = child.match.group('url')  # noqa
            child.expect(' -> master')
            child.expect(pexpect.EOF, timeout=2 * 60)
        finally:
            login(self.username, self.password)
            os.chdir(os.path.join(home, repo_name))
            # destroy the app
            child = pexpect.spawn(
                "{} apps:destroy --app={} --confirm={}".format(DEIS, app, app),
                timeout=5 * 60)
            child.expect('Git remote deis removed')
            child.expect(pexpect.EOF)

    def test_go(self):
        self._test_sharing('example-go')