def setUp(self): self.enbIoService = IoService("enb", 9000) self.ueIoService = IoService("ue", 9001) [s.start() for s in self.enbIoService, self.ueIoService] self.enbProcedure = EnbInitialSecurityActivationProcedureHandler( self.enbIoService, self.__procedureCompleteCallback__) self.enbIoService.addIncomingMessageCallback(self.enbProcedure.handleIncomingMessage) self.result = None
def setUp(self): self.enbIoService = IoService("enb", 9000) self.ueIoService = IoService("ue", 9001) [s.start() for s in self.enbIoService, self.ueIoService] self.enbProcedure = EnbInitialSecurityActivationProcedureHandler( self.enbIoService, self.__procedureCompleteCallback__) self.enbIoService.addIncomingMessageCallback( self.enbProcedure.handleIncomingMessage) self.result = None
class TestEnbInitialSecurityActivationProcedure(unittest.TestCase): def setUp(self): self.enbIoService = IoService("enb", 9000) self.ueIoService = IoService("ue", 9001) [s.start() for s in self.enbIoService, self.ueIoService] self.enbProcedure = EnbInitialSecurityActivationProcedureHandler( self.enbIoService, self.__procedureCompleteCallback__) self.enbIoService.addIncomingMessageCallback(self.enbProcedure.handleIncomingMessage) self.result = None def __procedureCompleteCallback__(self, result, rrcTransactionIdentifier): self.result = result def test_procedureSuccessful(self): self.enbProcedure.start((localhost(), 9001), 0, "cipherAlgo", "integProtAlgo") time.sleep(0.1) self.ueIoService.sendMessage("enb", *securityModeComplete(0)) time.sleep(0.1) self.assertEqual(self.result, EnbInitialSecurityActivationProcedureHandler.Complete)
class TestEnbInitialSecurityActivationProcedure(unittest.TestCase): def setUp(self): self.enbIoService = IoService("enb", 9000) self.ueIoService = IoService("ue", 9001) [s.start() for s in self.enbIoService, self.ueIoService] self.enbProcedure = EnbInitialSecurityActivationProcedureHandler( self.enbIoService, self.__procedureCompleteCallback__) self.enbIoService.addIncomingMessageCallback( self.enbProcedure.handleIncomingMessage) self.result = None def __procedureCompleteCallback__(self, result, rrcTransactionIdentifier): self.result = result def test_procedureSuccessful(self): self.enbProcedure.start((localhost(), 9001), 0, "cipherAlgo", "integProtAlgo") time.sleep(0.1) self.ueIoService.sendMessage("enb", *securityModeComplete(0)) time.sleep(0.1) self.assertEqual(self.result, EnbInitialSecurityActivationProcedureHandler.Complete)