def setUp(self):
self.clock = sip.clock = task.Clock()
self.dbdir = self.mktemp()
self.store = store.Store(self.dbdir)
self.login = userbase.LoginSystem(store=self.store)
installOn(self.login, self.store)
account = self.login.addAccount('bob', 'proxy2.org', None)
account2 = self.login.addAccount('alice', 'proxy1.org', None)
us = self.us = account.avatars.open()
installOn(FakeAvatar(store=us), us)
us2 = self.us2 = account2.avatars.open()
installOn(FakeAvatar(store=us2), us2)
self.tq = TaskQueue(self.clock)
self.uas = useragent.UserAgent.server(sip.IVoiceSystem(us), "10.0.0.2", FakeMediaController())
self.uas2 = useragent.UserAgent.server(sip.IVoiceSystem(us2), "10.0.0.1", FakeMediaController())
self.sip1 = sip.SIPTransport(self.uas, ["server.com"], 5060)
self.sip1.startProtocol()
self.sip2 = sip.SIPTransport(self.uas2, ["client.com"], 5060)
self.sip2.startProtocol()
self.svc = sipserver.SIPServer(store=self.store)
self.svc.mediaController = FakeMediaController()
portal = Portal(self.login, [self.login])
self.svc.dispatcher = sip.SIPDispatcher(portal, sip.Proxy(portal))
self.svc.transport = sip.SIPTransport(self.svc.dispatcher, sipserver.getHostnames(self.store), 5060)
self.svc.dispatcher.start(self.svc.transport)
dests = {('10.0.0.2', 5060): self.uas, ('10.0.0.1', 5060): self.uas2,
('127.0.0.1', 5060): self.svc}
self.messages = []
self.sip1.sendMessage = lambda msg, dest: self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("10.0.0.2", 5060))
self.sip2.sendMessage = lambda msg, dest: self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("10.0.0.1", 5060))
self.svc.transport.sendMessage = lambda msg, dest: self.messages.append(msg) or self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("127.0.0.1", 5060))
useragent.Dialog.genTag = lambda self: "314159"
class TPCCTest(unittest.TestCase):
def setUp(self):
self.clock = sip.clock = task.Clock()
self.dbdir = self.mktemp()
self.store = store.Store(self.dbdir)
self.login = userbase.LoginSystem(store=self.store)
installOn(self.login, self.store)
account = self.login.addAccount('bob', 'proxy2.org', None)
account2 = self.login.addAccount('alice', 'proxy1.org', None)
us = self.us = account.avatars.open()
installOn(FakeAvatar(store=us), us)
us2 = self.us2 = account2.avatars.open()
installOn(FakeAvatar(store=us2), us2)
self.tq = TaskQueue(self.clock)
self.uas = useragent.UserAgent.server(sip.IVoiceSystem(us), "10.0.0.2", FakeMediaController())
self.uas2 = useragent.UserAgent.server(sip.IVoiceSystem(us2), "10.0.0.1", FakeMediaController())
self.sip1 = sip.SIPTransport(self.uas, ["server.com"], 5060)
self.sip1.startProtocol()
self.sip2 = sip.SIPTransport(self.uas2, ["client.com"], 5060)
self.sip2.startProtocol()
self.svc = sipserver.SIPServer(store=self.store)
self.svc.mediaController = FakeMediaController()
portal = Portal(self.login, [self.login])
self.svc.dispatcher = sip.SIPDispatcher(portal, sip.Proxy(portal))
self.svc.transport = sip.SIPTransport(self.svc.dispatcher, sipserver.getHostnames(self.store), 5060)
self.svc.dispatcher.start(self.svc.transport)
dests = {('10.0.0.2', 5060): self.uas, ('10.0.0.1', 5060): self.uas2,
('127.0.0.1', 5060): self.svc}
self.messages = []
self.sip1.sendMessage = lambda msg, dest: self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("10.0.0.2", 5060))
self.sip2.sendMessage = lambda msg, dest: self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("10.0.0.1", 5060))
self.svc.transport.sendMessage = lambda msg, dest: self.messages.append(msg) or self.tq.addTask(dests[dest].transport.datagramReceived, str(msg.toString()), ("127.0.0.1", 5060))
useragent.Dialog.genTag = lambda self: "314159"
def tearDown(self):
self.clock.advance(33)
self.clock.advance(33)
def tearDown(self):
sip.clock = reactor
def test3PCC(self):
self.svc.setupCallBetween(sip.parseAddress("sip:[email protected]"),
sip.parseAddress("sip:[email protected]"))
def testDialogAddsSDP(self):
"""
Previously, there was a bug where no SDP was attached to
certain INVITEs. This ensures that it gets added.
"""
uac = useragent.UserAgent.server(sip.IVoiceSystem(self.us2), "10.0.0.1", FakeMediaController())
d = useragent.Dialog.forClient(uac, sip.URL(uac.host, "clicktocall"), sip.URL("10.0.0.2", "bob"), None, False, '')
def invite(dlg):
return dlg._generateInvite(sip.URL("localhost", "clicktocall"), "", sip.URL("10.0.0.2", "bob"), False)
def test(dlg):
sdplines = dlg.msg.body.split('\r\n')
#There needs to be a m= line at a minimum, probably some other stuff.
self.failIfEqual([line for line in sdplines if line.startswith("m=")], [])
return d.addCallback(invite).addCallback(test)
def testCreateRTPSocket(self):
lcp = useragent.LocalControlProtocol(False)
lcp.makeConnection(StringTransport())
lcp.createRTPSocket("A Dialog", u"server.com")
box = juice.parseString(lcp.transport.value())
self.assertEquals(box[0]['host'], "server.com")
lcp.dataReceived('-Answer: %s\r\nCookie: 123\r\n\r\n' % (box[0]['_ask']))
self.assertEquals(lcp.dialogs['123'], "A Dialog")
self.assertEquals(lcp.cookies["A Dialog"], '123')
