Introduce osmux support for msc,bsc,mgw

Change-Id: Ief3518e77650f2cddb48357b07fe95346e46ee1b
diff --git a/src/osmo_gsm_tester/osmo_mgw.py b/src/osmo_gsm_tester/osmo_mgw.py
index ca7bd51..ea8e652 100644
--- a/src/osmo_gsm_tester/osmo_mgw.py
+++ b/src/osmo_gsm_tester/osmo_mgw.py
@@ -31,6 +31,7 @@
         self.process = None
         self.suite_run = suite_run
         self.ip_address = ip_address
+        self.use_osmux = "off"
 
     def start(self):
         self.log('Starting osmo-mgw')
@@ -63,7 +64,11 @@
 
         values = dict(mgw=config.get_defaults('mgw'))
         config.overlay(values, self.suite_run.config())
-        config.overlay(values, dict(mgw=dict(ip_address=self.ip_address)))
+        config.overlay(values, { 'mgw': {
+                                        'ip_address': self.ip_address,
+                                        'use_osmux': self.use_osmux
+                                        }
+                                })
 
         self.dbg('MGW CONFIG:\n' + pprint.pformat(values))
 
@@ -81,4 +86,14 @@
     def running(self):
         return not self.process.terminated()
 
+    def set_use_osmux(self, use=False, force=False):
+        if not use:
+            self.use_osmux = "off"
+        else:
+            if not force:
+                self.use_osmux = "on"
+            else:
+                self.use_osmux = "only"
+
+
 # vim: expandtab tabstop=4 shiftwidth=4