def setUp(self):
yield super(AbstractTestTunnelCommunity, self).setUp()
self.tunnel_community = HiddenTunnelCommunity(self.dispersy, self.master_member, self.member)
self.tunnel_community._request_cache = RequestCache()
self.tunnel_community.socks_server = Socks5Server(self, 1234)
self.tunnel_community._initialize_meta_messages()
self.tunnel_community.add_conversion(TunnelConversion(self.tunnel_community))
class AbstractTestTunnelCommunity(AbstractTestCommunity):
# We have to initialize Dispersy and the tunnel community on the reactor thread
@blocking_call_on_reactor_thread
@inlineCallbacks
def setUp(self):
yield super(AbstractTestTunnelCommunity, self).setUp()
self.tunnel_community = HiddenTunnelCommunity(self.dispersy, self.master_member, self.member)
self.tunnel_community._request_cache = RequestCache()
self.tunnel_community.socks_server = Socks5Server(self, 1234)
self.tunnel_community._initialize_meta_messages()
self.tunnel_community.add_conversion(TunnelConversion(self.tunnel_community))
def setUp(self, autoload_discovery=True):
yield super(TestCircuitDebugEndpoint, self).setUp(autoload_discovery=autoload_discovery)
self.dispersy = Dispersy(ManualEnpoint(0), self.getStateDir())
self.dispersy._database.open()
master_member = DummyMember(self.dispersy, 1, "a" * 20)
member = self.dispersy.get_new_member(u"curve25519")
self.tunnel_community = HiddenTunnelCommunity(self.dispersy, master_member, member)
self.dispersy.get_communities = lambda: [self.tunnel_community]
self.session.get_dispersy_instance = lambda: self.dispersy
def test_circuit_tunnel_data(self):
"""
Test whether the right methods are called when tunneling data over a circuit
"""
proxy = HiddenTunnelCommunity.__new__(HiddenTunnelCommunity)
proxy.stats = {'bytes_up': 0}
proxy.send_data = lambda *_: 3
circuit = Circuit(1234L, 3, proxy=proxy, first_hop=("1.2.3.5", 1235))
circuit.tunnel_data(("1.2.3.4", 1234), 'abcd')
proxy.send_data = lambda *_: 0
circuit.tunnel_data(("1.2.3.4", 1234), 'abcd')
self.assertEqual(proxy.stats['bytes_up'], 3)