Migrate LISP tests to new Beryllium API

[integration/test.git] / csit / libraries / LISPFlowMapping.robot

diff --git a/csit/libraries/LISPFlowMapping.robot b/csit/libraries/LISPFlowMapping.robot
index 6f15ad4857c7e499f21b78babcbf99b3e229815b..6ee0ab481267064e2e92db592327156419d34d21 100644 (file)
--- a/csit/libraries/LISPFlowMapping.robot
+++ b/csit/libraries/LISPFlowMapping.robot
@@ -8,6 +8,76 @@ Documentation     This resource file defines keywords that are used in more
 ${ODL_VERSION}    Be
 
 *** Keywords ***
+Authentication Key Should Be
+    [Arguments]    ${resp}    ${password}
+    [Documentation]    Check if the authentication key in the ${resp} is ${password}
+    ${authkey}=    Get Authentication Key    ${resp}
+    Should Be Equal As Strings    ${authkey}    ${password}
+
+Get Authentication Key
+    [Arguments]    ${resp}
+    ${output}=    Get From Dictionary    ${resp.json()}    output
+    ${authkey}=    Run Keyword If    "${ODL_VERSION}" == "Li"    Get Authentication Key Lithium    ${output}
+    ...    ELSE    Get Authentication Key Beryllium    ${output}
+    [Return]    ${authkey}
+
+Get Authentication Key Beryllium
+    [Arguments]    ${output}
+    ${mapping_authkey}=    Get From Dictionary    ${output}    mapping-authkey
+    ${authkey}=    Get From Dictionary    ${mapping_authkey}    key-string
+    [Return]    ${authkey}
+
+Get Authentication Key Lithium
+    [Arguments]    ${output}
+    ${authkey}=    Get From Dictionary    ${output}    authkey
+    [Return]    ${authkey}
+
+Ipv4 Rloc Should Be
+    [Arguments]    ${resp}    ${address}
+    [Documentation]    Check if the RLOC in the ${resp} is ${address}
+    ${eid_record}=    Get Eid Record    ${resp}
+    ${loc_record}=    Get From Dictionary    ${eid_record}    LocatorRecord
+    ${loc_record_0}=    Get From List    ${loc_record}    0
+    ${ipv4}=    Get Ipv4 Rloc    ${loc_record_0}
+    Should Be Equal As Strings    ${ipv4}    ${address}
+
+Get Eid Record
+    [Arguments]    ${resp}
+    ${output}=    Get From Dictionary    ${resp.json()}    output
+    ${eid_record}=    Run Keyword If    "${ODL_VERSION}" == "Li"    Get Eid Record Lithium    ${output}
+    ...    ELSE    Get Eid Record Beryllium    ${output}
+    [Return]    ${eid_record}
+
+Get Eid Record Beryllium
+    [Arguments]    ${output}
+    ${eid_record}=    Get From Dictionary    ${output}    mapping-record
+    [Return]    ${eid_record}
+
+Get Eid Record Lithium
+    [Arguments]    ${output}
+    ${eid_record}=    Get From Dictionary    ${output}    eidToLocatorRecord
+    ${eid_record}=    Get From List    ${eid_record}    0
+    [Return]    ${eid_record}
+
+Get Ipv4 Rloc
+    [Arguments]    ${loc_record}
+    ${ipv4}=    Run Keyword If    "${ODL_VERSION}" == "Li"    Get Ipv4 Rloc Lithium    ${loc_record}
+    ...    ELSE    Get Ipv4 Rloc Beryllium    ${loc_record}
+    [Return]    ${ipv4}
+
+Get Ipv4 Rloc Beryllium
+    [Arguments]    ${loc_record}
+    ${loc}=    Get From Dictionary    ${loc_record}    rloc
+    ${ipv4}=    Get From Dictionary    ${loc}    ipv4
+    [Return]    ${ipv4}
+
+Get Ipv4 Rloc Lithium
+    [Arguments]    ${loc_record}
+    ${loc}=    Get From Dictionary    ${loc_record}    LispAddressContainer
+    ${address}=    Get From Dictionary    ${loc}    Ipv4Address
+    ${ipv4}=    Get From Dictionary    ${address}    Ipv4Address
+    [Return]    ${ipv4}
+
 Check Mapping Removal
     [Arguments]    ${json}
     Run Keyword If    "${ODL_VERSION}" == "Li"    Check Mapping Removal Lithium    ${json}