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
|
import unittest
from mock import patch, MagicMock, ANY
import pixelated
class ApplicationTest(unittest.TestCase):
class MockConfig:
def __init__(self, port, host, sslkey=None, sslcert=None):
self.port = port
self.host = host
self.sslkey = sslkey
self.sslcert = sslcert
self.home = 'leap_home'
@patch('pixelated.application.reactor')
@patch('pixelated.application.Services')
def test_that_create_app_binds_to_tcp_port_if_no_ssl_options(self, services_mock, reactor_mock):
app_mock = MagicMock()
leap_session = MagicMock()
config = ApplicationTest.MockConfig(12345, '127.0.0.1', leap_session)
d = pixelated.application.start_user_agent(app_mock, config.host, config.port, config.sslkey, config.sslcert, config.home, leap_session)
def _assert(_):
services_mock.assert_called_once_with(config.home, leap_session)
reactor_mock.listenTCP.assert_called_once_with(12345, ANY, interface='127.0.0.1')
app_mock.stopListening.assert_called()
d.addCallback(_assert)
return d
@patch('pixelated.application.reactor')
@patch('pixelated.application.Services')
def test_that_create_app_binds_to_ssl_if_ssl_options(self, services_mock, reactor_mock):
app_mock = MagicMock()
leap_session = MagicMock()
pixelated.application._ssl_options = lambda x, y: 'options'
config = ApplicationTest.MockConfig(12345, '127.0.0.1', sslkey="sslkey", sslcert="sslcert")
d = pixelated.application.start_user_agent(app_mock, config.host, config.port, config.sslkey, config.sslcert, config.home, leap_session)
def _assert(_):
services_mock.assert_called_once_with(config.home, leap_session)
reactor_mock.listenSSL.assert_called_once_with(12345, ANY, 'options', interface='127.0.0.1')
app_mock.stopListening.assert_called()
d.addCallback(_assert)
return d
|
