← Back to branch summary

~nskaggs/juju-ci-tools/add-essential-operations

« back to all changes in this revision

Viewing changes to tests/test_assess_update_mongo.py

clean up

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_assess_update_mongo.py
41
41
    def test_main(self):
42
42
        argv = ["an-env", "/bin/juju", "/tmp/logs", "an-env-mod", "--verbose",
43
43
                "--series", "trusty", "--bootstrap-host", "10.0.0.2"]
44
 
        env = object()
45
44
        client = Mock(spec=["is_jes_enabled"])
46
45
        with patch("assess_update_mongo.configure_logging",
47
46
                   autospec=True) as mock_cl:
48
47
            with patch("assess_update_mongo.BootstrapManager.booted_context",
49
48
                       autospec=True) as mock_bc:
50
 
                with patch("jujupy.SimpleEnvironment.from_config",
51
 
                           return_value=env) as mock_e:
52
 
                    with patch("jujupy.EnvJujuClient.by_version",
53
 
                               return_value=client) as mock_c:
54
 
                        with patch("assess_update_mongo.assess_update_mongo",
55
 
                                   autospec=True) as mock_assess:
56
 
                            main(argv)
 
49
                with patch("deploy_stack.client_from_config",
 
50
                           return_value=client) as mock_c:
 
51
                    with patch("assess_update_mongo.assess_update_mongo",
 
52
                               autospec=True) as mock_assess:
 
53
                        main(argv)
57
54
        mock_cl.assert_called_once_with(logging.DEBUG)
58
 
        mock_e.assert_called_once_with("an-env")
59
 
        mock_c.assert_called_once_with(env, "/bin/juju", debug=False)
 
55
        mock_c.assert_called_once_with('an-env', "/bin/juju", debug=False)
60
56
        self.assertEqual(mock_bc.call_count, 1)
61
57
        mock_assess.assert_called_once_with(client, 'trusty', '10.0.0.2')
62
58