def setUp(self): # Create sqlite3 database for testing self._tmpfile = tempfile.TemporaryDirectory() store = SqliteStore(self._tmpfile.name + '/') op = 16 * b'\x11' amf = b'\x80\x00' self._sub_profiles = { 'superfast': SubscriberDB.SubscriptionProfile(max_ul_bit_rate=100000, max_dl_bit_rate=50000) } self._default_sub_profile = SubscriberDB.SubscriptionProfile( max_ul_bit_rate=10000, max_dl_bit_rate=5000) self._processor = processor.Processor(store, self._default_sub_profile, self._sub_profiles, op, amf) # Add some test users (rand, sres, gsm_key) = _dummy_auth_tuple() gsm = GSMSubscription(state=GSMSubscription.ACTIVE, auth_tuples=[rand + sres + gsm_key]) lte_key = 16 * b'\x00' lte = LTESubscription(state=LTESubscription.ACTIVE, auth_key=lte_key) lte_opc = LTESubscription(state=LTESubscription.ACTIVE, auth_key=lte_key, auth_opc=Milenage.generate_opc(lte_key, op)) lte_opc_short = LTESubscription(state=LTESubscription.ACTIVE, auth_key=lte_key, auth_opc=b'\x00') state = SubscriberState(lte_auth_next_seq=1) sub1 = SubscriberData(sid=SIDUtils.to_pb('IMSI11111'), gsm=gsm, lte=lte, state=state, sub_profile='superfast') sub2 = SubscriberData( sid=SIDUtils.to_pb('IMSI22222'), # No auth keys gsm=GSMSubscription(state=GSMSubscription.ACTIVE), lte=LTESubscription(state=LTESubscription.ACTIVE)) sub3 = SubscriberData( sid=SIDUtils.to_pb('IMSI33333')) # No subscribtion sub4 = SubscriberData(sid=SIDUtils.to_pb('IMSI44444'), lte=lte_opc, state=state) sub5 = SubscriberData(sid=SIDUtils.to_pb('IMSI55555'), lte=lte_opc_short, state=state) store.add_subscriber(sub1) store.add_subscriber(sub2) store.add_subscriber(sub3) store.add_subscriber(sub4) store.add_subscriber(sub5)
def setUp(self): # Create sqlite3 database for testing self._tmpfile = tempfile.TemporaryDirectory() store = SqliteStore(self._tmpfile.name + "/") op = 16 * b"\x11" amf = b"\x80\x00" sub_network = CoreNetworkType() self._sub_profiles = { "superfast": SubscriberDB.SubscriptionProfile( max_ul_bit_rate=100000, max_dl_bit_rate=50000, ), } self._default_sub_profile = SubscriberDB.SubscriptionProfile( max_ul_bit_rate=10000, max_dl_bit_rate=5000, ) self._sub_network = sub_network self._processor = processor.Processor( store, self._default_sub_profile, self._sub_profiles, op, amf, self._sub_network, ) # Add some test users (rand, sres, gsm_key) = _dummy_auth_tuple() gsm = GSMSubscription( state=GSMSubscription.ACTIVE, auth_tuples=[rand + sres + gsm_key], ) lte_key = 16 * b"\x00" lte = LTESubscription(state=LTESubscription.ACTIVE, auth_key=lte_key) lte_opc = LTESubscription( state=LTESubscription.ACTIVE, auth_key=lte_key, auth_opc=Milenage.generate_opc(lte_key, op), ) lte_opc_short = LTESubscription( state=LTESubscription.ACTIVE, auth_key=lte_key, auth_opc=b"\x00", ) sub_network = CoreNetworkType(forbidden_network_types=[ CoreNetworkType.NT_EPC, CoreNetworkType.NT_5GC ], ) state = SubscriberState(lte_auth_next_seq=1) sub1 = SubscriberData( sid=SIDUtils.to_pb("IMSI11111"), gsm=gsm, lte=lte, state=state, sub_profile="superfast", ) sub2 = SubscriberData( sid=SIDUtils.to_pb("IMSI22222"), # No auth keys gsm=GSMSubscription(state=GSMSubscription.ACTIVE), lte=LTESubscription(state=LTESubscription.ACTIVE), ) sub3 = SubscriberData( sid=SIDUtils.to_pb("IMSI33333")) # No subscription sub4 = SubscriberData(sid=SIDUtils.to_pb("IMSI44444"), lte=lte_opc, state=state) sub5 = SubscriberData( sid=SIDUtils.to_pb("IMSI55555"), lte=lte_opc_short, state=state, ) sub6 = SubscriberData( sid=SIDUtils.to_pb("IMSI66666"), sub_network=CoreNetworkType(), ) store.add_subscriber(sub1) store.add_subscriber(sub2) store.add_subscriber(sub3) store.add_subscriber(sub4) store.add_subscriber(sub5) store.add_subscriber(sub6)