From 93f1dfa205ca76c0eb2f0fe3100a03b5700ffbf6 Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Mon, 28 Jul 2025 20:32:39 +0000
Subject: [PATCH] Add test for SR Sid Spotlight configuration and stats
 management

Co-authored-by: nilaa567.murugaa <nilaa567.murugaa@gmail.com>
---
 .../Valimar/Stats/TEST_Stats_Spotlight_Sid.py | 266 ++++++++++++++++++
 .../TEST_Stats_Spotlight_Sid.cpython-313.pyc  | Bin 0 -> 12295 bytes
 2 files changed, 266 insertions(+)
 create mode 100644 tests/Valimar/Stats/TEST_Stats_Spotlight_Sid.py
 create mode 100644 tests/Valimar/Stats/__pycache__/TEST_Stats_Spotlight_Sid.cpython-313.pyc

diff --git a/tests/Valimar/Stats/TEST_Stats_Spotlight_Sid.py b/tests/Valimar/Stats/TEST_Stats_Spotlight_Sid.py
new file mode 100644
index 0000000..cf3ae54
--- /dev/null
+++ b/tests/Valimar/Stats/TEST_Stats_Spotlight_Sid.py
@@ -0,0 +1,266 @@
+import time
+from .Stats_Common import Stats_CommonTestCase
+from .Stats_Common import (
+    STUB_srteSensorConfigCliCreate,
+    STUB_srteSensorConfigCliDelete,
+)
+from utilities.commonPredicates import checkTrue
+from utilities.prereq import Prereq
+import utilities.SR.Predicates as SR_Preds
+
+
+class TEST_Stats_Spotlight_Sid(Stats_CommonTestCase):
+    """
+    SR Sid Spotlight.
+    """
+
+    def testSetup(self):
+        super().testSetup()
+
+        for d in self.ALL_DEVICES:
+            d.stubEnable(STUB_srteSensorConfigCliCreate)
+            d.stubEnable(STUB_srteSensorConfigCliDelete)
+
+    def testExecute(self):
+        algo0_labels = [16001, 16002, 16003]
+        flex_labels = [17002, 17003, 18002, 18003, 19002, 19003]
+        adj_labels = [32000, 32001]
+
+        with self.testlog.step(
+            "Enable spotlight+collection on algo0 range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "16000" "16999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Disable spotlight on algo0 range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "16000" "16999" stats-spotlight-state off"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Re-Enable spotlight on algo0 range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "16000" "16999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Remove the algo0 range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """no segment-routing sid-stats-range "16000" "16999" """
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Enable spotlight+collection on flex range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "17000" "19999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Disable spotlight on flex range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "17000" "19999" stats-spotlight-state off"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Re-Enable spotlight on flex range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "17000" "19999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Remove the flex range",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """no segment-routing sid-stats-range "17000" "19999" """
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Enable spotlight+collection on adjacency SID labels",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing adj-sid-stats stats-collection on stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Disable spotlight on adjacency SID labels",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing adj-sid-stats stats-spotlight-state off"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Re-Enable spotlight on adjacency SID labels",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing adj-sid-stats stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Remove adjacency SID spotlight configurations",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig(
+                """no segment-routing adj-sid-stats stats-spotlight-state"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Re-Enable spotlight on adjacency SID labels and remove adjacency SID configurations",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing adj-sid-stats stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.sendConfig("""no segment-routing adj-sid-stats""")
+            self.sanityCheck(devices=[self.UUT])
+
+        with self.testlog.step(
+            "Re-Enable spotlight on SIDs and remove all segment-routing configurations",
+            pass_on_exit=True,
+            reraise_ex=True,
+        ):
+            # Re-enable spotlight on algo0 labels (16000-16999 range)
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "16000" "16999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+            
+            # Re-enable spotlight on flex labels (17000-19999 range)
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing sid-stats-range "17000" "19999" stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+            
+            # Re-enable spotlight on adjacency SID labels
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliCreate, f"0 1")
+            self.UUT.sendConfig(
+                """segment-routing adj-sid-stats stats-collection on stats-spotlight-state on"""
+            )
+            self.sanityCheck(devices=[self.UUT])
+            
+            # Now use "no segment-routing" to disable all spotlight configurations
+            # This should delete all the sensors that were created above
+            
+            # Expect deletion of algo0 labels
+            for l in algo0_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            
+            # Expect deletion of flex labels
+            for l in flex_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            # no labels alloc'ed for FA local, but stats still enabled
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            
+            # Expect deletion of adjacency SID labels
+            for l in adj_labels:
+                self.UUT.stubExpect(STUB_srteSensorConfigCliDelete, f"0 1")
+            
+            # Execute the "no segment-routing" command to disable all segment-routing features
+            self.UUT.sendConfig("""no segment-routing""")
+            self.sanityCheck(devices=[self.UUT])
+
+    def testCleanup(self):
+        for d in self.ALL_DEVICES:
+            d.stubDisable(STUB_srteSensorConfigCliCreate)
+            d.stubDisable(STUB_srteSensorConfigCliDelete)
+
+        super().testCleanup()
\ No newline at end of file
diff --git a/tests/Valimar/Stats/__pycache__/TEST_Stats_Spotlight_Sid.cpython-313.pyc b/tests/Valimar/Stats/__pycache__/TEST_Stats_Spotlight_Sid.cpython-313.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..d41d92ab3cdba2659fcdf6873ede28c84ea8691c
GIT binary patch
literal 12295
zcmeHNU2GHC6~5!KJ(Ka+B!PsGKs@<bEe_;|<u@#_n*??t0jtf#t+Y_4#-2C>;~DRr
z$qFpBqOH`DC$>_z?L!~7Pc855`>I-1eW+AWQ;24FyQqCaDi8>iRiAq9j1&K45<7Ol
zTd8B~{>+?v&$-_@_s-mV&QdVQA-MMc{U<Y9_8|06Qdz!yxx&&66h1;C5~&G<k?0ZW
z32NMfJpw6vp-f|X+>5<XN1|`SH}1!NyX>D}##zkTWon{foHL-!cmVSPYD3*fWZIF)
zUho{G9Zooyq9P3sNG1A!f;2)|+e9;N8q%1aPV3r~Y?xzlLymZ?PH`$WDjC?6MOia+
zJf>^YN@`41#;_bW<+Y!mlvUZ3q34FgjGVYKg=a~{{|zi-`Tc|=h?9D9m8BC<xCjVb
zK$sE{_J|awMGyAI&;d%M+tFnf`$TU$!hZXX&%VQme$2+uaSAu|q6lNPO`Q^_Bxf{Y
zMmJR@HDgMml6=5J_t~wSI4M9)Sa~+UB?tz~mlp~Ir=?#(a)SB@onL-}raUq99njXn
z+#LbM&?KQRxwfLy{RnAPjGAOh9zpdyyL|6pjxY4%PHEc=q0UpD8$>981e8LzsA$A*
z`Hb0&jI9O}R3^&iY{m+VoINX@JoT+JW2Z!mGtAjbr?mJbRmQ~5upg3JwB<Eqb=snm
zR)ZuZ)VN_tl0il-2-mHSwGbT0T*F&{$m(8`VnvK5pfip@^3`?pIUAf~?yx)N*^c{7
zE%%zb=9{{1j4bfMn`h_F&hI!p&mYNG?(#>T`cQNC60}7{njxawnz`6U=E)$;lW$TQ
z1-S=um@T4r(4*0y6VkNgnj|etF3~6Y;l~Wq9zZlkf!=yRZ(wh3T3uh4B6~1r#{xR{
znxmsN`U#YX0adXfuVH8vTutHL*B~Ww!<0SmVNJ+O0Mx6yK&}`BYQ`Y6%&JBV__|>b
ztQmuqIb9<L^|OOe%@}N!9c-za9W)kU;CV&s!Mg8=LVI#w11evaUQ&Cv^{lCGI&7|4
zht=4zn+{$3*?<mSWqUUrZmpXRx7Dn}YSv>n9lG|j0Uf@|_HH_?-#l!oS%<FnU*_A&
zO^2@iY(R&vvb~!QTkAFtw-@P<ensy?Hw@b9hQW@SF>tk?GIs2SL3`aW=%^V3*LAQ8
z40hHHgIzUauvvD{SvL&AHDmBKv4dDOYuRS$PpF&zx{CDoKYZR1`~Thy4qYnP8@jzm
zYWyawYVK^-ysz;)psF*z_2+#xmbiY8m7hMk_EYyS+j_38-J83%rcf8!i+1M@IFoQ*
zKkX4v7*2}?^ez|u-ElgMrm1ylPL`!<h|X6)lPpWaMt@e8hKYV!mc}1_Rsl_KSsGsj
z`1qpb@$puGk2hK#A3FN;ax$loLlNnTMrh>+;E+;&NRl5x67egSN1=R$B$MwzB9i3j
zI3#b91bq(@3X*J}<8cv;W&hKjgs!S`!c=rkfPY*~=|cjJYbkjOd~AG%IAC;5lCLVJ
z#es(`t{5`Zvh<L!FMHaMQ)yW<`>{T2Dq2b~lw`j_+;#nS7ee>GLqkJD-6S6#9_|*L
zXDiHYS{e9)|BXq?A1Ddg$O<PFgLsAVGL{IIJ+nz{r>C>slX8D4B&F<HC*`#Mfh?Fa
zav?t1{hBUpRFZ6@lv{}crd9c>gMTIH9?qc);I4yiHtY&%0jmlSt%wyM+O-Cv1>9N+
z04=y(8J@**WcQcBGoHL0Psm#0njoGzDX8&FvT9_<OVtKc{VTScv(n;;v6kg#Bd*xG
zvOQc0+f|bTFx|5TrmkELRmDZi_UE)x-06xqLc;d1&*C^)D~#+R7wdO}ais%ivS-B&
zj75lRNdY^vuUM=SVT5wRWY4U@7OcW3Vxy|7CUpt3=GDw1ti=!qv#O`Sfo;kei;l&n
zaz5^>8CYL!A2-~PwWJfMU<Hi0rkK~pNDPAQ{EqnWR(PLVyuAoZj;&e&GDDUt9H6jV
zrc{IY%>{DdR?;v)L^yR-PRv61L+`qt@EjGmtC7(Q&iw1h+P?P&Z?(RCBe2lg@ypnU
z7jJO)gH1Ot%w4z_?7S1~TmV=2j$I4eI~Exf3O_~O5Vypk*6t6fg~n~aV1LTqYwWqx
z*fZZd`bpw$<HQphp55mgZ=RkzeUA^{;lm4!7oX9dU}({gf=&1Mop<=1A18k^^Xr+9
z-@8p}LoZsO&&4(rYA)nDP%Brd8m>dNa;<8ty|r>ZQ_WZpyK(JWN3UJB3iFzs7fyWA
zGta-fP^j+8Z<aTVUBobi62s`PyV&I{%T?^bx{F=TG+o8+sk_+aY}-}r5jU}gb!=NG
zS+l;oX)|ggT+da*b-32Co~?%Kn|0wDf5}`wYYA6uB)5fkoW;-erRHju-(Dp?zp1a%
zB|mEYI<4SDo{iXlnOV0~4y`5jWskq?h!$D+%*nqaZwp7}`C)iNx*AEEB5yeIP4yz~
zgY@5|_@5VT8z>5S+Ooxs_WRrx$M`*bE&g^StRA5M&^HoJ`{_^jLe@tAaR9PD`cFfn
zZ>5jXe{LRqOAFC|IRKeJ|FspeH|V=ZM#8$6UN{O_D{UQvY&ZS+3CNDpU!u|D8J_-!
zFdoQs(*HWZrD^7&pG&te58Jr(>&(MGE<MaVJjQ7O=23{#b}^3xP8(z%y}{`e^VrMj
zTbakLoc<c~csHjXW*#5qG7Zck&t*E8#ZE4>mnjO?TopzO$C5Cx(yJg;)+Ipm{7P_b
z!Lc6Y7RGp<1c%yJo^m-<HXm?n2k^EO2Jog3Cc%~wKW>B!b_)<Lo7-yU<7F{9XLY{&
z$5c74&1P^j+*uBowYLR$520CqJCJMagJVSFD3C*F>_Txo?76vf^X-la2X8u+yZrDI
z5)b=~MGYYmM%(c>b{8gFid!hcS*#>!@w6;S7AHw*JvpnAGA~K*&&Ji<lP!`otzg(U
zDVnV5@R*ULq@Iu@JDL~24t+V|w?GsyoGU2_DQ=orxdb~L2n@D@xftPutg4a}p)f#*
z3>XD^+Zj8FQ^q7AA=XAGBwzg)J@I&a>~q%V8-5zPLi_f;c#rY<I+qY6w)l2~NXYV<
zN?Nvfu^|2&6TRp0x7y*{OI1uoHU<)QfI38xCzS-mpBtD&d{~Wnf{Yztj)}v;+Md5H
zP7dT7SsroH;tJ}lI4O~;VGD&JdK)=1+B-h>m;>wyd>kH-aT|nUk)|l>GZgp&4ZQHs
il;=5%sC{3co-a|~ZPfS7|9zV3ydB>E9Fc<E-G2epJam5m

literal 0
HcmV?d00001