← Back to branch summary

~bcsaller/pyjuju/status-state-nodes

« back to all changes in this revision

Viewing changes to ensemble/state/tests/test_service.py

  • Committer: Benjamin Saller
  • Date: 2011-01-27 18:20:36 UTC
  • Revision ID: bcsaller@gmail.com-20110127182036-gvgtgxsuf93pxqfz
use _internal_id_to_id for machine id conversion
tests for get_all_unit_states
more method pattern renames

Show diffs side-by-side

added added

removed removed

Lines of Context:
ensemble/state/tests/test_service.py
287
287
        self.assertEquals(unit_state3.unit_name, "mysql/1")
288
288
 
289
289
    @inlineCallbacks
 
290
    def test_get_all_unit_states(self):
 
291
        service_state0 = yield self.service_state_manager.add_service_state(
 
292
            "wordpress", self.formula_state)
 
293
        service_state1 = yield self.service_state_manager.add_service_state(
 
294
            "mysql", self.formula_state)
 
295
 
 
296
        yield service_state0.add_unit_state()
 
297
        yield service_state1.add_unit_state()
 
298
        yield service_state0.add_unit_state()
 
299
        yield service_state1.add_unit_state()
 
300
 
 
301
        unit_state0 = yield service_state0.get_unit_state("wordpress/0")
 
302
        unit_state1 = yield service_state1.get_unit_state("mysql/0")
 
303
        unit_state2 = yield service_state0.get_unit_state("wordpress/1")
 
304
        unit_state3 = yield service_state1.get_unit_state("mysql/1")
 
305
 
 
306
        wordpress_units = yield service_state0.get_all_unit_states()
 
307
        self.assertEquals(set(wordpress_units), set((unit_state0, unit_state2)))
 
308
 
 
309
        mysql_units = yield service_state1.get_all_unit_states()
 
310
        self.assertEquals(set(mysql_units), set((unit_state1, unit_state3)))
 
311
 
 
312
    @inlineCallbacks
290
313
    def test_get_service_unit_not_found(self):
291
314
        """
292
315
        Attempting to retrieve a non-existent service unit should