workflow-cli/api/tests/test_flavor.py at f7d887dda25e003908ab9377ef74372a472d7e09 · drycc/workflow-cli · 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
# -*- coding: utf-8 -*-
"""
Unit tests for the Deis api app.

Run the tests with "./manage.py test api"
"""

from __future__ import unicode_literals

import json

from django.test import TestCase

from api.models import Flavor


class FlavorTest(TestCase):

    """Tests creation of different node flavors"""

    fixtures = ['tests.json']

    def setUp(self):
        self.assertTrue(
            self.client.login(username='autotest', password='password'))
        url = '/api/providers'
        creds = {'secret_key': 'x' * 64, 'access_key': 1 * 20}
        body = {'id': 'autotest', 'type': 'mock', 'creds': json.dumps(creds)}
        response = self.client.post(url, json.dumps(body), content_type='application/json')
        self.assertEqual(response.status_code, 201)

    def test_flavor(self):
        """
        Test that a user can create, read, update and delete a node flavor
        """
        url = '/api/flavors'
        # Ensure that dashes and underscores are allowed in id, per Django's
        # definition of a SlugField.
        body = {'id': 'auto_test-1', 'provider': 'autotest',
                'params': json.dumps({'region': 'us-west-2', 'instance_size': 'm1.medium'})}
        response = self.client.post(url, json.dumps(body), content_type='application/json')
        self.assertEqual(response.status_code, 201)
        flavor_id = response.data['id']
        response = self.client.get('/api/flavors')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(len(response.data['results']), 1)
        url = "/api/flavors/{flavor_id}".format(**locals())
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
        response = self.client.delete(url)
        self.assertEqual(response.status_code, 204)

    def test_flavor_update(self):
        """Tests that flavors can be updated by the client."""
        url = '/api/flavors'
        params = {
            'region': 'us-west-2',
            'size': 't1.micro',
            'zone': 'any',
            'image': 'i-1234567'
        }
        body = {'id': 'auto_test', 'provider': 'autotest', 'params': json.dumps(params)}
        response = self.client.post(url, json.dumps(body), content_type='application/json')
        self.assertEqual(response.status_code, 201)
        flavor_id = response.data['id']
        response = self.client.get('/api/flavors')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(len(response.data['results']), 1)
        url = "/api/flavors/{flavor_id}".format(**locals())
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
        uuid = response.data['uuid']
        self.assertRegexpMatches(uuid, r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$')
        params = json.loads(response.data['params'])
        self.assertEqual(params['region'], 'us-west-2')
        self.assertEqual(params['zone'], 'any')
        self.assertEqual(params['size'], 't1.micro')
        self.assertEqual(params['image'], 'i-1234567')
        params = {
            'size': 'c1.xlarge',
            'image': 'ami-c98d1bf9',
            'zone': None
        }
        body = {'id': flavor_id, 'params': json.dumps(params)}
        response = self.client.patch(url, json.dumps(body), content_type='application/json')
        self.assertEqual(response.status_code, 200)
        flavor_id = response.data['id']
        response = self.client.get('/api/flavors')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(len(response.data['results']), 1)
        url = "/api/flavors/{flavor_id}".format(**locals())
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data['uuid'], uuid)
        params = json.loads(response.data['params'])
        self.assertIn('region', params)
        self.assertNotIn('zone', params)
        self.assertEqual(params['size'], 'c1.xlarge')
        self.assertEqual(params['image'], 'ami-c98d1bf9')

    def test_flavor_str(self):
        """Test the text representation of a flavor."""
        url = '/api/flavors'
        params = {
            'region': 'us-west-2',
            'size': 't1.micro',
            'zone': 'any',
            'image': 'i-1234567'
        }
        body = {'id': 'auto_test', 'provider': 'autotest', 'params': json.dumps(params)}
        response = self.client.post(url, json.dumps(body), content_type='application/json')
        self.assertEqual(response.status_code, 201)
        flavor = Flavor.objects.get(uuid=response.data['uuid'])
        self.assertEqual(str(flavor), 'auto_test')