def setUp(self): self.__mongoUtils = MongoSaveUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN) self.__simpleNormal = SimpleNormalization() self.__fileSourceCustom = FileSourceCustom() self.__fileBlockSource = FileBlockSource() self.__calcMongo = CalcMongo() self.__mongoStatistics = MongoStatistic(self.__mongoUtils) self.__dirPath = os.path.abspath(os.curdir)
class TestMongoStatistic(unittest.TestCase): def setUp(self): self.__mongoUtils = MongoSaveUtils(HOST, PORT, USR, PWD, DB, FC_N, FC_DN, MDN) self.__simpleNormal = SimpleNormalization() self.__fileSourceCustom = FileSourceCustom() self.__fileBlockSource = FileBlockSource() self.__calcMongo = CalcMongo() self.__mongoStatistics = MongoStatistic(self.__mongoUtils) self.__dirPath = os.path.abspath(os.curdir) def tearDown(self): try: self.__mongoUtils.deleteMergeDict() except DataNotFound: pass def testMakeDocStatisticCustomUtf8(self): filePath = os.path.join(self.__dirPath, "resources/test_mongo_index_utf8") self.__fileSourceCustom.custom = filePath self.__mongoStatistics.makeDocStatisticCustom(self.__fileBlockSource, self.__fileSourceCustom, self.__simpleNormal) self.__mongoStatistics.makeTotalStatistic() def testMakeDocStatisticCustomWin1251(self): filePath = os.path.join(self.__dirPath, "resources/test_mongo_index_win1251") self.__fileSourceCustom.custom = filePath self.__mongoStatistics.makeDocStatisticCustom(self.__fileBlockSource, self.__fileSourceCustom, self.__simpleNormal) self.__mongoStatistics.makeTotalStatistic() def testMakeDocStatisticCustomWin866(self): filePath = os.path.join(self.__dirPath, "resources/test_mongo_index_win866") self.__fileSourceCustom.custom = filePath self.__mongoStatistics.makeDocStatisticCustom(self.__fileBlockSource, self.__fileSourceCustom, self.__simpleNormal) self.__mongoStatistics.makeTotalStatistic() def testMakeDocStatistic(self): data = "Проверка проверка сохранения индекса. Check save statistic" self.__mongoStatistics.makeDocStatistic("test_source_name", 1234, data, self.__simpleNormal) self.__mongoStatistics.makeTotalStatistic() def testAddMoreStatistics(self): filePath = os.path.join(self.__dirPath, "resources/test_mongo_index_utf8") self.__fileSourceCustom.custom = filePath self.__mongoStatistics.makeDocStatisticCustom(self.__fileBlockSource, self.__fileSourceCustom, self.__simpleNormal) self.__mongoStatistics.makeTotalStatistic() self.__mongoUtils.addMoreStatistics(self.__calcMongo)