X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=csit%2Flibraries%2FClusterAdmin.robot;h=19d55e529a137ef37ef1c14517e46b317b53b4f2;hb=e2c94012be3706110b79ccca3dcb5aceefacfbfb;hp=3de2c8c603f0b07895d0f43a69373a5f1b4e53df;hpb=a2de221e2deea2906885c463181dd7fff8c59c4c;p=integration%2Ftest.git

diff --git a/csit/libraries/ClusterAdmin.robot b/csit/libraries/ClusterAdmin.robot
index 3de2c8c603..19d55e529a 100644
--- a/csit/libraries/ClusterAdmin.robot
+++ b/csit/libraries/ClusterAdmin.robot
@@ -17,6 +17,8 @@ ${ADD_SHARD_REPLICA_DIR}    ${CLUSTERADMIN_RPC_DIR}/add_shard_replica
 ${MAKE_LEADER_LOCAL_DIR}    ${CLUSTERADMIN_RPC_DIR}/make_leader_local
 ${REMOVE_PREFIX_SHARD_REPLICA_DIR}    ${CLUSTERADMIN_RPC_DIR}/remove_prefix_shard_replica
 ${REMOVE_SHARD_REPLICA_DIR}    ${CLUSTERADMIN_RPC_DIR}/remove_shard_replica
+${GET_SHARD_ROLE_DIR}    ${CLUSTERADMIN_RPC_DIR}/get_shard_role
+${GET_PREFIX_SHARD_ROLE_DIR}    ${CLUSTERADMIN_RPC_DIR}/get_prefix_shard_role
 
 *** Keywords ***
 Make_Leader_Local
@@ -53,3 +55,23 @@ Remove_Shard_Replica
     ${session} =    ClusterManagement.Resolve_Http_Session_For_Member    member_index=${member_index}
     &{mapping}    BuiltIn.Create_Dictionary    SHARD_NAME=${shard_name}    MEMBER_NAME=${member_name}    DATA_STORE_TYPE=${ds_type}
     ${text} =    TemplatedRequests.Post_As_Xml_Templated    ${REMOVE_SHARD_REPLICA_DIR}    mapping=${mapping}    session=${session}
+
+Get_Shard_Role
+    [Arguments]    ${member_index}    ${shard_name}    ${ds_type}
+    [Documentation]    Get shard member role.
+    ${session} =    ClusterManagement.Resolve_Http_Session_For_Member    member_index=${member_index}
+    &{mapping}    BuiltIn.Create_Dictionary    SHARD_NAME=${shard_name}    DATA_STORE_TYPE=${ds_type}
+    ${text} =    TemplatedRequests.Post_As_Xml_Templated    ${GET_SHARD_ROLE_DIR}    mapping=${mapping}    session=${session}
+    ${xml} =    XML.Parse_Xml    ${text}
+    ${role} =    XML.Get_Element_Text    ${xml}    xpath=role
+    BuiltIn.Return_From_Keyword    ${role}
+
+Get_Prefix_Shard_Role
+    [Arguments]    ${member_index}    ${shard_prefix}    ${ds_type}
+    [Documentation]    Get prefix shard member role.
+    ${session} =    ClusterManagement.Resolve_Http_Session_For_Member    member_index=${member_index}
+    &{mapping}    BuiltIn.Create_Dictionary    SHARD_PREFIX=${shard_prefix}    DATA_STORE_TYPE=${ds_type}
+    ${text} =    TemplatedRequests.Post_As_Xml_Templated    ${GET_PREFIX_SHARD_ROLE_DIR}    mapping=${mapping}    session=${session}
+    ${xml} =    XML.Parse_Xml    ${text}
+    ${role} =    XML.Get_Element_Text    ${xml}    xpath=role
+    BuiltIn.Return_From_Keyword    ${role}