root/trunk/test/test_storemsaccess.py

"""This works...most of the time. Sometimes during the Multithreading,
the Access SM will crash python."""

import warnings

try:
    import pythoncom
except ImportError:
    def run():
        warnings.warn("The pythoncom module could not be imported. "
                      "The MSAccess test will not be run.")
else:
    from dejavu.storage import storeado
    mod = storeado.gen_py()
    if mod is None:
        def run():
            warnings.warn("ADO 2.7 support could not be verified. "
                          "The MSAccess test will not be run.")
    else:
        
        class CurrencyAdapter(storeado.FieldTypeAdapter_MSAccess):
            """Stores Decimal and FixedPoint objects as CURRENCY."""
            
            def coerce_decimal_Decimal(self, cls, key):
                return "CURRENCY"

            def coerce_fixedpoint_FixedPoint(self, cls, key):
                return "CURRENCY"
            
        SM_class = "dejavu.storage.storeado.StorageManagerADO_MSAccess"
        opts = {u'Connect': "PROVIDER=MICROSOFT.JET.OLEDB.4.0;"
                            "DATA SOURCE=zoo.mdb;",
                }
        
        def run():
            import zoo_fixture
            
            # lists to keep track of the test_currency runs
            standard_runs = []
            altered_runs = []
            
            def test_currency(obj):
                sm = zoo_fixture.arena.stores['testSM']
                fta = sm.typeAdapter.__class__.__name__
                
                clsnames = ('Exhibit', 'Zoo')
                tblnames = [sm.table_name(cn).strip('[]') for cn in clsnames]
                colnames = ('Acreage', 'Admission')
                targets = dict(zip(tblnames, colnames))
                data, _ = sm.fetch(storeado.adSchemaColumns, conn=None,
                                   schema=True)
                for row in data:
                    match = targets.get(row[2])
                    if not match:
                        continue
                    if match == row[3]:
##                        print row[2], row[3], row[11]
                        dt = row[11]
                        
                        if fta in ("CurrencyAdapter",):
                            obj.assertEqual(dt, storeado.adCurrency)
                        else:
                            obj.assertEqual(dt, storeado.adDouble)
                            obj.assertEqual(len(standard_runs), 0)
            
            # Isolate schema changes from one test to the next.
            reload(zoo_fixture)
            
            # test the standard MS Access setup where Decimal and FixedPoint
            # objects are stored in the database as INTEGERS, LONGS or DOUBLES
            print
            print "Standard MSAccess test."
            zoo_fixture.ZooTests.test_currency = test_currency
            zoo_fixture.init()
            zoo_fixture.run(SM_class, opts)
            standard_runs.append(True)
            
            # plugin the adapter for testing CURRENCY columns
            storeado.StorageManagerADO_MSAccess.typeAdapter = CurrencyAdapter()
            
            reload(zoo_fixture)
            
            # test storing and recalling Decimal values to the database using
            # CURRENCY columns - the (current) default behavior of pythoncom
            # is to return CURRENCY values as a tuple
            print
            print "MSAccess test - CURRENCY returned as tuple."
            zoo_fixture.ZooTests.test_currency = test_currency
            zoo_fixture.init()
            zoo_fixture.run(SM_class, opts)
            altered_runs.append(True)
            
            # set pythoncom to return CURRENCY values as Decimal objects
            pythoncom.__future_currency__ = True
            
            reload(zoo_fixture)
            
            # test storing and recalling Decimal values to the database using
            # CURRENCY columns - CURRENCY values are now returned as Decimal
            # objects
            print
            print "MSAccess test - CURRENCY returned as Decimal."
            zoo_fixture.ZooTests.test_currency = test_currency
            zoo_fixture.init()
            zoo_fixture.run(SM_class, opts)
            altered_runs.append(True)


if __name__ == "__main__":
    run()