glusterd test case: validating add brick functionality

In this test case,
1. adding a single brick to replicated volume
2. adding non existing bricks to volume
3. adding bricks from node which is not a part of cluster
4. triggering rebalance start after add brick are validated.

Change-Id: I982ff42dcbe6cd0cfbf3653b8cee0b269314db3f
Signed-off-by: Sanju Rakonde <srakonde@redhat.com>

1 files changed, 134 insertions, 0 deletions

diff --git a/tests/functional/glusterd/test_add_brick_functionality.py b/tests/functional/glusterd/test_add_brick_functionality.py
new file mode 100644
index 000000000..bd6ce3ea0
--- /dev/null
+++ b/tests/functional/glusterd/test_add_brick_functionality.py
@@ -0,0 +1,134 @@
+#  Copyright (C) 2016-2017  Red Hat, Inc. <http://www.redhat.com>
+#
+#  This program is free software; you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation; either version 2 of the License, or
+#  any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License along
+#  with this program; if not, write to the Free Software Foundation, Inc.,
+#  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+from glusto.core import Glusto as g
+from glustolibs.gluster.gluster_base_class import GlusterBaseClass, runs_on
+from glustolibs.gluster.exceptions import ExecutionError
+from glustolibs.gluster.volume_libs import setup_volume, cleanup_volume
+from glustolibs.gluster.volume_ops import (get_volume_list)
+from glustolibs.gluster.brick_ops import add_brick
+from glustolibs.gluster.lib_utils import form_bricks_list
+from glustolibs.gluster.rebalance_ops import rebalance_start
+import random
+
+
+@runs_on([['distributed-replicated'], ['glusterfs']])
+class TestVolumeCreate(GlusterBaseClass):
+
+    @classmethod
+    def setUpClass(cls):
+
+        # Calling GlusterBaseClass setUpClass
+        GlusterBaseClass.setUpClass.im_func(cls)
+
+        # check whether peers are in connected state
+        ret = cls.validate_peers_are_connected()
+        if not ret:
+            raise ExecutionError("Peers are not in connected state")
+
+    def tearDown(self):
+
+        # clean up all volumes
+        vol_list = get_volume_list(self.mnode)
+        if vol_list is None:
+            raise ExecutionError("Failed to get the volume list")
+
+        for volume in vol_list:
+            ret = cleanup_volume(self.mnode, volume)
+            if not ret:
+                raise ExecutionError("Unable to delete volume % s" % volume)
+            g.log.info("Volume deleted successfully : %s" % volume)
+
+        GlusterBaseClass.tearDown.im_func(self)
+
+    def test_add_brick_functionality(self):
+
+        ret = setup_volume(self.mnode, self.all_servers_info, self.volume)
+        self.assertTrue(ret, ("Failed to create and start volume %s"
+                        % self.volname))
+        g.log.info("Volume created and started succssfully")
+
+        # form bricks list to test add brick functionality
+
+        replica_count_of_volume = self.volume['voltype']['replica_count']
+        num_of_bricks = 4 * replica_count_of_volume
+        bricks_list = form_bricks_list(self.mnode, self.volname,
+                                       num_of_bricks, self.servers,
+                                       self.all_servers_info)
+        self.assertIsNotNone(bricks_list, "Bricks list is None")
+
+        # Try to add a single brick to volume, which should fail as it is a
+        # replicated volume, we should pass multiple of replica count number
+        # of bricks
+
+        bricks_list_to_add = [bricks_list[0]]
+        ret, out, err = add_brick(self.mnode, self.volname, bricks_list_to_add)
+        self.assertNotEqual(ret, 0, "Expected: It should fail to add a single"
+                            "brick to a replicated volume. Actual: "
+                            "Successfully added single brick to volume")
+        g.log.info("failed to add a single brick to replicated volume")
+
+        # add brick replica count number of bricks in which one is
+        # non existing brick
+        kwargs = {}
+        kwargs['replica_count'] = replica_count_of_volume
+
+        bricks_list_to_add = bricks_list[1:replica_count_of_volume + 1]
+
+        num_of_bricks = len(bricks_list_to_add)
+        index_of_non_existing_brick = random.randint(0, num_of_bricks - 1)
+        complete_brick = bricks_list_to_add[index_of_non_existing_brick]
+        non_existing_brick = complete_brick + "/non_existing_brick"
+        bricks_list_to_add[index_of_non_existing_brick] = non_existing_brick
+
+        ret, out, err = add_brick(self.mnode, self.volname,
+                                  bricks_list_to_add, False, **kwargs)
+        self.assertNotEqual(ret, 0, "Expected: It should fail to add non"
+                            "existing brick to a volume. Actual: "
+                            "Successfully added non existing brick to volume")
+        g.log.info("failed to add a non existing brick to volume")
+
+        # adding brick from node which is not part of cluster
+        bricks_list_to_add = bricks_list[replica_count_of_volume + 1:
+                                         (2 * replica_count_of_volume) + 1]
+
+        num_of_bricks = len(bricks_list_to_add)
+        index_of_node = random.randint(0, num_of_bricks - 1)
+        complete_brick = bricks_list_to_add[index_of_node].split(":")
+        complete_brick[0] = "abc.def.ghi.jkl"
+        bricks_list_to_add[index_of_node] = ":".join(complete_brick)
+        ret, out, err = add_brick(self.mnode, self.volname,
+                                  bricks_list_to_add, False, **kwargs)
+        self.assertNotEqual(ret, 0, "Expected: It should fail to add brick "
+                            "from a node which is not part of a cluster."
+                            "Actual:Successfully added bricks from node which"
+                            " is not a part of cluster to volume")
+
+        g.log.info("Failed to add bricks form node which is not a part of "
+                   "cluster to volume")
+
+        # add correct number of valid bricks, it should succeed
+
+        bricks_list_to_add = bricks_list[(2 * replica_count_of_volume) + 1:
+                                         (3 * replica_count_of_volume) + 1]
+        ret, out, err = add_brick(self.mnode, self.volname,
+                                  bricks_list_to_add, False, **kwargs)
+        self.assertEqual(ret, 0, "Failed to add the bricks to the volume")
+        g.log.info("Successfully added bricks to volume")
+
+        # Perform rebalance start operation
+        ret, out, err = rebalance_start(self.mnode, self.volname)
+        self.assertEqual(ret, 0, "Rebalance start is success")