def getClientObject(): global MONGO_CLIENT if MONGO_CLIENT is None: MONGO_CLIENT = MongoClient(getHost(), getPort()) # print "getClientObject: {0}".format(hex(id(MONGO_CLIENT))) return MONGO_CLIENT
def testHost(self): self.assertEqual(HOST, config_reader.getHost())
# Collections TAGS = 'tags' COLLECTION = 'services' NAME = 'name' CONFIG = 'config' LOG_SIZE = 'log_size' OWNERID = 'owner_id' ID = '_id' LOG = 'log' BC = 'bc' METADATA = 'metadata' # db initialisation MONGO_CLIENT = None # MongoClient(getHost(), getPort()) db = MongoClient(getHost(), getPort())[getDbName()] # keys USER_ID = 'user_id' DATE = 'date' MESSAGE = 'message' LEVEL = 'level' ERROR_CODE = 'error_code' SERVICE = 'service' COLLECTION = 'services' CHANNELS_COLLECTION = 'channels' POINTS_COLLECTION = 'points' JSON = 'json' ACL = 'acl' OWNER_GROUP = 'owner_group' POINTS_COLLECTION = 'points' LOCATION = 'location'
#!/usr/bin/env python # -*- coding: utf-8 -*- from unittest import TestCase from pymongo import MongoClient from db_model import getAllChannelIds import config_reader SERVICE_NAME = 'testservice' TEST_ID = u'556721a52a2e7febd2744200' db = MongoClient(config_reader.getHost(), config_reader.getPort())[SERVICE_NAME] CHANNELS_COLLECTION = 'channels' class TestgetAllChannelIds(TestCase): def testgetAllChannelIds_id(self): result = getAllChannelIds(SERVICE_NAME) self.assertEqual(result[0], TEST_ID) def testgetAllChannelIds_count(self): count_channels = db[CHANNELS_COLLECTION].count() result = getAllChannelIds(SERVICE_NAME) self.assertEqual(len(result), count_channels)
import config_reader TEST_DB = 'master_db_template' DB_TAG = '--dbName ' CONF_TAG = '--config ' CONFIG_NAME = 'config' PATH_DB = 'python scripts/db/setupMasterDbTemplate.py ' + DB_TAG + TEST_DB PATH_CONF = \ 'python scripts/db/setupMasterDbTemplate.py ' + CONF_TAG + CONFIG_NAME TEST_PATH = os.getcwd() os.chdir('../../') os.system(PATH_CONF) os.chdir(TEST_PATH) db = MongoClient(config_reader.getHost(), config_reader.getPort())[TEST_DB] MYCOLLECTION = 'testdump_forimport' COUNT = 1 class TestImportDb(TestCase): def testMyImport_Db_Tag(self): os.chdir('../../') os.system('./scripts/db/drop_test_db.sh') os.system(PATH_DB) count_mycoll = db[MYCOLLECTION].count() self.assertEqual(count_mycoll, COUNT) os.system('./scripts/db/import_test_db.sh') os.chdir(TEST_PATH) def testMyImport_Conf_Tag(self):