diff options
-rw-r--r-- | test.py | 247 |
1 files changed, 0 insertions, 247 deletions
diff --git a/test.py b/test.py deleted file mode 100644 index a408cc4..0000000 --- a/test.py +++ /dev/null @@ -1,247 +0,0 @@ -# -*- coding: utf-8 -*- - -import datetime -import time -from kittystore.kittysastore import KittySAStore -from kittystore.mongostore import KittyMGStore - -# Define global constant - -TABLE = 'devel' -REP = 1 -URL = 'postgres://mm3:mm3@localhost/mm3' -DB_STORE = KittySAStore(URL) -MG_STORE = KittyMGStore(host='localhost', port=27017) - -START = datetime.datetime(2012, 3, 1) -END = datetime.datetime(2012, 3, 30) - -def output(name, post, mongo): - stream = open(name, 'w') - stream.write('postgresql\tMongodb') - for i in range(0, len(post)): - stream.write(str(post[i]) + '\t' + str(mongo[i])) - stream.close() - -def get_email(rep): - print 'get_email' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = DB_STORE.get_email(TABLE, '3D97B04F.7090405@terra.com.br') - post.append(time.time() - t0) - for i in range(0,rep): - t0 = time.time() - res_mg = MG_STORE.get_email(TABLE, '3D97B04F.7090405@terra.com.br') - mongo.append(time.time() - t0) - output('get_email', post, mongo) - if res_mg['Subject'] != res_pg.subject and res_mg['Date'] != res_pg.date: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def get_archives_range(rep): - print 'get_archives_range' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len(DB_STORE.get_archives(TABLE, START, END)) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len(MG_STORE.get_archives(TABLE, START, END)) - mongo.append(time.time() - t0) - output('get_thread_length', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def first_email_in_archives_range(rep): - print 'first_email_in_archives_range' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = DB_STORE.get_archives(TABLE, START, END)[0] - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = MG_STORE.get_archives(TABLE, START, END)[0] - mongo.append(time.time() - t0) - output('first_email_in_archives_range', post, mongo) - if res_mg['Subject'] != res_pg.subject and res_mg['Date'] != res_pg.date: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def get_thread_length(rep): - print 'get_thread_length' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = DB_STORE.get_thread_length(TABLE, - '4FCWUV6BCP3A5PASNFX6L5JOAE4GJ7F2') - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = MG_STORE.get_thread_length(TABLE, - '4FCWUV6BCP3A5PASNFX6L5JOAE4GJ7F2') - mongo.append(time.time() - t0) - output('get_thread_length', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def get_thread_participants(rep): - print 'get_thread_participants' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len(DB_STORE.get_thread_participants(TABLE, - '4FCWUV6BCP3A5PASNFX6L5JOAE4GJ7F2')) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len(MG_STORE.get_thread_participants(TABLE, - '4FCWUV6BCP3A5PASNFX6L5JOAE4GJ7F2')) - mongo.append(time.time() - t0) - output('get_thread_participants', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def get_archives_length(rep): - print 'get_archives_length' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = DB_STORE.get_archives_length(TABLE) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = MG_STORE.get_archives_length(TABLE) - mongo.append(time.time() - t0) - output('get_archives_length', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def search_subject(rep): - print 'search_subject' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len(DB_STORE.search_subject(TABLE, 'rawhid')) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len(MG_STORE.search_subject(TABLE, 'rawhid')) - mongo.append(time.time() - t0) - output('search_subject', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def search_content(rep): - print 'search_content' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len(DB_STORE.search_content(TABLE, 'rawhid')) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len(MG_STORE.search_content(TABLE, 'rawhid')) - mongo.append(time.time() - t0) - output('search_content', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def search_content_subject(rep): - print 'search_content_subject' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len( DB_STORE.search_content_subject(TABLE, 'rawhid')) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len( MG_STORE.search_content_subject(TABLE, 'rawhid')) - mongo.append(time.time() - t0) - output('search_content_subject', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def search_sender(rep): - print 'search_sender' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = len(DB_STORE.search_sender(TABLE, 'pingou')) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = len(MG_STORE.search_sender(TABLE, 'pingou')) - mongo.append(time.time() - t0) - output('search_sender', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - -def get_list_size(rep): - print 'get_list_size' - post = [] - mongo = [] - for i in range(0, rep): - t0 = time.time() - res_pg = res_pg = DB_STORE.get_list_size(TABLE) - post.append(time.time() - t0) - for i in range(0, rep): - t0 = time.time() - res_mg = MG_STORE.get_list_size(TABLE) - mongo.append(time.time() - t0) - output('get_list_size', post, mongo) - if res_mg != res_pg: - print '** Results differs' - print 'MG: %s' % res_mg - print 'PG: %s\n' % res_pg - - -if __name__ == '__main__': - t_start = time.time() - get_email(REP) - get_archives_range(REP) - first_email_in_archives_range(REP) - get_thread_length(REP) - get_thread_participants(REP) - get_archives_length(REP) - - search_subject(REP) - search_content(REP) - search_content_subject(REP) - - get_list_size(REP) - print "Ran for %s seconds" % (time.time() - t_start) - - - - |