← Back to branch summary

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

« back to all changes in this revision

Viewing changes to tests/test_assess_multi_series_charms.py

  • Committer: Curtis Hovey
  • Date: 2016-06-15 20:52:35 UTC
  • Revision ID: curtis@canonical.com-20160615205235-cf6hu9xt1qmbo1a4
Escape vars in run-unit-tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_assess_multi_series_charms.py
48
48
class TestMain(TestCase):
49
49
    def test_main(self):
50
50
        argv = ["an-env", "/bin/juju", "/tmp/logs", "an-env-mod", "--verbose"]
 
51
        env = object()
51
52
        client = Mock(spec=["is_jes_enabled"])
52
53
        with patch("assess_multi_series_charms.configure_logging",
53
54
                   autospec=True) as mock_cl:
54
55
            with patch("assess_multi_series_charms.BootstrapManager."
55
56
                       "booted_context",
56
57
                       autospec=True) as mock_bc:
57
 
                with patch("deploy_stack.client_from_config",
58
 
                           return_value=client) as mock_c:
59
 
                    with patch("assess_multi_series_charms."
60
 
                               "assess_multi_series_charms",
61
 
                               autospec=True) as mock_assess:
62
 
                        main(argv)
 
58
                with patch("jujupy.SimpleEnvironment.from_config",
 
59
                           return_value=env) as mock_e:
 
60
                    with patch("jujupy.EnvJujuClient.by_version",
 
61
                               return_value=client) as mock_c:
 
62
                        with patch("assess_multi_series_charms."
 
63
                                   "assess_multi_series_charms",
 
64
                                   autospec=True) as mock_assess:
 
65
                            main(argv)
63
66
        mock_cl.assert_called_once_with(logging.DEBUG)
64
 
        mock_c.assert_called_once_with('an-env', "/bin/juju", debug=False,
65
 
                                       soft_deadline=None)
 
67
        mock_e.assert_called_once_with("an-env")
 
68
        mock_c.assert_called_once_with(env, "/bin/juju", debug=False)
66
69
        self.assertEqual(mock_bc.call_count, 1)
67
70
        mock_assess.assert_called_once_with(client)
68
71