← Back to branch summary

~jorge/pyjuju/doc-bin-removal

« back to all changes in this revision

Viewing changes to ensemble/providers/orchestra/tests/test_state.py

  • Committer: William Reade
  • Date: 2011-08-05 07:17:27 UTC
  • mfrom: (281.3.10 webdav-storage)
  • Revision ID: fwereade@gmail.com-20110805071727-8oxoxnqgrpsgs6mt
orchestra provider now has a FileStorage implementation using WebDAV

Show diffs side-by-side

added added

removed removed

Lines of Context:
ensemble/providers/orchestra/tests/test_state.py
 
1
from yaml import dump
 
2
 
 
3
from twisted.internet.defer import succeed
 
4
 
 
5
from ensemble.lib.testing import TestCase
 
6
from ensemble.providers.orchestra import MachineProvider
 
7
 
 
8
 
 
9
def get_provider():
 
10
    config = {"orchestra-server": "somewhe.re",
 
11
              "orchestra-user": "user",
 
12
              "orchestra-pass": "pass",
 
13
              "acquired-mgmt-class": "acquired",
 
14
              "available-mgmt-class": "available"}
 
15
    return MachineProvider("tetrascape", config)
 
16
 
 
17
 
 
18
class StateTest(TestCase):
 
19
 
 
20
    def test_save(self):
 
21
        state = {"foo": "blah blah"}
 
22
        getPage = self.mocker.replace("twisted.web.client.getPage")
 
23
        getPage("http://somewhe.re/webdav/provider-state",
 
24
                method="PUT", postdata=dump(state))
 
25
        self.mocker.result(succeed(None))
 
26
        self.mocker.replay()
 
27
 
 
28
        provider = get_provider()
 
29
        d = provider.save_state(state)
 
30
 
 
31
        def verify(result):
 
32
            self.assertEquals(result, True)
 
33
        d.addCallback(verify)
 
34
        return d
 
35
 
 
36
    def test_load(self):
 
37
        expect_state = {"foo": "blah blah"}
 
38
        getPage = self.mocker.replace("twisted.web.client.getPage")
 
39
        getPage("http://somewhe.re/webdav/provider-state")
 
40
        self.mocker.result(succeed(dump(expect_state)))
 
41
        self.mocker.replay()
 
42
 
 
43
        provider = get_provider()
 
44
        d = provider.load_state()
 
45
 
 
46
        def verify(state):
 
47
            self.assertEquals(state, expect_state)
 
48
        d.addCallback(verify)
 
49
        return d