~hazmat/pyjuju/rapi-rollup

« back to all changes in this revision

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

  • Committer: Kapil Thangavelu
  • Date: 2012-12-17 22:14:37 UTC
  • mfrom: (616.1.2 rapi-login)
  • Revision ID: kapil@canonical.com-20121217221437-lhy4eck58wx3sf5n
merge rapi-login

Show diffs side-by-side

added added

removed removed

Lines of Context:
4
4
from txzookeeper.tests.utils import deleteTree
5
5
 
6
6
from juju.environment.tests.test_config import EnvironmentsConfigTestBase
7
 
from juju.state.auth import  make_identity
 
7
 
8
8
from juju.state.environment import (
9
9
    GlobalSettingsStateManager, EnvironmentStateManager)
10
10
from juju.state.initialize import StateHierarchy
11
11
from juju.state.machine import MachineStateManager
 
12
from juju.state.security import Principal
12
13
 
13
14
 
14
15
class LayoutTest(EnvironmentsConfigTestBase):
19
20
        self.log = self.capture_logging("juju.state.init")
20
21
        zookeeper.set_debug_level(0)
21
22
        self.client = self.get_zookeeper_client()
22
 
        self.identity = make_identity("admin:genie")
 
23
        self.principal = Principal("admin", "genie")
 
24
        self.identity = self.principal.get_token()
 
25
 
23
26
        constraints_data = {
24
27
            "arch": "arm",
25
28
            "cpu": None,
28
31
        self.layout = StateHierarchy(
29
32
            self.client, self.identity, "i-abcdef", constraints_data, "dummy")
30
33
        yield self.client.connect()
 
34
        yield self.principal.attach(self.client)
31
35
 
32
36
    def tearDown(self):
33
37
        deleteTree(handle=self.client.handle)
57
61
        yield self.assert_existence_and_acl("/machines")
58
62
        yield self.assert_existence_and_acl("/relations")
59
63
        yield self.assert_existence_and_acl("/initialized")
 
64
        yield self.assert_existence_and_acl("/login")
60
65
 
61
66
        # To check that the constraints landed correctly, we need the
62
67
        # environment config to have been sent, or we won't be able to