def test_CSV_converter_to_UTF8(self): #This call the to_utf8 function, and saves a file to tmp/ result = libs.to_utf8.main(self.file) self.assertTrue(os.path.isfile(result)) csv_reader = csv.reader(open(result, 'rb'),delimiter=';', quotechar='"') self.utf8_file = obp_import_post_bank.get_info_from_row(csv_reader.next()) for element in self.utf8_file: if type(element) is unicode: self.values_eq_unicode = 1 else: self.values_eq_unicode = 0 self.assertEqual(self.values_eq_unicode,1) os.remove(result) self.assertFalse(os.path.isfile(result)) result = os.getcwd() self.assertEqual(result,self.here)
def test_Import_CSV(self): result = libs.to_utf8.main(self.file) self.assertTrue(os.path.isfile(result)) csv_reader = csv.reader(open(result, 'rb'),delimiter=';', quotechar='"') self.utf8_file = obp_import_post_bank.get_info_from_row(csv_reader.next()) for element in self.utf8_file: if type(element) is unicode: self.values_eq_unicode = 1 else: self.values_eq_unicode = 0 self.assertEqual(self.values_eq_unicode,1) self.connection = Connection('obp_mongod', 27017) self.mongo_db = self.connection.test_obp_import_db self.collection = self.mongo_db.test_obp_import_db.insert(son.SON(self.utf8_file)) self.find_in_mongo = self.mongo_db.test_obp_import_db.find_one(self.collection) self.find_should = { u'obp_transaction_date_complete': u'07.11.2011', u'obp_transaction_new_balance': u'5.314', u'obp_transaction_comment2': u'PETRA PFIFFIG', u'obp_transaction_amount': u'-328', u'obp_transaction_comment1': u'111111/1000000000/37050198 Finanzkasse 3991234 Steuernummer 00703434', u'obp_transaction_data_blob': u'Finanzkasse K\xf6ln-S\xfcd', u'obp_transaction_transaction_type_de': u'\xdcberweisung', u'_id': "ObjectId('4eba776731533f464d000000')", u'obp_transaction_date_start': u'07.11.2011' } # This is a new test from Pyhon2.7, it will sort the input of the # test and then comapre them like assertEqual # LINK: # http://docs.python.org/library/unittest.html#unittest.TestCase.assertItemsEqual self.assertItemsEqual(self.find_should,self.find_in_mongo) os.remove(result) self.assertFalse(os.path.isfile(result))