encryption: Reuse code using test lib features
Change-Id: I9b6944516c524a0c11a47aec6839f8ee9d571661
diff --git a/suites/encryption/register_a5_1_authreq.py b/suites/encryption/register_a5_1_authreq.py
index e7aa76c..077819b 100755
--- a/suites/encryption/register_a5_1_authreq.py
+++ b/suites/encryption/register_a5_1_authreq.py
@@ -1,50 +1,8 @@
#!/usr/bin/env python3
from osmo_gsm_tester.testenv import *
-hlr = suite.hlr()
-bts = suite.bts()
-mgw_msc = suite.mgw()
-mgw_bsc = suite.mgw()
-stp = suite.stp()
-msc = suite.msc(hlr, mgw_msc, stp)
-bsc = suite.bsc(msc, mgw_bsc, stp)
-ms = suite.modem()
+import testlib
+suite.test_import_modules_register_for_cleanup(testlib)
+from testlib import encryption_test_setup_run
-print('start network...')
-msc.set_authentication(True)
-msc.set_encryption('a5_1')
-bsc.set_encryption('a5_1')
-hlr.start()
-stp.start()
-msc.start()
-mgw_msc.start()
-mgw_bsc.start()
-bsc.bts_add(bts)
-bsc.start()
-bts.start()
-wait(bsc.bts_is_connected, bts)
-
-ms.log_info()
-good_ki = ms.ki()
-bad_ki = ("%1X" % (int(good_ki[0], 16) ^ 0x01)) + good_ki[1:]
-
-print('KI changed: ' + good_ki + " => " + bad_ki)
-ms.set_ki(bad_ki)
-hlr.subscriber_add(ms)
-print('Attempt connection with wrong KI...')
-ms.connect(msc.mcc_mnc())
-
-sleep(40) # TODO: read pcap or CTRL interface and look for Rejected? (gsm_a.dtap.msg_mm_type == 0x04)
-print('Asserting modem did not register')
-# FIXME: this can fail because ofono qmi signals registered before being accepted by network. See OS#2458
-# assert not ms.is_connected(msc.mcc_mnc())
-assert not msc.subscriber_attached(ms)
-
-hlr.subscriber_delete(ms)
-print('KI changed: ' + bad_ki + " => " + good_ki)
-ms.set_ki(good_ki)
-hlr.subscriber_add(ms, ms.msisdn)
-print('Attempt connection with correct KI...')
-ms.connect(msc.mcc_mnc())
-wait(ms.is_connected, msc.mcc_mnc())
-wait(msc.subscriber_attached, ms)
+encryption_test_setup_run(True, 'a5_1')