def test_main_upgrade(self, mock_docopt_docopt, mock_cli_upgrade):
     self.assertEquals(0, cli.main())
     mock_docopt_docopt.assert_called_once_with(cli.__doc__,
                                                help=False,
                                                version=ANY)
     mock_cli_upgrade.assert_called_once_with(host='fake_host',
                                            release='fake_release')
 def test_main(self, mock_docopt_docopt, mock_cli_print):
     self.assertEquals(1, cli.main())
     version_string = (six.text_type
                       ("dcos quobyte version " + constants.version))
     mock_docopt_docopt.assert_called_once_with(cli.__doc__,
                                                help=False,
                                                version=version_string)
     mock_cli_print.assert_called_once_with(cli.__doc__)
 def test_main_config_schema(self, mock_docopt_docopt, mock_cli_cschema):
     self.assertEquals(0, cli.main())
     mock_docopt_docopt.assert_called_once_with(cli.__doc__,
                                                help=False,
                                                version=ANY)
     mock_cli_cschema.assert_called_once_with()
 def test_main_stop(self, mock_docopt_docopt, mock_cli_stop):
     self.assertEquals(0, cli.main())
     mock_docopt_docopt.assert_called_once_with(cli.__doc__,
                                                help=False,
                                                version=ANY)
     mock_cli_stop.assert_called_once_with(host='fake_host')
 def test_main_info(self, mock_docopt_docopt, mock_cli_info):
     self.assertEquals(0, cli.main())
     mock_docopt_docopt.assert_called_once_with(cli.__doc__,
                                                help=False,
                                                version=ANY)
     mock_cli_info.assert_called_once_with()