#!/usr/bin/python
import mysql.connector
import sys

port=3306
db1_host='10.28.239.48'
db2_host='10.180.231.50'

cnx1 = mysql.connector.connect( host=db1_host, user="myuser", password="mypwd", database="test", port=port )
cnx2 = mysql.connector.connect( host=db2_host, user="myuser", password="mypwd", database="test", port=port )

#cnx1.autocommit=True
cnx2.autocommit=True

cur1 = cnx1.cursor()
cur2 = cnx2.cursor()

cur1.execute("TRUNCATE t1")

cur2.execute("SET SESSION wsrep_sync_wait=%s",(int(sys.argv[1]),))

ins = "INSERT INTO t1 (k,j) VALUES (%s,%s)"
sel = "SELECT COUNT(*) FROM t1 WHERE k=%s"
upd = "UPDATE t1 SET v=v+1 WHERE k=%s"

for i in xrange(1,101):
    for j in xrange(0,20):
        cnx1.start_transaction()
        cur1.execute( ins, (i,j) )
        cnx1.commit()
    try:
        #print("iteration {0}".format(i))
        cur2.execute( upd, (i,) )
        if cur2.with_rows:
            ((count,),)=cur2.fetchall()
        else:
            count=cur2.rowcount
        if count != 20:
            print("ERROR: found {0} rows on iteration {1}".format(count,i))
    except mysql.connector.Error as err:
        print("ERROR: "+str(err))