From d95d494fcc1979299c19cd337fdfb21ae77739e6 Mon Sep 17 00:00:00 2001
From: Pau Espin Pedrol <pespin@espeweb.net>
Date: Mon, 21 Dec 2020 18:32:43 +0100
Subject: [PATCH] sysmocom: Port nitb_netreg_mass suite to osmo split network

Change-Id: I221df84d76f73e15eed98036b31adc8089c8dc99
---
 .../register_default_mass.py                  | 21 ++++++++++++++-----
 .../suite.conf                                |  3 +--
 2 files changed, 17 insertions(+), 7 deletions(-)
 rename sysmocom/suites/{nitb_netreg_mass => netreg_mass}/register_default_mass.py (83%)
 rename sysmocom/suites/{nitb_netreg_mass => netreg_mass}/suite.conf (71%)

diff --git a/sysmocom/suites/nitb_netreg_mass/register_default_mass.py b/sysmocom/suites/netreg_mass/register_default_mass.py
similarity index 83%
rename from sysmocom/suites/nitb_netreg_mass/register_default_mass.py
rename to sysmocom/suites/netreg_mass/register_default_mass.py
index f6b94ad5..ed411863 100644
--- a/sysmocom/suites/nitb_netreg_mass/register_default_mass.py
+++ b/sysmocom/suites/netreg_mass/register_default_mass.py
@@ -7,21 +7,32 @@ from osmo_gsm_tester.testenv import *
 from datetime import timedelta
 
 print('Claiming resources for the test')
-nitb = tenv.nitb()
+hlr = tenv.hlr()
+mgw_msc = tenv.mgw()
+mgw_bsc = tenv.mgw()
+stp = tenv.stp()
+msc = tenv.msc(hlr, mgw_msc, stp)
+bsc = tenv.bsc(msc, mgw_bsc, stp)
 bts = tenv.bts()
 ms_driver = tenv.ms_driver()
 ul = ms_driver.add_test('ul_test')
 modems = tenv.all_resources(tenv.modem)
 
 print('Launching a simple network')
-nitb.bts_add(bts)
-nitb.start()
+hlr.start()
+stp.start()
+msc.start()
+mgw_msc.start()
+mgw_bsc.start()
+
+bsc.bts_add(bts)
+bsc.start()
 bts.start()
-wait(nitb.bts_is_connected, bts)
+wait(bsc.bts_is_connected, bts)
 
 # Configure all MS that are available to this test.
 for modem in modems:
-    nitb.subscriber_add(modem)
+    hlr.subscriber_add(modem)
     ms_driver.subscriber_add(modem)
 
 # Run the base test.
diff --git a/sysmocom/suites/nitb_netreg_mass/suite.conf b/sysmocom/suites/netreg_mass/suite.conf
similarity index 71%
rename from sysmocom/suites/nitb_netreg_mass/suite.conf
rename to sysmocom/suites/netreg_mass/suite.conf
index 90e99460..68c43b71 100644
--- a/sysmocom/suites/nitb_netreg_mass/suite.conf
+++ b/sysmocom/suites/netreg_mass/suite.conf
@@ -1,9 +1,8 @@
 resources:
   ip_address:
-  - times: 1
+  - times: 6 # msc, bsc, hlr, stp, mgw*2
   bts:
   - type: osmo-bts-virtual
   modem:
   - times: 100
     type: osmo-mobile
-