Examples with Collection com.tvd12.dahlia.core.entity.Collection used on opensource projects

Example 31 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project ezyfox-server by youngmonkeys.

the class EzyAbstractStreamingApi method response.

@Override
public void response(EzyBytesPackage pack) throws Exception {
    EzyConstant connectionType = getConnectionType();
    Collection<EzySession> recipients = pack.getRecipients(connectionType);
    if (recipients.isEmpty()) {
        return;
    }
    byte[] bytes = pack.getBytes();
    for (EzySession session : recipients) {
        session.send(createPacket(bytes, pack));
    }
}

Example 32 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project ezyfox-server by youngmonkeys.

the class EzyAppResponseImplTest method test.

@SuppressWarnings("rawtypes")
@Test
public void test() throws Exception {
    EzyAppContext context = mock(EzyAppContext.class);
    EzyApplication application = mock(EzyApplication.class);
    EzyAppUserManager userManager = EzyAppUserManagerImpl.builder().build();
    when(context.getApp()).thenReturn(application);
    when(application.getUserManager()).thenReturn(userManager);
    EzyAppResponse response = new EzyAppResponseImpl(context);
    Field recipients = EzyAbstractResponse.class.getDeclaredField("recipients");
    Field exclusiveRecipients = EzyAbstractResponse.class.getDeclaredField("exclusiveRecipients");
    recipients.setAccessible(true);
    exclusiveRecipients.setAccessible(true);
    assert ((Collection) recipients.get(response)).size() == 0;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 0;
    EzyUser user1 = newUser("user1");
    response.user(user1);
    assert ((Collection) recipients.get(response)).size() == 2;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 0;
    EzyUser user2 = newUser("user2");
    response.user(user2);
    response.user(user2, true);
    assert ((Collection) recipients.get(response)).size() == 4;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 2;
    EzyUser user3 = newUser("user3");
    EzyUser user4 = newUser("user4");
    response.users(user3, user4);
    response.users(new EzyUser[] { user3, null }, true);
    assert ((Collection) recipients.get(response)).size() == 8;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 4;
    EzyUser user5 = newUser("user5");
    EzyUser user6 = newUser("user6");
    response.users(Lists.newArrayList(user5, user6));
    response.users(Lists.newArrayList(user5, null), true);
    assert ((Collection) recipients.get(response)).size() == 12;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 6;
    EzyUser user7 = newUser("user7");
    EzyUser user8 = newUser("user8");
    userManager.addUser(user7);
    userManager.addUser(user8);
    response.username("user7");
    response.username("user7", true);
    assert ((Collection) recipients.get(response)).size() == 14;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 8;
    EzyUser user9 = newUser("user9");
    EzyUser user10 = newUser("user10");
    EzyUser user11 = newUser("user11");
    userManager.addUser(user9);
    userManager.addUser(user10);
    userManager.addUser(user11);
    response.usernames("user9", "user10");
    response.usernames(new String[] { "user11" }, true);
    assert ((Collection) recipients.get(response)).size() == 18;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 10;
    EzyUser user12 = newUser("user12");
    EzyUser user13 = newUser("user13");
    EzyUser user14 = newUser("user14");
    userManager.addUser(user12);
    userManager.addUser(user13);
    userManager.addUser(user14);
    response.usernames(Lists.newArrayList("user12", "user13"));
    response.usernames(Lists.newArrayList("user14"), true);
    assert ((Collection) recipients.get(response)).size() == 22;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 12;
    EzySession session1 = new ExSession();
    response.session(session1);
    assert ((Collection) recipients.get(response)).size() == 23;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 12;
    EzySession session2 = new ExSession();
    EzySession session3 = new ExSession();
    response.sessions(session2, session3);
    response.sessions(new EzySession[] { session3 }, true);
    assert ((Collection) recipients.get(response)).size() == 25;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 13;
    EzySession session4 = new ExSession();
    EzySession session5 = new ExSession();
    response.sessions(Lists.newArrayList(session4, session5));
    response.sessions(Lists.newArrayList(session4, session5), true);
    assert ((Collection) recipients.get(response)).size() == 27;
    assert ((Collection) exclusiveRecipients.get(response)).size() == 15;
}