blob: 240b47dd16055b1f5e7d19319649f39882929442 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#!/usr/bin/env python
"""
To use with secret keys, do:
$ echo "<key>" | ./test.py -k
"""
import sys
import optparse
import logging
import socket
import bjsonrpc
import bjsonrpc.connection
import augeas
from exmachina import ExMachinaClient
# =============================================================================
# Command line handling
def main():
socket_path="/tmp/exmachina.sock"
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.connect(socket_path)
secret_key = None
if sys.argv[-1] == "-k":
print "waiting for key on stdin..."
secret_key = sys.stdin.readline()
print "sent!"
print "========= Testing low level connection"
c = bjsonrpc.connection.Connection(sock)
if secret_key:
c.call.authenticate(secret_key)
print "time: %s" % c.call.test_whattime()
print "/*: %s" % c.call.augeas_match("/*")
print "/augeas/*: %s" % c.call.augeas_match("/augeas/*")
print "/etc/* files:"
for name in c.call.augeas_match("/files/etc/*"):
print "\t%s" % name
print c.call.initd_status("bluetooth")
print "hostname: %s" % c.call.augeas_get("/files/etc/hostname/*")
print "localhost: %s" % c.call.augeas_get("/files/etc/hosts/1/canonical")
sock.close()
print "========= Testing user client library"
client = ExMachinaClient(secret_key=secret_key)
print client.augeas.match("/files/etc/*")
print client.initd.restart("bluetooth")
print client.initd.status("greentooth")
print "(expect Error on the above line)"
print client.initd.status("bluetooth")
client.close()
if __name__ == '__main__':
main()
|