def ping(self):
"""
Add a key to the `config.etcd_path` with a TTL of `config.ping_ttl`.
A wrapper around `client.add_key`.
"""
path = '{prefix}/{hostname}'.format(prefix=self._config.etcd_path,
hostname=util.get_hostname())
response = self._client.write(path,
time.time(),
ttl=self._config.ping_ttl)
if response.action == 'set':
# TODO(retr0h): log
print 'ping -> [ok]'
def test_get_hostname(self, mocked):
mocked.return_value = 'mocked-fqdn'
result = util.get_hostname()
self.assertEquals('mocked-fqdn', result)
