Ticket #23: selfref_test.patch

File selfref_test.patch, 0.8 kB (added by guest, 5 years ago)

Additional patch that adds a test for self-referential relations.

  • tests/test_selfref.py

     
    88class TestSelfRef(object): 
    99    def setup(self): 
    1010        metadata.bind = 'sqlite:///' 
    11      
     11 
     12    def test_self_referential(self): 
     13        class Node(Entity): 
     14            id = Field(Integer, primary_key=True) 
     15            parent = ManyToOne("Node", backref="children") 
     16        setup_all(True) 
     17         
     18        n1 = Node(id=1) 
     19        n2 = Node(id=2, parent=n1) 
     20        n3 = Node(id=3, parent=n1) 
     21        session.flush() 
     22        session.clear() 
     23        n_1 = Node.query.get(1) 
     24        assert Node.query.get(2) in n_1.children 
     25        assert Node.query.get(3) in n_1.children 
     26         
    1227    def teardown(self): 
    1328        cleanup_all() 
    1429