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

'''
Created on 8 Mar 2012

@author: mnagni
'''
import unittest
from testconfig import MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT
from MolesManager.moles3epb import Moles3EPB
from libs.commons_db import DbManager
from unittest import TestCase
import logging
from logging import StreamHandler
from ea_model.iso_19115_2006_metadata_corrigendum.reference_system_information.md_identifier import MD_Identifier
from test_utils import createObservationCollection
from ea_model.ceda_metadatamodel.ceda_observationcollection.ceda_observationcollection import CEDA_ObservationCollection
from ea_model.moles3_4.observationcollection.mo_observationcollection import MO_ObservationCollection
from ea_model.iso_19115_2006_metadata_corrigendum.citation_and_responsible_party_information.ci_citation import CI_Citation


class Moles3EPBTest(TestCase):

    def setUp(self):    
        Moles3EPBTest.molesDB = DbManager(MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT)
        Moles3EPB.overrrideDBManager(Moles3EPBTest.molesDB)
        self.logging = logging.getLogger('Moles3EPBTest')
        self.logging.addHandler(StreamHandler())
        self.logging.setLevel(logging.DEBUG)

    def tearDown(self):
        pass
        #self._dropAllTables()

    def runTest(self):
        self.checkExtractCollectionIdentifierByTitle()        

    def checkExtractCollectionIdentifierByTitle(self):
        session = Moles3EPB.getNewMolesSession();
        observationCollection = createObservationCollection()
        
        self.logging.info('Stores an new CEDA_ObservationCollection')
        Moles3EPB.addCedaObject(observationCollection, session)
        session.commit()

        my_identifier = Moles3EPB.extractCollectionIdentifierByTitle('test_title', session)            
        self.assertTrue(my_identifier.count() > 0, "Cannot find MD_Identifier")
        my_identifier = Moles3EPB.extractCollectionIdentifierByTitle('fake_title', session)        
        self.assertTrue(my_identifier.count() == 0, "Cannot find MD_Identifier")            
        
        self.logging.info('Deletes it')
        session.delete(observationCollection)
        session.commit()

if __name__ == "__main__":
    #import sys;sys.argv = ['', 'Test.testName']
    unittest.main()