test/design/box.py

   1 #!/usr/bin/env python
   2
   3 from neco.design.box import Box
   4
   5 import unittest
   6
   7 class BoxDesignTestCase(unittest.TestCase):
   8     def test_simple_design(self):
   9         node1 = Box()
  10         node2 = Box()
  11
  12         node1.label = "uno"
  13         node2.label = "dos"
  14
  15         node1.tadd('nodo')
  16         node2.tadd('mynodo')
  17
  18         self.assertEquals(node1.tags, set(['nodo']))
  19         self.assertEquals(node2.tags, set(['mynodo']))
  20
  21         node1.a.hola = "chau"
  22         node2.a.hello = "bye"
  23
  24         self.assertEquals(node1.a.hola, "chau")
  25         self.assertEquals(node2.a.hello, "bye")
  26
  27         node1.connect(node2)
  28
  29         self.assertEquals(node1.connections, set([node2]))
  30         self.assertEquals(node2.connections, set([node1]))
  31         self.assertTrue(node1.is_connected(node2))
  32         self.assertTrue(node2.is_connected(node1))
  33
  34         self.assertEquals(node1.c.dos.a.hello, "bye")
  35         self.assertEquals(node2.c.uno.a.hola, "chau")
  36
  37         node2.disconnect(node1)
  38
  39         self.assertEquals(node1.connections, set([]))
  40         self.assertEquals(node2.connections, set([]))
  41         self.assertFalse(node1.is_connected(node2))
  42         self.assertFalse(node2.is_connected(node1))
  43
  44         self.assertRaises(AttributeError, node1.c.dos)
  45         self.assertRaises(AttributeError, node2.c.uno)
  46
  47
  48 if __name__ == '__main__':
  49     unittest.main()
  50