tests/testRecord.py

   1 import unittest
   2 import xmlrpclib
   3 from record import *
   4 from cert import *
   5 from gid import *
   6
   7 class TestRecord(unittest.TestCase):
   8     def setUp(self):
   9         pass
  10
  11     def testCreate(self):
  12         r = GeniRecord()
  13
  14 class TestTable(unittest.TestCase):
  15
  16     def setUp(self):
  17         self.reg_hrn = "test.table"
  18         self.rec_hrn = self.reg_hrn + "." + "record"
  19         pass
  20
  21     def testCreate(self):
  22         t = GeniTable(hrn = self.reg_hrn)
  23         t.create()
  24
  25     def testInsert(self):
  26         t = GeniTable(hrn = self.reg_hrn)
  27
  28         k = Keypair(create=True)
  29         gid = GID(subject="scott.foo", uuid=create_uuid(), hrn=self.rec_hrn)
  30         gid.set_pubkey(k)
  31         gid.set_issuer(key=k, subject=self.rec_hrn)
  32         gid.encode()
  33         gid.sign()
  34
  35         r = GeniRecord(name=self.rec_hrn, gid=gid.save_to_string(), type="user", pointer=3)
  36         t.insert(r)
  37
  38     def testLookup(self):
  39         t = GeniTable(hrn = self.reg_hrn)
  40
  41         rec_list = t.lookup(self.rec_hrn)
  42         self.assertEqual(len(rec_list), 1)
  43         r = rec_list[0]
  44         self.assertEqual(r.name, self.rec_hrn)
  45         self.assertEqual(r.pointer, 3)
  46
  47     def testUpdate(self):
  48         t = GeniTable(hrn = self.reg_hrn)
  49
  50         rec_list = t.lookup(self.rec_hrn)
  51         r = rec_list[0]
  52
  53         r.set_pointer(4)
  54         t.update(r)
  55
  56         rec_list = t.lookup(self.rec_hrn)
  57         self.assertEqual(len(rec_list), 1)
  58         r = rec_list[0]
  59         self.assertEqual(r.name, self.rec_hrn)
  60         self.assertEqual(r.pointer, 4)
  61
  62 if __name__ == "__main__":
  63     unittest.main()