twoway.py

from uprclient import UprClient
from mcdclient import McdClient
from constants import *
import time
import logging
import pprint

def get_resp(op, text):
    print text
    print op
    response = op.next_response()
    print 'Response: %s\n' % response
    return response


def two_way_demo():
    consumer = UprClient('127.0.0.1', 12000)
    producer = UprClient('127.0.0.1', 12002)
    consumer.set_proxy(producer)
    producer.set_proxy(consumer)

    mcd_consumer = McdClient('127.0.0.1', 12000)
    mcd_producer = McdClient('127.0.0.1', 12002)

    response = get_resp(consumer.open_consumer("mystream"), 'Sending open connection (consumer)')
    assert response['status'] == SUCCESS

    response = get_resp(producer.open_producer("mystream"), 'Sending open connection (producer)')
    assert response['status'] == SUCCESS

    response = get_resp(consumer.add_stream(0, 0), 'Sending add stream request for vbucket 0')
    assert response['status'] == SUCCESS

    op = mcd_consumer.stats('upr')
    response = op.next_response()
    print response['value']
    assert response['value']['ep_upr_count'] == '1'

    op = mcd_producer.stats('upr')
    response = op.next_response()
    print response['value']
    assert response['value']['ep_upr_count'] == '1'

    get_resp(mcd_producer.set("key", "blah", 0, 0, 0), 'Create mutation on producer')

    response = get_resp(consumer.close_stream(0), 'Sending close stream request for vbucket 0')
    assert response['status'] == SUCCESS

    time.sleep(10)

    consumer.shutdown()
    producer.shutdown()
    mcd_consumer.shutdown()
    mcd_producer.shutdown()


if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG)
    two_way_demo()