コード例 #1
0
ファイル: BaseTest.java プロジェクト: juansarabia/atmosphere
  @Test(timeOut = 60000, enabled = true)
  public void onResumeWithTimerTest() {
    logger.info("{}: running test: onResumeWithTimerTest", getClass().getSimpleName());

    final AtomicBoolean onResume = new AtomicBoolean();
    final CountDownLatch eventReceived = new CountDownLatch(1);
    final CountDownLatch suspended = new CountDownLatch(1);
    final AtomicReference<AtmosphereResource> resource = new AtomicReference<AtmosphereResource>();

    final CountDownLatch latch = new CountDownLatch(1);
    atmoServlet
        .framework()
        .addAtmosphereHandler(
            ROOT,
            new AbstractHttpAtmosphereHandler() {

              public void onRequest(final AtmosphereResource r) throws IOException {
                r.addEventListener(
                    new WebSocketEventListenerAdapter() {

                      @Override
                      public void onSuspend(AtmosphereResourceEvent event) {
                        logger.trace("{}", event);
                        suspended.countDown();
                      }

                      @Override
                      public void onResume(AtmosphereResourceEvent event) {
                        onResume.set(true);
                        eventReceived.countDown();
                      }
                    });

                if (suspended.getCount() != 0) {
                  resource.set(r.suspend());
                } else {
                  try {
                    suspended.await(5, TimeUnit.SECONDS);
                  } catch (InterruptedException e) {
                    e.printStackTrace();
                  }
                  String message = r.getRequest().getReader().readLine();
                  r.getBroadcaster().broadcast(message);
                  new Timer()
                      .schedule(
                          new TimerTask() {
                            @Override
                            public void run() {
                              resource.get().resume();
                            }
                          },
                          2000);
                }
              }

              public void onStateChange(AtmosphereResourceEvent event) throws IOException {
                if (event.isSuspended()) {
                  event.getResource().write(event.getMessage().toString().getBytes());
                }
              }
            },
            BroadcasterFactory.getDefault().get(DefaultBroadcaster.class, "suspend"));

    AsyncHttpClient c = new AsyncHttpClient();
    try {
      final AtomicReference<String> response = new AtomicReference<String>(null);
      WebSocket w =
          c.prepareGet(urlTarget).execute(new WebSocketUpgradeHandler.Builder().build()).get();
      w.addWebSocketListener(
              new WebSocketTextListener() {

                @Override
                public void onMessage(String message) {
                  response.set(message);
                  latch.countDown();
                }

                @Override
                public void onFragment(String fragment, boolean last) {
                  fail();
                }

                @Override
                public void onOpen(WebSocket websocket) {}

                @Override
                public void onClose(WebSocket websocket) {
                  latch.countDown();
                }

                @Override
                public void onError(Throwable t) {
                  t.printStackTrace();
                  fail(t.getMessage());
                }
              })
          .sendTextMessage("echo");

      try {
        latch.await(20, TimeUnit.SECONDS);
      } catch (InterruptedException e) {
        fail(e.getMessage());
      }

      eventReceived.await(10, TimeUnit.SECONDS);
      assertNotNull(response.get());
      assertEquals(response.get(), "echo");
      assertTrue(onResume.get());

    } catch (Exception e) {
      logger.error("test failed", e);
      fail(e.getMessage());
    }
    c.close();
  }
コード例 #2
0
ファイル: BaseTest.java プロジェクト: juansarabia/atmosphere
  @Test(timeOut = 60000, enabled = true)
  public void sessionTest() {
    logger.info("{}: running test: sessionTest", getClass().getSimpleName());

    final AtomicReference<HttpSession> session = new AtomicReference<HttpSession>();
    final AtomicReference<HttpSession> sessionOnPost = new AtomicReference<HttpSession>();

    final CountDownLatch latch = new CountDownLatch(1);
    atmoServlet
        .framework()
        .addAtmosphereHandler(
            ROOT,
            new AbstractHttpAtmosphereHandler() {

              boolean isSuspended = false;

              public void onRequest(AtmosphereResource r) throws IOException {
                if (!isSuspended) {
                  session.set(r.suspend().session(true));
                  isSuspended = true;
                } else {
                  String message = r.getRequest().getReader().readLine();
                  r.getBroadcaster().broadcast(message);
                }
              }

              public void onStateChange(AtmosphereResourceEvent event) throws IOException {
                if (event.isSuspended()) {
                  sessionOnPost.set(event.getResource().session());
                  event.getResource().write(event.getMessage().toString().getBytes());
                }
              }
            },
            BroadcasterFactory.getDefault().get(DefaultBroadcaster.class, "suspend"));

    AsyncHttpClient c = new AsyncHttpClient();
    try {
      final AtomicReference<String> response = new AtomicReference<String>(null);
      WebSocket w =
          c.prepareGet(urlTarget).execute(new WebSocketUpgradeHandler.Builder().build()).get();
      w.addWebSocketListener(
              new WebSocketTextListener() {

                @Override
                public void onMessage(String message) {
                  response.set(message);
                  latch.countDown();
                }

                @Override
                public void onFragment(String fragment, boolean last) {
                  fail();
                }

                @Override
                public void onOpen(WebSocket websocket) {}

                @Override
                public void onClose(WebSocket websocket) {
                  latch.countDown();
                }

                @Override
                public void onError(Throwable t) {
                  t.printStackTrace();
                  fail(t.getMessage());
                }
              })
          .sendTextMessage("echo");

      try {
        latch.await(20, TimeUnit.SECONDS);
      } catch (InterruptedException e) {
        fail(e.getMessage());
      }

      assertNotNull(response.get());
      assertEquals(response.get(), "echo");
      assertEquals(session.get().getId(), sessionOnPost.get().getId());

    } catch (Exception e) {
      logger.error("test failed", e);
      fail(e.getMessage());
    }
    c.close();
  }