package com.pubnub.api;

import org.json.JSONException;

import org.json.JSONObject;

import org.junit.Before;

import org.junit.Test;

import java.util.UUID;

import java.util.concurrent.CountDownLatch;

import java.util.concurrent.TimeUnit;

import static org.junit.Assert.assertEquals;

public class StateTest {

Pubnub pubnub;

String channel;

String group;

String random;

@Before

public void setUp() throws InterruptedException {

random = UUID.randomUUID().toString().substring(0, 8);

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

pubnub.setCacheBusting(false);

group = "jtest-" + random;

channel = "ch1" + random;

}

@Test

public void testStateForChannel() throws InterruptedException, PubnubException, JSONException {

JSONObject state = new JSONObject();

state.put("nickname", "jtester");

state.put("status", "online");

state.put("age", 32);

final CountDownLatch latch1 = new CountDownLatch(1);

final CountDownLatch latch2 = new CountDownLatch(1);

final CountDownLatch latch3 = new CountDownLatch(1);

final TestHelper.SimpleCallback cb1 = new TestHelper.SubscribeCallback(latch1);

final TestHelper.SimpleCallback cb2 = new TestHelper.SimpleCallback(latch2);

final TestHelper.SimpleCallback cb3 = new TestHelper.SimpleCallback(latch3);

pubnub.setHeartbeat(6);

pubnub.subscribe(channel, cb1);

latch1.await(10, TimeUnit.SECONDS);

pubnub.setState(channel, pubnub.getUUID(), state, cb2);

latch2.await(10, TimeUnit.SECONDS);

pubnub.getState(channel, pubnub.getUUID(), cb3);

latch3.await(10, TimeUnit.SECONDS);

JSONObject result = (JSONObject) cb3.getResponse();

assertEquals(state.getString("nickname"), result.getString("nickname"));

assertEquals(state.getString("status"), result.getString("status"));

assertEquals(state.getInt("age"), result.getInt("age"));

}

@Test

public void testStateForChannelGroup() throws InterruptedException, PubnubException, JSONException {

JSONObject state = new JSONObject();

state.put("nickname", "jtester");

state.put("status", "online");

state.put("age", 32);

final CountDownLatch latch1 = new CountDownLatch(1);

final CountDownLatch latch3 = new CountDownLatch(1);

final CountDownLatch latch4 = new CountDownLatch(1);

final TestHelper.SimpleCallback cb1 = new TestHelper.SimpleCallback(latch1);

final TestHelper.SimpleCallback cb3 = new TestHelper.SimpleCallback(latch3);

final TestHelper.SimpleCallback cb4 = new TestHelper.SimpleCallback(latch4);

pubnub.setHeartbeat(6);

pubnub.channelGroupAddChannel(group, channel, cb1);

latch1.await(10, TimeUnit.SECONDS);

pubnub.channelGroupSetState(group, pubnub.getUUID(), state, cb3);

latch3.await(10, TimeUnit.SECONDS);

pubnub.getState(channel, pubnub.getUUID(), cb4);

latch4.await(10, TimeUnit.SECONDS);

JSONObject result = (JSONObject) cb4.getResponse();

assertEquals(state.getString("nickname"), result.getString("nickname"));

assertEquals(state.getString("status"), result.getString("status"));

assertEquals(state.getInt("age"), result.getInt("age"));

}

}