source: ndgCommon/trunk/ndg/common/unittests/testutils.py @ 4991

"""
Utilities to use in the various test cases

NERC Data Grid Project
"""
__author__ = "C Byrom - Tessella"
__date__ = "16/12/08"
__copyright__ = "(C) 2008 STFC & NERC"
__license__ = \
"""This software may be distributed under the terms of the Q Public 
License, version 1.0 or later."""
__contact__ = "b.n.lawrence@rl.ac.uk"
__revision__ = '$Id: $'

from ndg.common.src.models.vocabtermdata import VocabTermData as VTD
from ndg.common.src.clients.xmldb.eXist.atomclient import AtomClient as ac
import ndg.common.src.clients.xmldb.eXist.dbconstants as dc
from ndg.common.src.models.Atom import Atom
import ndg.common.unittests.testconstants as tc
import os

class testUtils(object):
    def __init__(self, confFile):
        self.host = tc.EXIST_DB
        self.ac = ac(dbHostName = self.host, configFileName = confFile, setUpDB = True)
    
    def createAtom(self, content):
        atom = Atom(VTD.GRANULE_TERM)
        atom.fromString(content)
        return atom
    
    def createAtomInEXist(self, content):
        atom = self.createAtom(content)
        return self.ac.createAtom(atom)
    
    def deleteDoc(self, path):
        self.ac.deleteDoc(path)
        
    def createOldMolesDoc(self):
        '''
        Create an old moles1.0 doc
        '''
        # firstly, load a moles do to eXist
        f = open(tc.MOLES_FILE_NAME, 'r')
        xml = f.read()
        f.close()
        fileName = f.name.split(os.sep)[-1]
        path = dc.MOLES_COLLECTION_PATH + tc.GRANULITE_PROVIDER_ID
        self.ac.createDoc(xml, path, fileName)
        return path + '/' + fileName