def mongo_check_data(name_collection, name_test): # CONNECT AND GET COLLECTION name_database = "arkis" mongo = Mongo() conn = mongo.create_connexion(host_resilience_mongo, port_resilience_mongo) database = mongo.create_database(conn, name_database) mongo.create_collection(database, name_collection) # GET COUNT AND GET REAL COUNT try: r = mongo.get_all_data(database[name_collection]) real_count = r.count() count = get_count(name_test) res = {'real_count': real_count, 'count': count} mongo.write_results(res, name_test) print real_count print count except Exception, e: print e print "Problem geting data"