← Back to branch summary

~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-03-18 03:02:01 UTC
  • mto: This revision was merged to the branch mainline in revision 1323.
  • Revision ID: aaron.bentley@canonical.com-20160318030201-ki8wfpaeawgh9oqz
assess_cs_staging pokes state server via admin model.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_assess_recovery.py
33
33
        self.assertEqual(args.juju_path, 'foo')
34
34
        self.assertEqual(args.env_name, 'bar')
35
35
        self.assertEqual(args.logs, 'baz')
36
 
        self.assertEqual(args.charm_series, '')
 
36
        self.assertEqual(args.charm_prefix, '')
37
37
        self.assertEqual(args.strategy, 'backup')
38
38
        self.assertEqual(args.debug, False)
39
39
        self.assertIs(args.agent_stream, None)
51
51
        args = parse_args(['foo', 'bar', 'baz', '--ha', '--backup'])
52
52
        self.assertEqual(args.strategy, 'backup')
53
53
 
54
 
    def test_parse_args_charm_series(self):
55
 
        args = parse_args(['foo', 'bar', 'baz', '--charm-series', 'qux'])
56
 
        self.assertEqual(args.charm_series, 'qux')
 
54
    def test_parse_args_charm_prefix(self):
 
55
        args = parse_args(['foo', 'bar', 'baz', '--charm-prefix', 'qux'])
 
56
        self.assertEqual(args.charm_prefix, 'qux')
57
57
 
58
58
    def test_parse_args_debug(self):
59
59
        args = parse_args(['foo', 'bar', 'baz', '--debug'])
176
176
        with patch('assess_recovery.make_client_from_args', autospec=True,
177
177
                   return_value=client) as mc_mock:
178
178
            main(['./', 'foo', 'log_dir',
179
 
                  '--ha', '--charm-series', 'charm-series'])
 
179
                  '--ha', '--charm-prefix', 'prefix'])
180
180
        mc_mock.assert_called_once_with(Namespace(
181
 
            agent_stream=None, charm_series='charm-series', debug=False,
 
181
            agent_stream=None, charm_prefix='prefix', debug=False,
182
182
            env_name='foo', juju_path='./', logs='log_dir', strategy='ha',
183
183
            temp_env_name=None, series=None))
184
184
        client.wait_for_ha.assert_called_once_with()
186
186
        self.assertEqual(2, client.kill_controller.call_count)
187
187
        dns_mock.assert_called_once_with(client.get_admin_client.return_value,
188
188
                                         '0')
189
 
        ds_mock.assert_called_once_with(client, 'charm-series')
 
189
        ds_mock.assert_called_once_with(client, 'prefix')
190
190
        dcm_mock.assert_called_once_with(client, leader_only=True)
191
191
        self.assertTrue(dal_mock.called)
192
192
        self.assertEqual(0, ns_mock.call_count)
199
199
                   return_value=client) as mc_mock:
200
200
            with self.assertRaises(SystemExit):
201
201
                    main(['./', 'foo', 'log_dir',
202
 
                          '--ha', '--charm-series', 'prefix'])
 
202
                          '--ha', '--charm-prefix', 'prefix'])
203
203
        mc_mock.assert_called_once_with(Namespace(
204
 
            agent_stream=None, charm_series='prefix', debug=False,
 
204
            agent_stream=None, charm_prefix='prefix', debug=False,
205
205
            env_name='foo', juju_path='./', logs='log_dir', strategy='ha',
206
206
            temp_env_name=None, series=None))
207
207
        client.wait_for_ha.assert_called_once_with()
223
223
            with patch.object(client, 'bootstrap', side_effect=Exception):
224
224
                with self.assertRaises(SystemExit):
225
225
                    main(['./', 'foo', 'log_dir',
226
 
                          '--ha', '--charm-series', 'prefix'])
 
226
                          '--ha', '--charm-prefix', 'prefix'])
227
227
        self.assertEqual(2, client.kill_controller.call_count)
228
228
 
229
229