Ticket #1151: testParentChildWithRefToNonPK.py

#!/usr/bin/env python
# -*- coding: iso8859_15 -*-

import dabo
import unittest
from dabo.dLocalize import _
from dabo.dApp import dApp
from dabo.db.dConnectInfo import dConnectInfo

class BizMaster(dabo.biz.dBizobj):
    def initProperties(self):
        self.DataSource = "master"
        self.KeyField = "id"
        self.AutoPopulatePK = True
        self.DataStructure = (
                ("id", "I", True, "master", "id"),
                ("key", "I", False, "master", "key"),
                ("info", "C", False, "master", "info"),
        )       
        self.addFrom("master")
        self.addField("id")
        self.addField("key")
        self.addField("info")

class BizDetails(dabo.biz.dBizobj):
    def initProperties(self):
        self.DataSource = "details"
        self.KeyField = "id"
        self.AutoPopulatePK = True
        self.DataStructure = (
                ("id", "I", True, "details", "id"),
                ("key_ref", "I", False, "details", "key_ref"),
                ("optional_info", "C", False, "details", "optional_info"),
        )       
        self.addFrom("details")
        self.addField("id")
        self.addField("optional_info")
        self.addField("key_ref")
            
class BlackHole:
    def write(self, s):
        pass

class TestParentChildWithRefToNonPK(unittest.TestCase):
    def setUp(self):
        dabo.infoLog.LogObject = BlackHole()
        app = dApp()
        
        app.MainFormClass = None

        dbFileName = ":memory:"
        connInfo = dConnectInfo(Name="myconn", DbType="SQLite", Database=dbFileName) 
        app.addConnectInfo(connInfo)
            
        ## create database on the fly:
        conn = app.getConnectionByName("myconn")
        curs= conn.getDaboCursor()
        # create tables and testdata
        scriptList = """
            create table master (
                id integer primary key autoincrement not null,
                key integer,
                info text
            );
            create table details (
                id integer primary key autoincrement not null,
                key_ref integer,
                optional_info text
            );
            insert into master (key, info) values (1, 'row1key1');
            insert into master (key, info) values (1, 'row2key1');
            insert into master (key, info) values (2, 'row3key2');
            insert into details (key_ref, optional_info) values (1, 'option1forkey1');
            insert into details (key_ref, optional_info) values (1, 'option2forkey1');
            insert into details (key_ref, optional_info) values (1, 'option3forkey1');
            insert into details (key_ref, optional_info) values (2, 'option1forkey2');
            """.split(";")
        for stmt in scriptList:
            if stmt.strip(): 
                curs.execute(stmt)

        ## start the app
        app.setup()
        app.start()
 
        # create bizobjs
        conn = app.getConnectionByName("myconn")
        self.master = master = BizMaster(conn)
        self.details = details = BizDetails(conn)
        
        # link details to master
        # we use details only for lookup, so do not 
        # create new child rows on master.new()
        details.LinkField = "key_ref"
        details.ParentLinkField = "key"
        details.NewRecordOnNewParent = False
        details.FillLinkFromParent = False
        master.NewChildOnNew = False
        master.addChild(details)
        
        master.requery()

    def test_setFieldVal(self):
        """After the key field is set, does the right child dataset show up?"""
        master = self.master
        details = self.details
        master.setFieldVal("key", 2)
        master.requeryAllChildren()
        self.assertEqual(details.RowCount, 1)
        self.assertEqual(details.getFieldVal("optional_info"), "option1forkey2")
        master.next()
        master.prior()
        self.assertEqual(details.RowCount, 1)
        self.assertEqual(details.getFieldVal("optional_info"), "option1forkey2")
        
    def test_new(self):
        """After new, after the key field is set, does the right child dataset show up?"""
        master = self.master
        details = self.details
        master.new()
        master.setFieldVal("key", 1)
        master.requeryAllChildren()
        print "details.getDataSet:", details.getDataSet()
        self.assertEqual(details.RowCount, 3)
    
if __name__ == "__main__":
    suite = unittest.TestLoader().loadTestsFromTestCase(TestParentChildWithRefToNonPK)
    unittest.TextTestRunner(verbosity=2).run(suite)