blob: f073a3caa418c2b9cc44b959376c1810dc8cd1d9 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
package org.gnunet.util;
import org.gnunet.testing.TestingServer;
import org.gnunet.testing.TestingSetup;
import org.junit.Assert;
import org.junit.Test;
/**
* ...
*
* @author Florian Dold
*/
public class ClientServerTest {
/**
* Test if the server receives a message sent by a client.
*/
@Test(timeout = 1000)
public void test_testing_server() {
final TestingSetup setup = new TestingSetup();
Program.configureLogging("DEBUG", null);
final TestingServer srv = setup.createServer();
final Wrapper<Boolean> gotMessage = new Wrapper<Boolean>(false);
srv.server.setHandler(new Server.MessageRunabout() {
public void visit(TESTMessage tm) {
gotMessage.set(true);
srv.server.destroy();
}
});
Scheduler.run(new Scheduler.Task() {
@Override
public void run(Scheduler.RunContext ctx) {
final Client cli = srv.createClient();
cli.notifyTransmitReady(RelativeTime.FOREVER,true, 0, new MessageTransmitter() {
@Override
public void transmit(Connection.MessageSink sink) {
System.out.println("ntr!");
sink.send(new TESTMessage());
}
@Override
public void handleError() {
Assert.fail();
}
});
System.out.println("done");
}
});
Assert.assertTrue(gotMessage.get());
}
/**
* Test what happens when a client calls notifyTransmitReady, but does not send
* a message in the callback but disconnects.
*/
@Test(timeout = 1000)
public void test_premature_disconnect() {
final TestingSetup setup = new TestingSetup();
Program.configureLogging("DEBUG", null);
final TestingServer srv = setup.createServer();
srv.server.notifyDisconnect(new Server.DisconnectHandler() {
@Override
public void onDisconnect(Server.ClientHandle clientHandle) {
srv.server.destroy();
}
});
Scheduler.run(new Scheduler.Task() {
@Override
public void run(Scheduler.RunContext ctx) {
final Client cli = srv.createClient();
cli.notifyTransmitReady(RelativeTime.FOREVER,true, 0, new MessageTransmitter() {
@Override
public void transmit(Connection.MessageSink sink) {
sink.send(new TESTMessage());
cli.disconnect();
}
@Override
public void handleError() {
Assert.fail();
}
});
}
});
}
}
|