← Back to branch summary

~hazmat/pyjuju/rapi-delta

« back to all changes in this revision

Viewing changes to juju/state/tests/test_environment.py

  • Committer: Kapil Thangavelu
  • Date: 2012-09-19 20:38:44 UTC
  • mfrom: (573.1.6 rest-context)
  • Revision ID: kapil@canonical.com-20120919203844-dc2pf82ttm7xj3xs
Merged rest-context into rapi-delta.

Show diffs side-by-side

added added

removed removed

Lines of Context:
juju/state/tests/test_environment.py
1
 
import yaml
2
 
 
3
1
from twisted.internet.defer import inlineCallbacks, Deferred
4
2
 
5
3
from juju.environment.tests.test_config import (
6
4
    EnvironmentsConfigTestBase, SAMPLE_ENV)
7
5
from juju.machine.tests.test_constraints import dummy_cs
 
6
from juju.lib import serializer
8
7
from juju.state.errors import EnvironmentStateNotFound
9
8
from juju.state.environment import (
10
9
    EnvironmentStateManager, GlobalSettingsStateManager, SETTINGS_PATH)
38
37
 
39
38
        serialized = self.config.serialize("myfirstenv")
40
39
        content, stat = yield self.client.get("/environment")
41
 
        self.assertEquals(yaml.load(content), yaml.load(serialized))
 
40
        self.assertEquals(serializer.load(content),
 
41
                          serializer.load(serialized))
42
42
 
43
43
    @inlineCallbacks
44
44
    def test_set_config_state_replaces_environment(self):
53
53
 
54
54
        serialized = self.config.serialize("myfirstenv")
55
55
        content, stat = yield self.client.get("/environment")
56
 
        self.assertEquals(yaml.load(content), yaml.load(serialized))
 
56
        self.assertEquals(serializer.load(content),
 
57
                          serializer.load(serialized))
57
58
 
58
59
    @inlineCallbacks
59
60
    def test_get_config(self):
65
66
        config = yield manager.get_config()
66
67
        serialized1 = self.config.serialize("myfirstenv")
67
68
        serialized2 = config.serialize("myfirstenv")
68
 
        self.assertEquals(yaml.load(serialized1), yaml.load(serialized2))
 
69
        self.assertEquals(serializer.load(serialized1),
 
70
                          serializer.load(serialized2))
69
71
 
70
72
    def test_get_config_when_missing(self):
71
73
        """
140
142
        yield self.manager.set_provider_type("ec2")
141
143
        self.assertEqual((yield self.manager.get_provider_type()), "ec2")
142
144
        content, stat = yield self.client.get("/settings")
143
 
        self.assertEqual(yaml.load(content),
 
145
        self.assertEqual(serializer.load(content),
144
146
                         {"provider-type": "ec2"})
145
147
 
146
148
    @inlineCallbacks
211
213
 
212
214
        data = dict(x=1, y=2, z=3, moose=u"moon")
213
215
        yield self.client.set(
214
 
            SETTINGS_PATH, yaml.safe_dump(data))
 
216
            SETTINGS_PATH, serializer.dump(data))
215
217
        value = yield callbacks[1]
216
218
        self.assertEqual(value.type_name, "changed")
217
219