blob: 463e4ebbae542a5c87ce697517be74f125667c86 (
plain)
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
|
<?php
class Swift_Plugins_AntiFloodPluginTest extends \PHPUnit_Framework_TestCase
{
public function testThresholdCanBeSetAndFetched()
{
$plugin = new Swift_Plugins_AntiFloodPlugin(10);
$this->assertEquals(10, $plugin->getThreshold());
$plugin->setThreshold(100);
$this->assertEquals(100, $plugin->getThreshold());
}
public function testSleepTimeCanBeSetAndFetched()
{
$plugin = new Swift_Plugins_AntiFloodPlugin(10, 5);
$this->assertEquals(5, $plugin->getSleepTime());
$plugin->setSleepTime(1);
$this->assertEquals(1, $plugin->getSleepTime());
}
public function testPluginStopsConnectionAfterThreshold()
{
$transport = $this->_createTransport();
$transport->expects($this->once())
->method('start');
$transport->expects($this->once())
->method('stop');
$evt = $this->_createSendEvent($transport);
$plugin = new Swift_Plugins_AntiFloodPlugin(10);
for ($i = 0; $i < 12; ++$i) {
$plugin->sendPerformed($evt);
}
}
public function testPluginCanStopAndStartMultipleTimes()
{
$transport = $this->_createTransport();
$transport->expects($this->exactly(5))
->method('start');
$transport->expects($this->exactly(5))
->method('stop');
$evt = $this->_createSendEvent($transport);
$plugin = new Swift_Plugins_AntiFloodPlugin(2);
for ($i = 0; $i < 11; ++$i) {
$plugin->sendPerformed($evt);
}
}
public function testPluginCanSleepDuringRestart()
{
$sleeper = $this->getMockBuilder('Swift_Plugins_Sleeper')->getMock();
$sleeper->expects($this->once())
->method('sleep')
->with(10);
$transport = $this->_createTransport();
$transport->expects($this->once())
->method('start');
$transport->expects($this->once())
->method('stop');
$evt = $this->_createSendEvent($transport);
$plugin = new Swift_Plugins_AntiFloodPlugin(99, 10, $sleeper);
for ($i = 0; $i < 101; ++$i) {
$plugin->sendPerformed($evt);
}
}
private function _createTransport()
{
return $this->getMockBuilder('Swift_Transport')->getMock();
}
private function _createSendEvent($transport)
{
$evt = $this->getMockBuilder('Swift_Events_SendEvent')
->disableOriginalConstructor()
->getMock();
$evt->expects($this->any())
->method('getSource')
->will($this->returnValue($transport));
$evt->expects($this->any())
->method('getTransport')
->will($this->returnValue($transport));
return $evt;
}
}
|