~andrewjbeach/juju-ci-tools/make-local-patcher

« back to all changes in this revision

Viewing changes to tests/test_assess_recovery.py

  • Committer: Aaron Bentley
  • Date: 2016-10-22 02:03:03 UTC
  • mto: (1674.1.10 indirect-config-4)
  • mto: This revision was merged to the branch mainline in revision 1677.
  • Revision ID: aaron.bentley@canonical.com-20161022020303-de22m9wug4t2chrz
Use provider rather than get_provider.

Show diffs side-by-side

added added

removed removed

Lines of Context:
180
180
    def test_delete_controller_members(self, ti_mock, wsss_mock):
181
181
        client = Mock(spec=['env', 'get_controller_members'])
182
182
        client.env = sentinel.env
183
 
        client.env.get_provider = lambda: 'lxd'
 
183
        client.env.provider = 'lxd'
184
184
        client.get_controller_members.return_value = [
185
185
            Machine('3', {
186
186
                'dns-name': '10.0.0.3',
221
221
    def test_delete_controller_members_leader_only(self, ti_mock, wsss_mock):
222
222
        client = Mock(spec=['env', 'get_controller_leader'])
223
223
        client.env = sentinel.env
224
 
        client.env.get_provider = lambda: 'lxd'
 
224
        client.env.provider = 'lxd'
225
225
        client.get_controller_leader.return_value = Machine('3', {
226
226
            'dns-name': '10.0.0.3',
227
227
            'instance-id': 'juju-dddd-machine-3',
240
240
    def test_delete_controller_members_azure(self, ti_mock, wsss_mock):
241
241
        client = Mock(spec=['env', 'get_controller_leader'])
242
242
        client.env = sentinel.env
243
 
        client.env.get_provider = lambda: 'azure'
 
243
        client.env.provider = 'azure'
244
244
        client.get_controller_leader.return_value = Machine('3', {
245
245
            'dns-name': '10.0.0.3',
246
246
            'instance-id': 'juju-dddd-machine-3',