def test_roamingNotAllowedError(self): visitedPlmnId = "47836" imsi = "286031595270296" mmeAuthProcHandler = MmeAuthProcedureHandler((localhost(), 9000), self.mmeIoService, 1, self.__mmeAuthCompleteCallback__) self.mmeIoService.addIncomingMessageCallback(mmeAuthProcHandler.handleIncomingMessage) hssAuthProcHandler = HssAuthProcedureHandler(self.hssIoService, self.__hssAuthCompleteCallback__) self.hssIoService.addIncomingMessageCallback(hssAuthProcHandler.handleIncomingMessage) mmeAuthProcHandler.execute(imsi, visitedPlmnId) self.assertEqual(self.mmeSuccessCount,0)
def test_unknownImsiError(self): n = 10 visitedPlmnId = "28603" imsi = "286056548201466" mmeAuthProcHandler = MmeAuthProcedureHandler((localhost(), 9000), self.mmeIoService, 1, self.__mmeAuthCompleteCallback__) self.mmeIoService.addIncomingMessageCallback(mmeAuthProcHandler.handleIncomingMessage) hssAuthProcHandler = HssAuthProcedureHandler(self.hssIoService, self.__hssAuthCompleteCallback__) self.hssIoService.addIncomingMessageCallback(hssAuthProcHandler.handleIncomingMessage) for _ in range(n): randomImsi = visitedPlmnId + "".join([str(random.randrange(0, 10)) for __ in range(10)]) hssAuthProcHandler.knownIMSIs.append(randomImsi) mmeAuthProcHandler.execute(imsi, visitedPlmnId) self.assertEqual(self.mmeSuccessCount,0)
def test_nAuthInfoRetrievalProcedureSuccess(self): n = 100 visitedPlmnId = "28603" mmeAuthProcHandler = MmeAuthProcedureHandler((localhost(), 9000), self.mmeIoService, self.__mmeAuthCompleteCallback__) self.mmeIoService.addIncomingMessageCallback(mmeAuthProcHandler.handleIncomingMessage) hssAuthProcHandler = HssAuthProcedureHandler(self.hssIoService, self.__hssAuthCompleteCallback__) self.hssIoService.addIncomingMessageCallback(hssAuthProcHandler.handleIncomingMessage) for _ in range(n): randomImsi = visitedPlmnId + "".join([str(random.randrange(0, 10)) for __ in range(10)]) mmeAuthProcHandler.execute(randomImsi, visitedPlmnId) time.sleep(0.1) time.sleep(1.0) self.assertEqual(self.mmeSuccess, n) self.assertEqual(self.hssSuccess, n)
def test_responseTimeoutError(self): n = 100 visitedPlmnId = "28603" mmeAuthProcHandler = MmeAuthProcedureHandler((localhost(), 9000), self.mmeIoService, 1, self.__mmeAuthCompleteCallback__) self.mmeIoService.addIncomingMessageCallback(mmeAuthProcHandler.handleIncomingMessage) hssAuthProcHandler = HssAuthProcedureHandler(self.hssIoService, self.__hssAuthCompleteCallback__) self.hssIoService.addIncomingMessageCallback(hssAuthProcHandler.handleIncomingMessage) for _ in range(n): randomImsi = visitedPlmnId + "".join([str(random.randrange(0, 10)) for __ in range(10)]) hssAuthProcHandler.knownIMSIs.append(randomImsi) mmeAuthProcHandler.execute(randomImsi, visitedPlmnId) time.sleep(0.05) time.sleep(5.0) self.assertEqual(self.mmeSuccessCount, n - mmeAuthProcHandler.timeoutCounter)