class TestDataSourceWorkerFS(unittest.TestCase):
def setUp(self):
self.__dirPath = os.path.abspath(os.curdir)
firstPath = os.path.join(self.__dirPath, "resources/first")
secondPath = os.path.join(self.__dirPath, "resources/second")
self.__mongoSaveUtils = MongoSaveRetrievalUtils(
HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
self.__mongoReadUtils = MongoReadUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
self.__simN = SimpleNormalization()
self.__simNamesN = FileNameNormalization()
self.__ms = StatisticFactory().createStatistic(MONGO_TYPE,
self.__mongoSaveUtils)
self.__scc = FSSourceCustomCallback()
self.__rnFS = ReaderNameFS([firstPath, secondPath])
def tearDown(self):
try:
self.__mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testCreateStatistics(self):
fsWorker = DataSourceWorkerFS()
fsWorker.createStatistics(self.__ms, self.__rnFS, self.__simN,
self.__scc)
def testCreateStatisticsException(self):
fsWorker = DataSourceWorkerFS()
self.assertRaises(ParamError, fsWorker.createStatistics, None, 2, 3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, 1, 2, 3, 4)
self.assertRaises(ParamError, fsWorker.createStatistics, self.__ms,
None, 3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, self.__ms, 2,
3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, self.__ms,
self.__rnFS, self.__simN, 5)
def testCreateSourceNameIndex(self):
fsWorker = DataSourceWorkerFS()
fsWorker.createStatistics(self.__ms, self.__rnFS, self.__simN,
self.__scc)
fsWorker.createSourceNameIndex(self.__ms, self.__mongoReadUtils,
self.__mongoSaveUtils, self.__simNamesN)
def testCreateSourceNameIndexException(self):
fsWorker = DataSourceWorkerFS()
csn_m = fsWorker.createSourceNameIndex
self.assertRaises(ParamError, csn_m, None, 2, 3, 4)
self.assertRaises(TypeError, csn_m, 1, 2, 3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, None, 3, 4)
self.assertRaises(TypeError, csn_m, self.__ms, 2, 3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, self.__mongoReadUtils,
None, 4)
self.assertRaises(TypeError, csn_m, self.__ms, self.__mongoReadUtils,
3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, self.__mongoReadUtils,
self.__mongoSaveUtils, None)
self.assertRaises(TypeError, csn_m, self.__ms, self.__mongoReadUtils,
self.__mongoSaveUtils, 1)
def setUp(self):
self.__dirPath = os.path.abspath(os.curdir)
firstPath = os.path.join(self.__dirPath, "resources/first")
secondPath = os.path.join(self.__dirPath, "resources/second")
self.__mongoSaveUtils = MongoSaveRetrievalUtils(
HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
self.__mongoReadUtils = MongoReadUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
self.__simN = SimpleNormalization()
self.__simNamesN = FileNameNormalization()
self.__ms = StatisticFactory().createStatistic(MONGO_TYPE,
self.__mongoSaveUtils)
self.__scc = FSSourceCustomCallback()
self.__rnFS = ReaderNameFS([firstPath, secondPath])
class TestDataSourceWorkerFS(unittest.TestCase):
def setUp(self):
self.__dirPath = os.path.abspath(os.curdir)
firstPath = os.path.join(self.__dirPath, "resources/first")
secondPath = os.path.join(self.__dirPath, "resources/second")
self.__mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
self.__mongoReadUtils = MongoReadUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
self.__simN = SimpleNormalization()
self.__simNamesN = FileNameNormalization()
self.__ms = StatisticFactory().createStatistic(MONGO_TYPE, self.__mongoSaveUtils)
self.__scc = FSSourceCustomCallback()
self.__rnFS = ReaderNameFS([firstPath, secondPath])
def tearDown(self):
try:
self.__mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testCreateStatistics(self):
fsWorker = DataSourceWorkerFS()
fsWorker.createStatistics(self.__ms, self.__rnFS, self.__simN, self.__scc)
def testCreateStatisticsException(self):
fsWorker = DataSourceWorkerFS()
self.assertRaises(ParamError, fsWorker.createStatistics, None, 2, 3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, 1, 2, 3, 4)
self.assertRaises(ParamError, fsWorker.createStatistics, self.__ms, None, 3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, self.__ms, 2, 3, 4)
self.assertRaises(TypeError, fsWorker.createStatistics, self.__ms, self.__rnFS, self.__simN, 5)
def testCreateSourceNameIndex(self):
fsWorker = DataSourceWorkerFS()
fsWorker.createStatistics(self.__ms, self.__rnFS, self.__simN, self.__scc)
fsWorker.createSourceNameIndex(self.__ms, self.__mongoReadUtils, self.__mongoSaveUtils, self.__simNamesN)
def testCreateSourceNameIndexException(self):
fsWorker = DataSourceWorkerFS()
csn_m = fsWorker.createSourceNameIndex
self.assertRaises(ParamError, csn_m, None, 2, 3, 4)
self.assertRaises(TypeError, csn_m, 1, 2, 3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, None, 3, 4)
self.assertRaises(TypeError, csn_m, self.__ms, 2, 3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, self.__mongoReadUtils, None, 4)
self.assertRaises(TypeError, csn_m, self.__ms, self.__mongoReadUtils, 3, 4)
self.assertRaises(ParamError, csn_m, self.__ms, self.__mongoReadUtils, self.__mongoSaveUtils, None)
self.assertRaises(TypeError, csn_m, self.__ms, self.__mongoReadUtils, self.__mongoSaveUtils, 1)
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {"the": 1, "test": 2},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"], "testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {"some": 20, "gg": 50},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {"some": 20, "gg": 50, "n1": 80},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"], "some doc 2~")
def setUp(self):
self.__dirPath = os.path.abspath(os.curdir)
firstPath = os.path.join(self.__dirPath, "resources/first")
secondPath = os.path.join(self.__dirPath, "resources/second")
self.__mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
self.__mongoReadUtils = MongoReadUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
self.__simN = SimpleNormalization()
self.__simNamesN = FileNameNormalization()
self.__ms = StatisticFactory().createStatistic(MONGO_TYPE, self.__mongoSaveUtils)
self.__scc = FSSourceCustomCallback()
self.__rnFS = ReaderNameFS([firstPath, secondPath])
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {
"the": 1,
"test": 2
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"],
"testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {
"some": 20,
"gg": 50
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {
"some": 20,
"gg": 50,
"n1": 80
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"],
"some doc 2~")
def setUp(self):
self.__dirPath = os.path.abspath(os.curdir)
firstPath = os.path.join(self.__dirPath, "resources/first")
secondPath = os.path.join(self.__dirPath, "resources/second")
self.__mongoUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN, MDN)
self.__mongoUtilsTypeError = MongoSaveUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN, MDN)
self.__mongoReadUtils = MongoReadUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
self.__smN = SimpleNormalization()
self.__scc = FSSourceCustomCallback()
self.__rFS = ReaderNameFS([firstPath, secondPath])
self.__fsWorker = DataSourceWorkerFS()
class TestMongoSaveRetrievalUtils(unittest.TestCase):
def setUp(self):
self.__mongoUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
def tearDown(self):
try:
self.__mongoUtils.deleteMergeDict()
except DataNotFound:
pass
def testSaveFilename(self):
retID = self.__mongoUtils.saveDict("testing name", "utf-8", 1234, {"the": 1, "test": 2},
"utf-8", datetime.datetime.now())
self.__mongoUtils.saveFilename(retID, ["one", "test", "run"], "testing name")
def testSaveFilenameException(self):
self.assertRaises(ParamError, self.__mongoUtils.saveFilename, None, 2, 3)
self.assertRaises(ParamError, self.__mongoUtils.saveFilename, 1, None, 3)
self.assertRaises(ParamError, self.__mongoUtils.saveFilename, 1, 2, None)
self.assertRaises(TypeError, self.__mongoUtils.saveFilename, 1, 2, 3)
self.__mongoUtils.deleteMergeDict()
self.assertRaises(DataNotFound, self.__mongoUtils.saveFilename, 1, ["one"], "one")
class TestMongoSearchRetrievalUtils(unittest.TestCase):
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {"the": 1, "test": 2},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"], "testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {"some": 20, "gg": 50},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {"some": 20, "gg": 50, "n1": 80},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"], "some doc 2~")
def tearDown(self):
try:
self.mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testSearchFilename(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcn"}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchFilename([TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
self.assertIsNotNone(findObject, "findObject == None")
mongoSearch.removeSearchData(findObject)
def testSearchFilenameException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcn"}
mdID = self.mongoSaveUtils.getMergeDictID()
self.assertRaises(ParamError, mongoSearch.searchFilename, None, customDict, mdID)
self.assertRaises(ParamError, mongoSearch.searchFilename, [TYPE_Q_LOGIC, [["doc"], ["doc"]]], None, mdID)
self.assertRaises(ParamError, mongoSearch.searchFilename, [TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, None)
self.assertRaises(TypeError, mongoSearch.searchFilename, "string", customDict, mdID)
self.assertRaises(TypeError, mongoSearch.searchFilename, [TYPE_Q_LOGIC, [["doc"], ["doc"]]], "string", mdID)
self.assertRaises(KeyError, mongoSearch.searchFilename, [TYPE_Q_LOGIC, [["doc"], ["doc"]]], {"aa": 10}, mdID)
def testSearchData(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcd"}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchData([None, ["the"]], customDict, mdID)
self.assertIsNotNone(findObject, "findObject == None")
mongoSearch.removeSearchData(findObject)
def testSearchDataException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcd"}
mdID = self.mongoSaveUtils.getMergeDictID()
self.assertRaises(ParamError, mongoSearch.searchData, None, customDict, mdID)
self.assertRaises(ParamError, mongoSearch.searchData, [None, [["doc"], ["doc"]]], None, mdID)
self.assertRaises(ParamError, mongoSearch.searchData, [None, [["doc"], ["doc"]]], customDict, None)
self.assertRaises(TypeError, mongoSearch.searchData, "string", customDict, mdID)
self.assertRaises(TypeError, mongoSearch.searchData, [None, [["doc"], ["doc"]]], "string", mdID)
self.assertRaises(KeyError, mongoSearch.searchData, [None, [["doc"], ["doc"]]], {"aa": 10}, mdID)
def testGetSearchData(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcn"}
customDictData = {"is_lazy": False}
mdID = self.mongoSaveUtils.getMergeDictID()
# поиск по именам источников
# логический запрос
findObject = mongoSearch.searchFilename([TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
self.assertEqual(int(doc["value"]["count"]), 2, "count != 2")
# запрос точного совпадения
customDictData["is_lazy"] = False
findObject = mongoSearch.searchFilename([TYPE_Q_EXACT, "some doc~"], customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
mongoSearch.removeSearchData(findObject)
# поиск по данным из источника
customDict["result_cn"] = "rcd"
customDictData["is_lazy"] = False
findObject = mongoSearch.searchData([None, ["the"]], customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
self.assertEqual(int(doc["value"]["count"]), 1, "count != 1")
mongoSearch.removeSearchData(findObject)
def testGetSearchDataException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
customDict = {"result_cn": "rcn"}
customDictData = {"is_lazy": False}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchFilename([TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
self.assertRaises(ParamError, mongoSearch.getSearchData, None, customDictData)
self.assertRaises(ParamError, mongoSearch.getSearchData, findObject, None)
self.assertRaises(TypeError, mongoSearch.getSearchData, "string", customDictData)
self.assertRaises(TypeError, mongoSearch.getSearchData, findObject, "string")
self.assertRaises(KeyError, mongoSearch.getSearchData, findObject, {"bb": 40})
mongoSearch.removeSearchData(findObject)
class TestMongoSearchRetrievalUtils(unittest.TestCase):
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {
"the": 1,
"test": 2
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"],
"testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {
"some": 20,
"gg": 50
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {
"some": 20,
"gg": 50,
"n1": 80
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"],
"some doc 2~")
def tearDown(self):
try:
self.mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testSearchFilename(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcn"}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchFilename(
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
self.assertIsNotNone(findObject, "findObject == None")
mongoSearch.removeSearchData(findObject)
def testSearchFilenameException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcn"}
mdID = self.mongoSaveUtils.getMergeDictID()
self.assertRaises(ParamError, mongoSearch.searchFilename, None,
customDict, mdID)
self.assertRaises(ParamError, mongoSearch.searchFilename,
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], None, mdID)
self.assertRaises(ParamError, mongoSearch.searchFilename,
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, None)
self.assertRaises(TypeError, mongoSearch.searchFilename, "string",
customDict, mdID)
self.assertRaises(TypeError, mongoSearch.searchFilename,
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], "string", mdID)
self.assertRaises(KeyError, mongoSearch.searchFilename,
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], {"aa": 10}, mdID)
def testSearchData(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcd"}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchData([None, ["the"]], customDict, mdID)
self.assertIsNotNone(findObject, "findObject == None")
mongoSearch.removeSearchData(findObject)
def testSearchDataException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcd"}
mdID = self.mongoSaveUtils.getMergeDictID()
self.assertRaises(ParamError, mongoSearch.searchData, None, customDict,
mdID)
self.assertRaises(ParamError, mongoSearch.searchData,
[None, [["doc"], ["doc"]]], None, mdID)
self.assertRaises(ParamError, mongoSearch.searchData,
[None, [["doc"], ["doc"]]], customDict, None)
self.assertRaises(TypeError, mongoSearch.searchData, "string",
customDict, mdID)
self.assertRaises(TypeError, mongoSearch.searchData,
[None, [["doc"], ["doc"]]], "string", mdID)
self.assertRaises(KeyError, mongoSearch.searchData,
[None, [["doc"], ["doc"]]], {"aa": 10}, mdID)
def testGetSearchData(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcn"}
customDictData = {"is_lazy": False}
mdID = self.mongoSaveUtils.getMergeDictID()
# поиск по именам источников
# логический запрос
findObject = mongoSearch.searchFilename(
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
self.assertEqual(int(doc["value"]["count"]), 2, "count != 2")
# запрос точного совпадения
customDictData["is_lazy"] = False
findObject = mongoSearch.searchFilename([TYPE_Q_EXACT, "some doc~"],
customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
mongoSearch.removeSearchData(findObject)
# поиск по данным из источника
customDict["result_cn"] = "rcd"
customDictData["is_lazy"] = False
findObject = mongoSearch.searchData([None, ["the"]], customDict, mdID)
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
customDictData["is_lazy"] = True
result = mongoSearch.getSearchData(findObject, customDictData)
for doc in result:
self.assertIsNotNone(doc, "map-reduce doc == None")
self.assertEqual(int(doc["value"]["count"]), 1, "count != 1")
mongoSearch.removeSearchData(findObject)
def testGetSearchDataException(self):
mongoSearch = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N,
FC_DN)
customDict = {"result_cn": "rcn"}
customDictData = {"is_lazy": False}
mdID = self.mongoSaveUtils.getMergeDictID()
findObject = mongoSearch.searchFilename(
[TYPE_Q_LOGIC, [["doc"], ["doc"]]], customDict, mdID)
self.assertRaises(ParamError, mongoSearch.getSearchData, None,
customDictData)
self.assertRaises(ParamError, mongoSearch.getSearchData, findObject,
None)
self.assertRaises(TypeError, mongoSearch.getSearchData, "string",
customDictData)
self.assertRaises(TypeError, mongoSearch.getSearchData, findObject,
"string")
self.assertRaises(KeyError, mongoSearch.getSearchData, findObject,
{"bb": 40})
mongoSearch.removeSearchData(findObject)
def setUp(self):
self.__mongoUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
class TestMongoSearch(unittest.TestCase):
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {"the": 1, "test": 2},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"], "testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {"some": 20, "gg": 50},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {"some": 20, "gg": 50, "n1": 80},
"utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"], "some doc 2~")
def tearDown(self):
try:
self.mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testInitException(self):
self.assertRaises(ParamError, MongoSearch, None)
self.assertRaises(TypeError, MongoSearch, "string")
self.assertRaises(TypeError, MongoSearch, MongoReadUtils, HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
def testSearchNames(self):
mongoSearchUtils = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
cd = {"result_cn": "rcn"} # customDict
cdd = {"is_lazy": False} # customDictData
mdID = self.mongoSaveUtils.getMergeDictID()
mongoSearch = MongoSearch(mongoSearchUtils)
self.assertIsNotNone(mongoSearch.searchNames([TYPE_Q_LOGIC, [["doc"], ["doc"]]], cd, cdd, mdID))
mongoSearch.removeFindObject()
def testSearchData(self):
mongoSearchUtils = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN)
cd = {"result_cn": "rcd"} # customDict
cdd = {"is_lazy": False} # customDictData
mdID = self.mongoSaveUtils.getMergeDictID()
mongoSearch = MongoSearch(mongoSearchUtils)
self.assertIsNotNone(mongoSearch.searchData([None, ["the"]], cd, cdd, mdID))
mongoSearch.removeFindObject()
class TestMongoSearch(unittest.TestCase):
def setUp(self):
self.mongoSaveUtils = MongoSaveRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN, MDN)
retID = self.mongoSaveUtils.saveDict("testing name", "utf-8", 1234, {
"the": 1,
"test": 2
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["testing", "name", "other"],
"testing name~")
retID = self.mongoSaveUtils.saveDict("some doc", "utf-8", 1234, {
"some": 20,
"gg": 50
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc"], "some doc~")
retID = self.mongoSaveUtils.saveDict("some doc 2", "utf-8", 1234, {
"some": 20,
"gg": 50,
"n1": 80
}, "utf-8", datetime.datetime.now())
self.mongoSaveUtils.saveFilename(retID, ["some", "doc", "2"],
"some doc 2~")
def tearDown(self):
try:
self.mongoSaveUtils.deleteMergeDict()
except DataNotFound:
pass
def testInitException(self):
self.assertRaises(ParamError, MongoSearch, None)
self.assertRaises(TypeError, MongoSearch, "string")
self.assertRaises(TypeError, MongoSearch, MongoReadUtils, HOST, PORT,
USR, PWD, DB, FC_N, FC_DN)
def testSearchNames(self):
mongoSearchUtils = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN)
cd = {"result_cn": "rcn"} # customDict
cdd = {"is_lazy": False} # customDictData
mdID = self.mongoSaveUtils.getMergeDictID()
mongoSearch = MongoSearch(mongoSearchUtils)
self.assertIsNotNone(
mongoSearch.searchNames([TYPE_Q_LOGIC, [["doc"], ["doc"]]], cd,
cdd, mdID))
mongoSearch.removeFindObject()
def testSearchData(self):
mongoSearchUtils = MongoSearchRetrievalUtils(HOST, PORT, USR, PWD, DB,
FC_N, FC_DN)
cd = {"result_cn": "rcd"} # customDict
cdd = {"is_lazy": False} # customDictData
mdID = self.mongoSaveUtils.getMergeDictID()
mongoSearch = MongoSearch(mongoSearchUtils)
self.assertIsNotNone(
mongoSearch.searchData([None, ["the"]], cd, cdd, mdID))
mongoSearch.removeFindObject()
