workflow/celerytasks/tests/chef.py at d64892abef69967fc2e5d28b196ad189b0b407ed · drycc/workflow · 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
"""
Unit tests for the Deis Chef module

Run the tests with "python -m unittest discover"
"""

from __future__ import unicode_literals

import json
import unittest

from celerytasks.chef import ChefAPI
from deis import settings


@unittest.skip('Need to set up TEST_CHEF_SERVER somehow.')
class ChefAPITest(unittest.TestCase):
    """Tests the client interface to Chef Server API."""

    def setUp(self):
        self.client = ChefAPI(
            settings.TEST_CHEF_SERVER_URL, settings.TEST_CHEF_CLIENT_NAME,
            settings.TEST_CHEF_CLIENT_KEY)

    def test_databag(self):
        databag_name = 'testing'
        ditem_name = 'item1'
        ditem_value = {'something': 1, 'else': 2}

        # delete the databag to make sure we are creating a new one
        resp, status = self.client.delete_databag(databag_name)

        resp, status = self.client.create_databag(databag_name)
        self.assertEqual(status, 201)
        self.assertTrue(resp)

        resp = self.client.create_databag_item(
            databag_name, ditem_name, ditem_value)
        self.assertEqual(status, 201)
        self.assertTrue(resp)

        resp, status = self.client.get_databag(databag_name)
        self.assertEqual(status, 200)
        resp, status = self.client.get_databag_item(databag_name, ditem_name)
        self.assertEqual(status, 200)

        ditem_value = json.loads(resp)
        ditem_value['newvalue'] = 'databag'
        resp, status = self.client.update_databag_item(
            databag_name, ditem_name, ditem_value)
        self.assertEqual(status, 200)
        resp, status = self.client.get_databag_item(databag_name, ditem_name)
        self.assertEqual(status, 200)
        self.assertTrue('newvalue' in json.loads(resp))