X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FSRv6.py;h=cafc4a075d19c7815e7f0809cbba4a7240182ab6;hp=a22bb5e8328045afac4ee3e1d420aa1be9e8b0aa;hb=03f1fa8fbf089b48365fff2bead6a2afb56efd89;hpb=92cbb44a89ca808df32e4a4cb137bed076a68a94

diff --git a/resources/libraries/python/SRv6.py b/resources/libraries/python/SRv6.py
index a22bb5e832..cafc4a075d 100644
--- a/resources/libraries/python/SRv6.py
+++ b/resources/libraries/python/SRv6.py
@@ -285,3 +285,21 @@ class SRv6(object):
         with VatTerminal(node, json_param=False) as vat:
             vat.vat_terminal_exec_cmd_from_template(
                 'srv6/sr_steer_policies_show.vat')
+
+    @staticmethod
+    def set_sr_encaps_source_address(node, ip6_addr):
+        """Set SRv6 encapsulation source address on the given node.
+
+        :param node: Given node to set SRv6 encapsulation source address on.
+        :param ip6_addr: Local SID IPv6 address.
+        :type node: dict
+        :type ip6_addr: str
+        """
+        with VatTerminal(node) as vat:
+            resp = vat.vat_terminal_exec_cmd_from_template(
+                'srv6/sr_set_encaps_source.vat', ip6_addr=ip6_addr)
+
+        VatJsonUtil.verify_vat_retval(
+            resp[0],
+            err_msg='Set SRv6 encapsulation source address {0} failed on node'
+                    ' {1}'.format(ip6_addr, node['host']))