Initial check-in

Author: wcooley

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+.*.swp

genders.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python26
+
+#from ctypes import CDLL, c_char_p, pointer, byref
+from ctypes import *
+
+libgenders = CDLL("libgenders.so")
+
+class Genders(object):
+    def __init__(self, genders_file=None):
+        pass
+#        self.handle_create()
+
+#        if self.load_data(genders_file) != 0:
+#            raise Exception(self.errormsg())
+
+    def handle_create(self):
+        self._handle = c_void_p(libgenders.genders_handle_create())
+
+        if not self._handle:
+            raise Exception("Error allocating memory")
+
+    def handle_destroy(self):
+        if libgenders.genders_handle_destroy(self._handle) != 0:
+            raise Exception(self.errormsg())
+
+    def load_data(self, genders_file=None):
+        if libgenders.genders_load_data(self._handle, c_char_p(genders_file)) != 0:
+            raise Exception(self.errormsg())
+
+    def isnode(self, node=None):
+        node = c_char_p(node)
+        return bool(libgenders.genders_isnode(self._handle, node))
+
+    def isattr(self, attr=None):
+        attr = c_char_p(attr)
+        return bool(libgenders.genders_isattr(self._handle, attr))
+
+    def isattrval(self, attr=None, val=None):
+        attr = c_char_p(attr)
+        val = c_char_p(val)
+        return bool(libgenders.genders_isattrval(self._handle, attr, val))
+
+    def errnum(self):
+        return libgenders.genders_errnum(self._handle)
+
+    def strerror(self, err):
+        libgenders.genders_strerror.restype = c_char_p
+        return libgenders.genders_strerror(err)
+
+    def errormsg(self):
+        libgenders.genders_errormsg.restype = c_char_p
+        return libgenders.genders_errormsg(self._handle)
+
+    def perror(self, msg=None):
+        libgenders.genders_perror(self._handle, msg)
+
+    def nodelist_create(self):
+        self.node_list = POINTER(POINTER(POINTER(c_char)))
+        #libgenders.genders_nodelist_create.
+        libgenders.genders_nodelist_create(self._handle, self.node_list)
+
+    def query(self, query_str):
+        query_str = c_char_p(query_str)
+
+        #nodes = byref(c_char_p * 500)
+        #nodes = pointer(c_char_p * 500)
+        #nodes = POINTER(c_char_p * 500)
+        node_buf = (c_char_p * 500)()
+        libgenders.genders_query.argtypes = [c_int, POINTER(c_char_p), c_int, c_char_p]
+        query_ret = libgenders.genders_query(self._handle, cast(node_buf, POINTER(c_char_p)), 500, query_str)
+        print "ret:", query_ret
+
+        return node_buf

test-data/genders

@@ -0,0 +1 @@
+host1   os=rhel5,testhost

test-data/genders.dup-attr

@@ -0,0 +1 @@
+host1   attrdup,attrdup

tests/genders_core.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python26
+
+from pprint import pprint as pp
+import sys
+import unittest
+
+sys.path.insert(0, '.')
+sys.path.insert(0, '..')
+
+import genders
+
+class TestGendersCore(unittest.TestCase):
+
+    def setUp(self):
+        self.genders = genders.Genders()
+
+    def test_handle_create(self):
+        self.assertEqual(self.genders.handle_create(), None)
+
+    def test_handle_destroy(self):
+        self.genders.handle_create()
+        self.assertEqual(self.genders.handle_destroy(), None)
+
+class TestGendersLoad(unittest.TestCase):
+
+    def setUp(self):
+        self.genders = genders.Genders()
+        self.genders.handle_create()
+
+    def test_load_data(self):
+        self.assertEqual(self.genders.load_data("test-data/genders"), None)
+
+#    def test_load_data_fail(self):
+#        self.assertRaises(Exception("genders file parse error"), self.genders.load_data("test-data/genders.dup-attr"))
+
+class TestGendersTests(unittest.TestCase):
+
+    def setUp(self):
+        self.genders = genders.Genders()
+        self.genders.handle_create()
+        self.genders.load_data("test-data/genders")
+
+    def test_isnode_true(self):
+        self.assertTrue(self.genders.isnode("host1"))
+
+    def test_isnode_false(self):
+        self.assertFalse(self.genders.isnode("Xhost1"))
+
+if __name__ == '__main__':
+    loader = unittest.TestLoader()
+    suite = loader.loadTestsFromTestCase(TestGendersCore)
+    suite.addTest(loader.loadTestsFromTestCase(TestGendersLoad))
+    suite.addTest(loader.loadTestsFromTestCase(TestGendersTests))
+
+    unittest.TextTestRunner(verbosity=2).run(suite)
+