This is so that config unit tests can pass

Signed-off-by: BSRK Aditya <[email protected]>
---
 test/py/testutils/config_mock.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/test/py/testutils/config_mock.py b/test/py/testutils/config_mock.py
index e994bda..48857cf 100644
--- a/test/py/testutils/config_mock.py
+++ b/test/py/testutils/config_mock.py
@@ -892,3 +892,29 @@ class ConfigMock(config.ConfigWriter):
 
   def AllocatePort(self):
     return 1
+
+  def Update(self, target, feedback_fn, ec_id=None):
+    def replace_in(target, tdict):
+      tdict[target.uuid] = target
+
+    update_serial = False
+    if isinstance(target, objects.Cluster):
+      self._ConfigData().cluster = target
+    elif isinstance(target, objects.Node):
+      replace_in(target, self._ConfigData().nodes)
+      update_serial = True
+    elif isinstance(target, objects.Instance):
+      replace_in(target, self._ConfigData().instances)
+    elif isinstance(target, objects.NodeGroup):
+      replace_in(target, self._ConfigData().nodegroups)
+    elif isinstance(target, objects.Network):
+      replace_in(target, self._ConfigData().networks)
+    elif isinstance(target, objects.Disk):
+      replace_in(target, self._ConfigData().disks)
+
+    target.serial_no += 1
+    target.mtime = now = time.time()
+
+    if update_serial:
+      self._ConfigData().cluster.serial_no += 1
+      self._ConfigData().cluster.mtime = now
-- 
2.2.0.rc0.207.ga3a616c

Reply via email to