controller/rootfs/api/tests/test_service.py at e611c434a607f7b8670589f6c5f7bb46f23e660e · 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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
from django.contrib.auth import get_user_model
from django.core.cache import cache

from api.tests import DryccTransactionTestCase
from api.tests.test_gateway import RouteTest

User = get_user_model()


class ServiceTest(DryccTransactionTestCase):

    """Tests push notification from build system"""

    fixtures = ['tests.json']

    def setUp(self):
        self.user = User.objects.get(username='autotest')
        self.token = self.get_or_create_token(self.user)
        self.client.credentials(HTTP_AUTHORIZATION='Token ' + self.token)
        self.test_route = RouteTest()
        self.test_route.user = self.user
        self.test_route.token = self.token
        self.test_route.client = self.client

    def tearDown(self):
        # make sure every test has a clean slate for k8s mocking
        cache.clear()

    def test_service_basic_ops(self):
        """Test basic service operations."""
        app_id = self.create_app()
        # list non-existing services
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(response.status_code, 200, response.data)
        self.assertEqual(len(response.data['services']), 0)
        # create 1st service
        response = self.client.post(
            '/v2/apps/{}/services'.format(app_id),
            {
                'port': 5000,
                'protocol': 'UDP',
                'target_port': 5000,
                'ptype': 'test'
            }
        )
        self.assertEqual(response.status_code, 201, response.data)
        # list 1st service
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(response.status_code, 200, response.data)
        self.assertEqual(len(response.data['services']), 1)
        expected0 = {
            "domain": "%s-%s.%s.svc" % (app_id, "test", app_id),
            "ports": [{
                'name': "%s-%s-%s-%s" % (app_id, "test", 'udp', 5000),
                'port': 5000,
                'protocol': 'UDP',
                'targetPort': 5000,
            }],
            "ptype": "test"
        }
        self.assertEqual(response.data['services'][0], expected0 | response.data['services'][0])
        # port is occupied
        response = self.client.post(
            '/v2/apps/{}/services'.format(app_id),
            {
                'port': 5000,
                'protocol': 'UDP',
                'target_port': 5000,
                'ptype': 'test'
            }
        )
        self.assertEqual(response.status_code, 400, response.data)

        # add new port
        expected1 = {
            "domain": "%s-%s.%s.svc" % (app_id, "test", app_id),
            "ports": [
                {
                    'name': "%s-%s-%s-%s" % (app_id, "test", 'udp', 5000),
                    'port': 5000,
                    'protocol': 'UDP',
                    'targetPort': 5000,
                },
                {
                    'name': "%s-%s-%s-%s" % (app_id, "test", 'tcp', 6000),
                    'port': 6000,
                    'protocol': 'TCP',
                    'targetPort': 6000,
                }
            ],
            "ptype": "test"
        }
        response = self.client.post(
            '/v2/apps/{}/services'.format(app_id),
            {
                'port': 6000,
                'protocol': 'TCP',
                'target_port': 6000,
                'ptype': 'test'
            }
        )
        self.assertEqual(response.status_code, 204, response.data)
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(response.status_code, 200, response.data)
        self.assertEqual(
            response.data['services'][0], expected1 | response.data['services'][0])

        # create 2nd service
        response = self.client.post(
            '/v2/apps/{}/services'.format(app_id),
            {
                'port': 5000,
                'protocol': 'UDP',
                'target_port': 5000,
                'ptype': 'test2'
            }
        )
        self.assertEqual(response.status_code, 201, response.data)
        # list two services
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(response.status_code, 200, response.data)
        self.assertEqual(len(response.data['services']), 2)
        expected2 = {
            "domain": "%s-%s.%s.svc" % (app_id, "test2", app_id),
            "ports": [{
                'name': "%s-%s-%s-%s" % (app_id, "test2", 'udp', 5000),
                'port': 5000,
                'protocol': 'UDP',
                'targetPort': 5000,
            }],
            "ptype": "test2"
        }
        self.assertEqual(response.data['services'][0], expected2 | response.data['services'][0])
        self.assertEqual(response.data['services'][1], expected1 | response.data['services'][1])
        # delete port
        response = self.client.delete(
            '/v2/apps/{}/services'.format(app_id),
            {'ptype': 'test', "protocol": "TCP", "port": 6000}
        )
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(response.data['services'][1], expected0 | response.data['services'][1])
        # delete 1st
        response = self.client.delete(
            '/v2/apps/{}/services'.format(app_id),
            {'ptype': 'test', "protocol": "UDP", "port": 5000}
        )
        self.assertEqual(response.status_code, 204, response.data)
        # delete 2nd
        response = self.client.delete(
            '/v2/apps/{}/services'.format(app_id),
            {'ptype': 'test2',  "protocol": "UDP", "port": 5000}
        )
        self.assertEqual(response.status_code, 204, response.data)
        response = self.client.get('/v2/apps/{}/services'.format(app_id))
        self.assertEqual(len(response.data["services"]), 0)
        # delete non-existing (1st again)
        response = self.client.delete(
            '/v2/apps/{}/services'.format(app_id),
            {'ptype': 'test', "protocol": "UDP", "port": 5000}
        )
        self.assertEqual(response.status_code, 404, response.data)