@@ -17,6 +17,8 @@ limitations under the License.
1717package v1beta1
1818
1919import (
20+ "context"
21+
2022 . "github.com/onsi/ginkgo"
2123 . "github.com/onsi/gomega"
2224 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -51,55 +53,55 @@ var _ = Describe("ValidateCreate", func() {
5153 agentClassification .Spec .LabelKey = invalidKey
5254 agentClassification .Spec .LabelValue = validValue
5355 agentClassification .Spec .Query = validQuery
54- warn , err := agentClassification .ValidateCreate ()
56+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
5557 Expect (warn ).To (BeNil ())
5658 Expect (err ).NotTo (BeNil ())
5759 })
5860 It ("fails if label key has a prefix" , func () {
5961 agentClassification .Spec .LabelKey = keyWithPrefix
6062 agentClassification .Spec .LabelValue = validValue
6163 agentClassification .Spec .Query = validQuery
62- warn , err := agentClassification .ValidateCreate ()
64+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
6365 Expect (warn ).To (BeNil ())
6466 Expect (err ).NotTo (BeNil ())
6567 })
6668 It ("fails if label value is invalid" , func () {
6769 agentClassification .Spec .LabelKey = validKey
6870 agentClassification .Spec .LabelValue = invalidValue
6971 agentClassification .Spec .Query = validQuery
70- warn , err := agentClassification .ValidateCreate ()
72+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
7173 Expect (warn ).To (BeNil ())
7274 Expect (err ).NotTo (BeNil ())
7375 })
7476 It ("fails if label value starts with QUERYERROR" , func () {
7577 agentClassification .Spec .LabelKey = validKey
7678 agentClassification .Spec .LabelValue = "QUERYERROR-foo"
7779 agentClassification .Spec .Query = validQuery
78- warn , err := agentClassification .ValidateCreate ()
80+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
7981 Expect (warn ).To (BeNil ())
8082 Expect (err ).NotTo (BeNil ())
8183 })
8284 It ("fails if label query is invalid" , func () {
8385 agentClassification .Spec .LabelKey = validKey
8486 agentClassification .Spec .LabelValue = validValue
8587 agentClassification .Spec .Query = invalidQuery
86- warn , err := agentClassification .ValidateCreate ()
88+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
8789 Expect (warn ).To (BeNil ())
8890 Expect (err ).NotTo (BeNil ())
8991 })
9092 It ("fails if everything is invalid" , func () {
9193 agentClassification .Spec .LabelKey = invalidKey
9294 agentClassification .Spec .LabelValue = invalidValue
9395 agentClassification .Spec .Query = invalidQuery
94- warn , err := agentClassification .ValidateCreate ()
96+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
9597 Expect (warn ).To (BeNil ())
9698 Expect (err ).NotTo (BeNil ())
9799 })
98100 It ("succeeds if everything is valid" , func () {
99101 agentClassification .Spec .LabelKey = validKey
100102 agentClassification .Spec .LabelValue = validValue
101103 agentClassification .Spec .Query = validQuery
102- warn , err := agentClassification .ValidateCreate ()
104+ warn , err := agentClassification .ValidateCreate (context . TODO (), agentClassification )
103105 Expect (warn ).To (BeNil ())
104106 Expect (err ).To (BeNil ())
105107 })
@@ -127,7 +129,7 @@ var _ = Describe("ValidateUpdate", func() {
127129 oldAgentClassification .Spec .Query = validQuery
128130 newAgentClassification := oldAgentClassification .DeepCopy ()
129131 newAgentClassification .Spec .LabelKey = "newKey"
130- warn , err := newAgentClassification .ValidateUpdate (oldAgentClassification )
132+ warn , err := newAgentClassification .ValidateUpdate (context . TODO (), oldAgentClassification , newAgentClassification )
131133 Expect (warn ).To (BeNil ())
132134 Expect (err ).NotTo (BeNil ())
133135 })
@@ -137,17 +139,7 @@ var _ = Describe("ValidateUpdate", func() {
137139 oldAgentClassification .Spec .Query = validQuery
138140 newAgentClassification := oldAgentClassification .DeepCopy ()
139141 newAgentClassification .Spec .LabelValue = "newvalue"
140- warn , err := newAgentClassification .ValidateUpdate (oldAgentClassification )
141- Expect (warn ).To (BeNil ())
142- Expect (err ).NotTo (BeNil ())
143- })
144- It ("fails if label query is changed" , func () {
145- oldAgentClassification .Spec .LabelKey = validKey
146- oldAgentClassification .Spec .LabelValue = validValue
147- oldAgentClassification .Spec .Query = ".cpu.count == 2"
148- newAgentClassification := oldAgentClassification .DeepCopy ()
149- newAgentClassification .Spec .Query = validQuery
150- warn , err := newAgentClassification .ValidateUpdate (oldAgentClassification )
142+ warn , err := newAgentClassification .ValidateUpdate (context .TODO (), oldAgentClassification , newAgentClassification )
151143 Expect (warn ).To (BeNil ())
152144 Expect (err ).NotTo (BeNil ())
153145 })
0 commit comments