import sys

sys.path.append('../../common')
sys.path.append('..')
sys.path.append('../common')
sys.path.append('.')

from PubnubUnitTest import Suite
from Pubnub import Pubnub

pubnub = Pubnub("demo","demo")

tests_count = 1 + 2
test_suite = Suite(pubnub,tests_count)

tests = []


def test_publish():
	name = "Publish Test"
	def success(r):
		test_suite.test(r[0] == 1, name)

	def fail(e):
		test_suite.test(False, msg , e)


	pubnub.publish({
		'channel' : 'hello',
		'message' : 'hi',
		'callback' : success,
		'error' : fail
	})
tests.append(test_publish)


def test_subscribe_publish():
	channel = "hello"
	name = "Subscribe Publish Test"
	publish_msg = "This is Pubnub Python-Twisted"
	def connect():
		def success(r):
			test_suite.test(r[0] == 1, name, "publish success")

		def fail(e):
			test_suite.test(False, name , "Publish Failed", e)

		pubnub.publish({
			'channel' : channel,
			'message' : publish_msg,
			'callback' : success,
			'error' : fail
		})

	def callback(r):
		test_suite.test(r == publish_msg, name, "message received")

	pubnub.subscribe({
		'channel' : channel,
		'callback' : callback,
		'connect' : connect
	})
tests.append(test_subscribe_publish)





for t in tests:
	t()

pubnub.start()