source: mauRepo/MolesManager/trunk/src/libs/migration/tests/migrationprocess.py @ 8198

'''
Created on 7 Mar 2012

@author: mnagni
'''
from unittest import TestCase
from libs.commons_db import DbManager
from libs.migration.MigrationEPB import MigrationEPB
from libs.migration.processor.loadResources import LoadResources
from libs.migration.db.classes import MigrationObject
from libs.migration.processor.commons import DO_BADC, DS_pUBLISHED
from MolesManager.moles3epb import Moles3EPB
from libs.migration.processor.migrationProcess import MigrationProcess
from testconfig import MIGRATION_DB_CONNECTION,\
    MIGRATION_DB_SCRIPT, MOLES3_DB_SCRIPT, MOLES3_DB_CONNECTION

class LoadResourceTest(TestCase):
    '''
    classdocs
    '''

    def setUp(self):
        #self._dropAllTables()
        
        migrationDB = DbManager(MIGRATION_DB_CONNECTION, MIGRATION_DB_SCRIPT)
        MigrationEPB.overrrideDBManager(migrationDB)        

        molesDB = DbManager(MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT)
        Moles3EPB.overrrideDBManager(molesDB)
        
        lr = LoadResources()
        ex = []
        ex = lr.process()
        for e in ex:
            print e
         

    def testMigrationProcess(self):
        mp = MigrationProcess()

        dataEntity = MigrationObject()
        dataEntity.doc_status = DS_pUBLISHED
        dataEntity.doc_owner = DO_BADC
        dataEntity.doc_name = 'dataent_csip.atom'
        #dataEntity.doc_name = 'DE_dcaa78b2-4008-11e0-88c9-00e081470265.atom'
        
        ex = mp.process(dataEntity)
        #ex = mp.process()
        for e in ex:
            print e
        
    def _dropAllTables(self):
        molesDB = DbManager(MOLES3_DB_CONNECTION, None)
        Moles3EPB.overrrideDBManager(molesDB)
        session = Moles3EPB.getNewMolesSession()
        f = open('dropAllTables.sql', 'r')
        for line in f.readlines():
            stripped = line.strip()
            if len(stripped) > 0:
                Moles3EPB.executeNative(line.strip())
        session.commit()
        session.close()
        molesDB = None