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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
|
import unittest
import sys
import os
__mydir = os.path.dirname(sys.argv[0])
if not os.path.isdir(__mydir):
__mydir = os.getcwd()
sys.path.append(__mydir + "/..")
import augeas
MYROOT = __mydir + "/testroot"
def recurmatch(aug, path):
if path:
if path != "/":
val = aug.get(path)
if val:
yield (path, val)
m = []
if path != "/":
aug.match(path)
for i in m:
for x in recurmatch(aug, i):
yield x
else:
for i in aug.match(path + "/*"):
for x in recurmatch(aug, i):
yield x
class TestAugeas(unittest.TestCase):
def test01Get(self):
"test aug_get"
a = augeas.Augeas(root=MYROOT)
self.failUnless(a.get("/wrong/path") == None)
del a
def test02Match(self):
"test aug_match"
a = augeas.Augeas(root=MYROOT)
matches = a.match("/files/etc/hosts/*")
self.failUnless(matches)
for i in matches:
for attr in a.match(i+"/*"):
self.failUnless(a.get(attr) != None)
del a
def test03PrintAll(self):
"print all tree elements"
a = augeas.Augeas(root=MYROOT)
path = "/"
matches = recurmatch(a, path)
for (p, attr) in matches:
print >> sys.stderr, p, attr
self.failUnless(p != None and attr != None)
def test04Grub(self):
"test default setting of grub entry"
a = augeas.Augeas(root=MYROOT)
num = 0
for entry in a.match("/files/etc/grub.conf/title"):
num += 1
self.failUnless(num == 2)
default = int(a.get("/files/etc/grub.conf/default"))
self.failUnless(default == 0)
a.set("/files/etc/grub.conf/default", str(1))
a.save()
default = int(a.get("/files/etc/grub.conf/default"))
self.failUnless(default == 1)
a.set("/files/etc/grub.conf/default", str(0))
a.save()
def test05Defvar(self):
"test defvar"
a = augeas.Augeas(root=MYROOT)
a.defvar("hosts", "/files/etc/hosts")
matches = a.match("$hosts/*")
self.failUnless(matches)
for i in matches:
for attr in a.match(i+"/*"):
self.failUnless(a.get(attr) != None)
del a
def test06Defnode(self):
"test defnode"
a = augeas.Augeas(root=MYROOT)
a.defnode("bighost", "/files/etc/hosts/50/ipaddr", "192.168.1.1")
value = a.get("$bighost")
self.failUnless(value == "192.168.1.1")
del a
def test07Setm(self):
"test setm"
a = augeas.Augeas(root=MYROOT)
matches = a.match("/files/etc/hosts/*/ipaddr")
self.failUnless(matches)
a.setm("/files/etc/hosts", "*/ipaddr", "192.168.1.1")
for i in matches:
self.failUnless(a.get(i) == "192.168.1.1")
del a
def getsuite():
suite = unittest.TestSuite()
suite = unittest.makeSuite(TestAugeas, 'test')
return suite
if __name__ == "__main__":
__testRunner = unittest.TextTestRunner(verbosity=2)
__result = __testRunner.run(getsuite())
sys.exit(not __result.wasSuccessful())
__author__ = "Harald Hoyer <harald@redhat.com>"
|