1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package org.apache.mina.filter.firewall;
22
23 import static org.junit.Assert.assertFalse;
24 import static org.junit.Assert.assertTrue;
25
26 import java.net.InetSocketAddress;
27
28 import org.apache.mina.core.session.DummySession;
29 import org.junit.After;
30 import org.junit.Before;
31 import org.junit.Test;
32
33
34
35
36
37
38 public class ConnectionThrottleFilterTest {
39 private ConnectionThrottleFilter filter;
40
41 private DummySession sessionOne;
42
43 private DummySession sessionTwo;
44
45 @Before
46 public void setUp() throws Exception {
47 filter = new ConnectionThrottleFilter();
48
49 sessionOne = new DummySession();
50 sessionOne.setRemoteAddress(new InetSocketAddress(1234));
51 sessionTwo = new DummySession();
52 sessionTwo.setRemoteAddress(new InetSocketAddress(1235));
53 }
54
55 @After
56 public void tearDown() throws Exception {
57 filter = null;
58 }
59
60 @Test
61 public void testGoodConnection() {
62 filter.setAllowedInterval(100);
63 filter.isConnectionOk(sessionOne);
64
65 try {
66 Thread.sleep(1000);
67 } catch (InterruptedException e) {
68
69 }
70
71 boolean result = filter.isConnectionOk(sessionOne);
72 assertTrue(result);
73 }
74
75 @Test
76 public void testBadConnection() {
77 filter.setAllowedInterval(1000);
78 filter.isConnectionOk(sessionTwo);
79 assertFalse(filter.isConnectionOk(sessionTwo));
80 }
81 }