コード例 #1
0
  @Test
  public void testHeartbeat() throws IOException, ServletException {
    // Open connection
    AtmosphereRequest request =
        new AtmosphereRequest.Builder().pathInfo("/heartbeat").method("GET").build();

    request.header(X_ATMOSPHERE_TRANSPORT, WEBSOCKET_TRANSPORT);
    framework.doCometSupport(request, AtmosphereResponse.newInstance());

    // Check suspend
    final AtmosphereResource res = r.get();
    assertNotNull(res);

    // Send heartbeat
    request =
        new AtmosphereRequest.Builder()
            .pathInfo("/heartbeat")
            .method("GET")
            .body(Heartbeat.paddingData)
            .build();
    request.header(X_ATMOSPHERE_TRANSPORT, WEBSOCKET_TRANSPORT);
    request.setAttribute(HeartbeatInterceptor.INTERCEPTOR_ADDED, "");
    res.initialize(
        res.getAtmosphereConfig(),
        res.getBroadcaster(),
        request,
        AtmosphereResponse.newInstance(),
        framework.getAsyncSupport(),
        res.getAtmosphereHandler());
    request.setAttribute(FrameworkConfig.INJECTED_ATMOSPHERE_RESOURCE, res);
    framework.doCometSupport(request, AtmosphereResponse.newInstance());
    assertNotNull(message.get());
    assertEquals(message.get(), Heartbeat.paddingData);
  }
コード例 #2
0
  @Test
  public void testSuspend() throws IOException, ServletException {

    AtmosphereRequest request =
        new AtmosphereRequest.Builder().pathInfo("/j").method("GET").build();
    request.header(X_ATMOSPHERE_TRANSPORT, LONG_POLLING_TRANSPORT);
    framework.doCometSupport(request, AtmosphereResponse.newInstance());
    assertNotNull(r.get());
  }
コード例 #3
0
  @Test
  public void testPut() throws IOException, ServletException {

    AtmosphereRequest request =
        new AtmosphereRequest.Builder().pathInfo("/d").method("PUT").build();
    framework.doCometSupport(request, AtmosphereResponse.newInstance());
    assertNotNull(r.get());
    r.get().resume();
  }
コード例 #4
0
 @Test
 public void testInputStreamMessage() throws IOException, ServletException {
   AtmosphereRequest request =
       new AtmosphereRequest.Builder().pathInfo("/inputStreamInjection").method("GET").build();
   framework.doCometSupport(request, AtmosphereResponse.newInstance());
   assertNotNull(r.get());
   r.get().resume();
   assertNotNull(message.get());
   assertEquals(message.get(), "message");
 }
コード例 #5
0
 @Test
 public void createTest() {
   AtmosphereResource r =
       AtmosphereResourceFactory.getDefault()
           .create(
               mock(AtmosphereConfig.class),
               mock(Broadcaster.class),
               AtmosphereResponse.newInstance().request(AtmosphereRequest.create()),
               mock(AsyncSupport.class),
               mock(AtmosphereHandler.class));
   assertNotNull(r);
 }
コード例 #6
0
  @Test
  public void testOverrideBroadcaster() throws IOException, ServletException {
    framework.setDefaultBroadcasterClassName(SimpleBroadcaster.class.getName());

    AtmosphereRequest request =
        new AtmosphereRequest.Builder().pathInfo("/override").method("GET").build();
    request.header(X_ATMOSPHERE_TRANSPORT, LONG_POLLING_TRANSPORT);
    framework.doCometSupport(request, AtmosphereResponse.newInstance());

    assertNotNull(r.get());
    assertEquals(r.get().getBroadcaster().getClass().getName(), SimpleBroadcaster.class.getName());
  }
コード例 #7
0
  @Test
  public void testOnClose() throws IOException, ServletException {
    framework.addAtmosphereHandler(
        "/a",
        new AbstractReflectorAtmosphereHandler() {
          @Override
          public void onRequest(AtmosphereResource resource) throws IOException {}

          @Override
          public void destroy() {}
        });

    AtmosphereRequest request = new AtmosphereRequest.Builder().pathInfo("/a").build();

    final AtomicReference<Boolean> closed = new AtomicReference<Boolean>();

    final AtmosphereResourceEventListener listener =
        new AtmosphereResourceEventListenerAdapter() {
          @Override
          public void onClose(AtmosphereResourceEvent event) {
            closed.set(true);
          }
        };

    framework.interceptor(
        new AtmosphereInterceptor() {
          @Override
          public void configure(AtmosphereConfig config) {}

          @Override
          public Action inspect(AtmosphereResource r) {
            r.addEventListener(listener).suspend();
            try {
              r.getBroadcaster().broadcast("void").get();
            } catch (InterruptedException e) {
            } catch (ExecutionException e) {
            }
            return Action.CONTINUE;
          }

          @Override
          public void postInspect(AtmosphereResource r) {
            try {
              r.close();
            } catch (IOException e) {
              e.printStackTrace();
            }
          }
        });
    framework.doCometSupport(request, AtmosphereResponse.newInstance());
    assertTrue(closed.get());
  }
コード例 #8
0
  @Test
  public void testPriority() throws IOException, ServletException {

    AtmosphereRequest request =
        new AtmosphereRequest.Builder().pathInfo("/priority").method("GET").build();
    request.header(X_ATMOSPHERE_TRANSPORT, LONG_POLLING_TRANSPORT);
    framework.doCometSupport(request, AtmosphereResponse.newInstance());
    assertEquals(
        framework.getAtmosphereHandlers().get("/priority").interceptors.getFirst().toString(),
        "XXX");

    assertNotNull(r.get());
  }
コード例 #9
0
 @Test
 public void notFoundTest() {
   for (int i = 0; i < 10; i++) {
     BroadcasterFactory.getDefault().get(String.valueOf(i));
   }
   AtmosphereResource r =
       AtmosphereResourceFactory.getDefault()
           .create(
               mock(AtmosphereConfig.class),
               BroadcasterFactory.getDefault().lookup("1"),
               AtmosphereResponse.newInstance().request(AtmosphereRequest.create()),
               mock(AsyncSupport.class),
               mock(AtmosphereHandler.class));
   assertNotNull(r);
   assertNull(AtmosphereResourceFactory.getDefault().find(r.uuid()));
 }
コード例 #10
0
  @Test
  public void findTest() {
    Broadcaster b1 = BroadcasterFactory.getDefault().get("b1");
    Broadcaster b2 = BroadcasterFactory.getDefault().get("b2");
    AtmosphereResource r =
        AtmosphereResourceFactory.getDefault()
            .create(
                mock(AtmosphereConfig.class),
                b1,
                AtmosphereResponse.newInstance().request(AtmosphereRequest.create()),
                mock(AsyncSupport.class),
                mock(AtmosphereHandler.class));
    assertNotNull(r);

    b2.addAtmosphereResource(r);

    assertNotNull(AtmosphereResourceFactory.getDefault().find(r.uuid()));
  }