/*
* Copyright 2000-2006 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package jetbrains.communicator.p2p;
import jetbrains.communicator.core.impl.BaseTestCase;
import jetbrains.communicator.core.impl.users.UserImpl;
import jetbrains.communicator.core.users.PresenceMode;
import jetbrains.communicator.core.users.User;
import jetbrains.communicator.core.users.UserPresence;
import jetbrains.communicator.core.Pico;
import jetbrains.communicator.ide.NullProgressIndicator;
import jetbrains.communicator.ide.ProgressIndicator;
import jetbrains.communicator.util.WaitFor;
import org.jmock.Mock;
import org.jmock.core.constraint.IsGreaterThan;
import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.*;
/**
* @author Kir Maximov
*/
public class UserMonitorThreadTest extends BaseTestCase {
private Mock myUserMonitorClientMock;
private UserMonitorThread myUserMonitorThread;
private MulticastPingThread myMulticastThread;
private static final int PORT = 12234;
private static final int WAIT_USER_RESPONSES_TIMEOUT = 500;
@Override
protected void setUp() throws Exception {
super.setUp();
myUserMonitorClientMock = mock(UserMonitorClient.class);
final boolean[] started = new boolean[1];
myMulticastThread = new MulticastPingThread(InetAddress.getByName("localhost"), null, (UserMonitorClient) myUserMonitorClientMock.proxy()) {
@Override
public void run() {
myStarted = true;
try {
sleep(100000);
} catch (InterruptedException e) {
assert false;
}
}
@Override
public void sendMulticastPingRequest() throws IOException {
started[0] = true;
}
};
myUserMonitorClientMock.stubs().method("getPort").will(returnValue(PORT));
myUserMonitorThread = new UserMonitorThread(new MulticastPingThread[]{myMulticastThread},
(UserMonitorClient) myUserMonitorClientMock.proxy(), WAIT_USER_RESPONSES_TIMEOUT);
myUserMonitorThread.start();
triggerFind();
new WaitFor() { @Override
protected boolean condition() { return started[0]; } };
}
@Override
protected void tearDown() throws Exception {
myUserMonitorThread.shutdown();
new WaitFor(5000) {
@Override
protected boolean condition() {
return !myUserMonitorThread._isAlive();
}
};
super.tearDown();
}
private void triggerFind() {
myUserMonitorThread.triggerFindNow();
}
public void testFlushOnlineUsers_NoUsers() throws Exception {
assert myUserMonitorThread.isFinding();
myUserMonitorClientMock.expects(once()).method("setOnlineUsers").with(eq(Collections.emptySet()));
myUserMonitorThread.flushOnlineUsers();
assertTrue("Explicit flushing user list should not interrupt find process", myUserMonitorThread.isFinding());
}
public void testFlushOnlineUsers_WithUsers() throws Exception {
expectSetOneOnlineUser();
myUserMonitorThread.flushOnlineUsers();
}
public void testSetOnlineUsers() throws Exception {
expectSetOneOnlineUser();
assertTrue("Should be finding", myUserMonitorThread.isFinding());
Thread.sleep(myUserMonitorThread.getWaitUserResponsesTimeout());
new WaitFor(1000) {
@Override
protected boolean condition() {
return !myUserMonitorThread.isFinding();
}
};
assertFalse("Should be waiting wait for next cycle of user finding", myUserMonitorThread.isFinding());
}
public void testForceFind_WhenInFindState() throws Exception {
expectSetOneOnlineUser();
myUserMonitorThread.findNow(createProgressIndicator()); // should wait until the end of finding process;
assertFalse("Should wait for next cycle of user finding", myUserMonitorThread.isFinding());
}
public void testProgressIndicatorInvocations() throws Exception {
expectSetOneOnlineUser();
Mock mockProgressIndicator = mock(ProgressIndicator.class);
mockProgressIndicator.expects(atLeastOnce()).method("setText").with(eq("Found in local network: 1 user"));
mockProgressIndicator.expects(atLeastOnce()).method("checkCanceled");
final double[] lastFraction = new double[]{-1};
mockProgressIndicator.expects(atLeastOnce()).method("setFraction").with(new IsGreaterThan(new Comparable() {
@Override
public int compareTo(Object o) {
double v = ((Number) o).doubleValue();
int result = (int) ((lastFraction[0] - v) * 100);
lastFraction[0] = v;
if (result >= 0) {
System.out.println("result = " + result);
}
return result;
}
public String toString() {
return "" + lastFraction[0];
}
}));
myUserMonitorThread.findNow((ProgressIndicator) mockProgressIndicator.proxy());
assertEquals("Should end at 100%", 1, lastFraction[0], 0.01);
}
public void testAddRemoteUser_LoopbackAddress() throws Exception {
Pico.setUnitTest(false);
try {
myUserMonitorClientMock.expects(once()).method("setOnlineUsers") .with(eq(new HashSet()));
myUserMonitorThread.addOnlineUser("localhost", "nick", new Integer(PORT), new HashSet<String>(), new UserPresence(PresenceMode.AWAY));
myUserMonitorThread.flushOnlineUsers();
}
finally {
Pico.setUnitTest(true);
}
}
private void expectSetOneOnlineUser() throws UnknownHostException {
User p2PUser = UserImpl.create("nick", P2PTransport.CODE);
List<String> projects = new ArrayList<String>();
projects.add("project1");
OnlineUserInfo onlineUserInfo = new OnlineUserInfo(InetAddress.getByName("localhost"), PORT, projects, new UserPresence(PresenceMode.AWAY));
myUserMonitorClientMock.expects(once()).method("createUser").with(
eq("nick"),
eq(onlineUserInfo)
).will(returnValue(p2PUser));
myUserMonitorThread.addOnlineUser("localhost", "nick", new Integer(PORT), projects, new UserPresence(PresenceMode.AWAY));
myUserMonitorClientMock.expects(once()).method("setOnlineUsers")
.with(eq(new HashSet(Arrays.asList(p2PUser))));
}
private static ProgressIndicator createProgressIndicator() {
return new NullProgressIndicator();
}
}