def test_setup_clean(self, kazoo_cls):
kazoo_cls.return_value = kazoo = mock.Mock()
zkcli.main(['--config', self.zk_config, "setup", "--clean"])
kazoo.ensure_path.assert_called_once_with("/base")
kazoo.delete.assert_called_once_with("/base", recursive=True)
kazoo.create.assert_called_once_with("/base/pd/system_boot", "",
makepath=True)
def test_setup_clean_epu(self, kazoo_cls):
kazoo_cls.return_value = kazoo = mock.Mock()
zkcli.main(['--config', self.zk_config, "setup", "--clean-epu"])
kazoo.ensure_path.assert_called_once_with("/base")
self.assertEqual(kazoo.delete.call_count, 2)
kazoo.delete.assert_any_call("/base/provisioner", recursive=True)
kazoo.delete.assert_any_call("/base/epum", recursive=True)
kazoo.create.assert_called_once_with("/base/pd/system_boot", "",
makepath=True)
def test_setup(self, kazoo_cls):
kazoo_cls.return_value = kazoo = mock.Mock()
zkcli.main(['--config', self.zk_config, "setup"])
kazoo.ensure_path.assert_called_once_with("/base")
kazoo.delete.assert_called_once_with("/base/provisioner/disabled")
kazoo.create.assert_called_once_with("/base/pd/system_boot", "",
makepath=True)
kazoo.reset_mock()
kazoo.delete.side_effect = NoNodeException()
zkcli.main(['--config', self.zk_config, "setup"])
kazoo.delete.assert_called_once_with("/base/provisioner/disabled")
kazoo.reset_mock()
kazoo.create.side_effect = NodeExistsException()
zkcli.main(['--config', self.zk_config, "setup"])
kazoo.create.assert_called_once_with("/base/pd/system_boot", "",
makepath=True)
def test_destroy(self, kazoo_cls):
kazoo_cls.return_value = kazoo = mock.Mock()
zkcli.main(['--config', self.zk_config, "destroy"])
kazoo.delete.assert_called_once_with("/base", recursive=True)
