MCOL-5496: Merge CMAPI code to engine repo.

[add] cmapi code to engine

cmapi/cmapi_server/test/test_failover_agent.py

@@ -0,0 +1,124 @@
+import logging
+import socket
+
+from cmapi_server.failover_agent import FailoverAgent
+from cmapi_server.node_manipulation import add_node, remove_node
+from mcs_node_control.models.node_config import NodeConfig
+from cmapi_server.test.unittest_global import (
+    tmp_mcs_config_filename, BaseNodeManipTestCase
+)
+
+
+logging.basicConfig(level='DEBUG')
+
+
+class TestFailoverAgent(BaseNodeManipTestCase):
+
+    def test_activateNodes(self):
+        self.tmp_files = ('./activate0.xml', './activate1.xml')
+        hostaddr = socket.gethostbyname(socket.gethostname())
+        fa = FailoverAgent()
+        fa.activateNodes(
+            [self.NEW_NODE_NAME], tmp_mcs_config_filename, self.tmp_files[0],
+            test_mode=True
+        )
+        add_node(
+            hostaddr, self.tmp_files[0], self.tmp_files[1]
+        )
+
+        nc = NodeConfig()
+        root = nc.get_current_config_root(self.tmp_files[1])
+        pm_count = int(root.find('./PrimitiveServers/Count').text)
+        self.assertEqual(pm_count, 2)
+        node = root.find('./PMS1/IPAddr')
+        self.assertEqual(node.text, self.NEW_NODE_NAME)
+        node = root.find('./pm1_WriteEngineServer/IPAddr')
+        self.assertEqual(node.text, self.NEW_NODE_NAME)
+        node = root.find('./PMS2/IPAddr')
+        self.assertEqual(node.text, hostaddr)
+        node = root.find('./pm2_WriteEngineServer/IPAddr')
+        self.assertEqual(node.text, hostaddr)
+        remove_node(self.NEW_NODE_NAME, self.tmp_files[1], self.tmp_files[1])
+
+    def test_deactivateNodes(self):
+        self.tmp_files = (
+            './deactivate0.xml','./deactivate1.xml', './deactivate2.xml'
+        )
+        fa = FailoverAgent()
+        hostname = socket.gethostname()
+        hostaddr = socket.gethostbyname(hostname)
+        add_node(
+            hostaddr, tmp_mcs_config_filename, self.tmp_files[0]
+        )
+        fa.activateNodes(
+            [self.NEW_NODE_NAME], self.tmp_files[0], self.tmp_files[1],
+            test_mode=True
+        )
+        fa.deactivateNodes(
+            [self.NEW_NODE_NAME], self.tmp_files[1], self.tmp_files[2],
+            test_mode=True
+        )
+
+        nc = NodeConfig()
+        root = nc.get_current_config_root(self.tmp_files[2])
+        pm_count = int(root.find('./PrimitiveServers/Count').text)
+        self.assertEqual(pm_count, 1)
+        node = root.find('./PMS1/IPAddr')
+        self.assertEqual(node.text, hostaddr)
+        # TODO: Fix node_manipulation add_node logic and _replace_localhost
+        # node = root.find('./PMS2/IPAddr')
+        # self.assertEqual(node, None)
+        node = root.find('./pm1_WriteEngineServer/IPAddr')
+        self.assertTrue(node.text, hostaddr)
+        node = root.find('./pm2_WriteEngineServer/IPAddr')
+        self.assertTrue(node is None)
+        #node = root.find("./ConfigRevision")
+        #self.assertEqual(node.text, "3")
+
+        # make sure there are no traces of mysql.com,
+        # or an ip addr that isn't localhost or 127.0.0.1
+        all_nodes = root.findall('./')
+        for node in all_nodes:
+            self.assertFalse(node.text == self.NEW_NODE_NAME)
+            if node.tag in ['IPAddr', 'Node']:
+                self.assertTrue(node.text in [hostname, hostaddr])
+
+    def test_designatePrimaryNode(self):
+        self.tmp_files = (
+            './primary-node0.xml', './primary-node1.xml', './primary-node2.xml'
+        )
+        fa = FailoverAgent()
+        hostaddr = socket.gethostbyname(socket.gethostname())
+        fa.activateNodes(
+            [self.NEW_NODE_NAME], tmp_mcs_config_filename, self.tmp_files[0],
+            test_mode=True
+        )
+        add_node(
+            hostaddr, self.tmp_files[0], self.tmp_files[1]
+        )
+        fa.movePrimaryNode(
+            'placeholder', self.tmp_files[1], self.tmp_files[2], test_mode=True
+        )
+
+        nc = NodeConfig()
+        root = nc.get_current_config_root(self.tmp_files[2])
+        pm_count = int(root.find('./PrimitiveServers/Count').text)
+        self.assertEqual(pm_count, 2)
+        node = root.find('./PMS1/IPAddr')
+        self.assertEqual(node.text, self.NEW_NODE_NAME)
+        node = root.find('./PMS2/IPAddr')
+        self.assertEqual(node.text, hostaddr)
+        node = root.find('./pm1_WriteEngineServer/IPAddr')
+        self.assertEqual(node.text, self.NEW_NODE_NAME)
+        node = root.find('./pm2_WriteEngineServer/IPAddr')
+        self.assertEqual(node.text, hostaddr)
+
+        for tag in ['ExeMgr1', 'DMLProc', 'DDLProc']:
+            node = root.find(f'./{tag}/IPAddr')
+            self.assertEqual(node.text, self.NEW_NODE_NAME)
+
+        self.assertEqual(self.NEW_NODE_NAME, root.find('./PrimaryNode').text)
+
+    def test_enterStandbyMode(self):
+        fa = FailoverAgent()
+        fa.enterStandbyMode(test_mode=True)