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

'''
Created on 7 Mar 2012

@author: mnagni
'''
from unittest import TestCase
from libs.commons_db import DbManager
from libs.migration.MigrationEPB import MigrationEPBFactory
from libs.migration.processor.loadResources import LoadResources
from libs.migration.db.classes import MigrationObject
from libs.migration.processor.commons import DO_BADC, DS_pUBLISHED, DS_PUBLISHED
from MolesManager.moles3epb import Moles3EPBFactory
from libs.migration.processor.migrationProcess import MigrationProcess
from testconfig import MIGRATION_DB_CONNECTION,\
    MIGRATION_DB_SCRIPT, MOLES3_DB_SCRIPT, MOLES3_DB_CONNECTION, INFO_DB_CONNECTION 
from libs.migration.InfodbEPB import InfodbEPBFactory
from MolesManager.settings import EVENTS_DB
from libs.migration.client import EPBRepo

class LoadResourceTest(TestCase):
    '''
    classdocs
    '''
    
    def setUp(self):
        migrationDB = DbManager(MIGRATION_DB_CONNECTION, MIGRATION_DB_SCRIPT, sql_echo=True)
        #migrationDB = DbManager(MIGRATION_DB_CONNECTION, MIGRATION_DB_SCRIPT)
        #MigrationEPB.overrrideDBManager(migrationDB)      

        molesDB = DbManager(MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT, sql_echo=True, session_event_manager=EVENTS_DB)
        #molesDB = DbManager(MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT, session_event_manager=EVENTS_DB)         
        
        
        #infoDB = DbManager(INFO_DB_CONNECTION, sql_echo=True)
        infoDB = DbManager(INFO_DB_CONNECTION)        
        infoFactory = InfodbEPBFactory(infoDB)
        
        moles3Factory = Moles3EPBFactory(molesDB)
        migrationFactory = MigrationEPBFactory(migrationDB)
        
        EPBRepo.moles3EPB = moles3Factory.createEPB() 
        EPBRepo.migrationEPB = migrationFactory.createEPB()
        EPBRepo.infodbEPB = infoFactory.createEPB()
        '''
        lr = LoadResources()
        ex = lr.process()
        for e in ex:
            print e
        '''
    def testMigrationProcess(self):        
        mp = MigrationProcess()

        dataEntity = MigrationObject()
        dataEntity.doc_owner = DO_BADC
                
        #dataEntity.doc_status = DS_pUBLISHED
        #dataEntity.doc_name = 'dataent_csip.atom'
        
        #Has a DOI
        #dataEntity.doc_status = DS_PUBLISHED
        #dataEntity.doc_name = 'DE_095e8da2-cf02-11e0-8b7a-00e081470265.atom'
        
        #Has 3 DOI
        dataEntity.doc_status = DS_pUBLISHED
        dataEntity.doc_name = 'dataent_chobs.atom'        
                
        
        ex = mp.process(dataEntity)
        #ex = []
        #ex = mp.process()
        for e in ex:
            print e
        #self._dropAllTables()

    """        
    def _dropAllTables(self):
        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)
        session.commit()
        session.close()        
    """