def test_connect_cli_option(self):
"""The zookeeper server can be passed via cli argument."""
mock_client = self.mocker.mock()
client = self.mocker.patch(ZookeeperClient)
client.connect("x2.example.com")
self.mocker.result(succeed(mock_client))
self.mocker.replay()
agent = BaseAgent()
agent.configure({"zookeeper_servers": "x2.example.com", "juju_directory": self.makeDir()})
result = agent.connect()
self.assertEqual(result.result, mock_client)
self.assertEqual(agent.client, mock_client)
def test_connect_cli_option(self):
"""The zookeeper server can be passed via cli argument."""
mock_client = self.mocker.mock()
client = self.mocker.patch(ZookeeperClient)
client.connect("x2.example.com")
self.mocker.result(succeed(mock_client))
self.mocker.replay()
agent = BaseAgent()
agent.configure({
"zookeeper_servers": "x2.example.com",
"juju_directory": self.makeDir()
})
result = agent.connect()
self.assertEqual(result.result, mock_client)
self.assertEqual(agent.client, mock_client)
