← Back to branch summary

~hopem/charms/trusty/keystone/fix-ssl-disable

« back to all changes in this revision

Viewing changes to unit_tests/test_keystone_utils.py

[hopem,r=gnuoy]

Fixes is_db_ready() logic

Closes-Bug: 1417108

Show diffs side-by-side

added added

removed removed

Lines of Context:
unit_tests/test_keystone_utils.py
352
352
        self.assertEqual(utils.get_admin_passwd(), 'supersecretgen')
353
353
 
354
354
    def test_is_db_ready(self):
 
355
        allowed_units = None
 
356
 
 
357
        def fake_rel_get(attribute=None, *args, **kwargs):
 
358
            if attribute == 'allowed_units':
 
359
                return allowed_units
 
360
 
 
361
        self.relation_get.side_effect = fake_rel_get
 
362
 
355
363
        self.relation_id.return_value = 'shared-db:0'
356
 
        self.relation_ids.return_value = [self.relation_id.return_value]
 
364
        self.relation_ids.return_value = ['shared-db:0']
357
365
        self.local_unit.return_value = 'unit/0'
358
 
        self.relation_get.return_value = 'unit/0'
 
366
        allowed_units = 'unit/0'
359
367
        self.assertTrue(utils.is_db_ready(use_current_context=True))
360
368
 
361
369
        self.relation_ids.return_value = ['acme:0']
362
370
        self.assertRaises(utils.is_db_ready, use_current_context=True)
363
371
 
364
372
        self.related_units.return_value = ['unit/0']
365
 
        self.relation_ids.return_value = [self.relation_id.return_value]
 
373
        self.relation_ids.return_value = ['shared-db:0', 'shared-db:1']
366
374
        self.assertTrue(utils.is_db_ready())
367
375
 
368
 
        self.relation_get.return_value = 'unit/1'
 
376
        allowed_units = 'unit/1'
369
377
        self.assertFalse(utils.is_db_ready())
370
378
 
371
379
        self.related_units.return_value = []