Skip to content

Commit

Permalink
Fix broken tests
Browse files Browse the repository at this point in the history
Signed-off-by: Max Smythe <smythe@google.com>
  • Loading branch information
maxsmythe committed Jan 24, 2023
1 parent 3ce6b43 commit 2ecccf2
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -579,7 +579,7 @@ violation[{"msg": "denied!"}] {
}

// Set up tracker
tracker, err := readiness.SetupTracker(mgr, false, false)
tracker, err := readiness.SetupTrackerNoReadyz(mgr, false, false)
if err != nil {
t.Fatal(err)
}
Expand Down
19 changes: 15 additions & 4 deletions pkg/readiness/setup.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,21 @@ import (
// provided Manager object.
// NOTE: Must be called _before_ the manager is started.
func SetupTracker(mgr manager.Manager, mutationEnabled bool, externalDataEnabled bool) (*Tracker, error) {
tracker, err := SetupTrackerNoReadyz(mgr, mutationEnabled, externalDataEnabled)
if err != nil {
return nil, err
}

if err := mgr.AddReadyzCheck("tracker", tracker.CheckSatisfied); err != nil {
return nil, fmt.Errorf("registering readiness check: %w", err)
}

return tracker, nil
}

// SetupTracker sets up a readiness tracker and registers it to run under control of the
// provided Manager object without instantiating /readyz (used for testing).
func SetupTrackerNoReadyz(mgr manager.Manager, mutationEnabled bool, externalDataEnabled bool) (*Tracker, error) {
tracker := NewTracker(mgr.GetAPIReader(), mutationEnabled, externalDataEnabled)

err := mgr.Add(manager.RunnableFunc(func(ctx context.Context) error {
Expand All @@ -35,9 +50,5 @@ func SetupTracker(mgr manager.Manager, mutationEnabled bool, externalDataEnabled
return nil, fmt.Errorf("adding tracker to manager: %w", err)
}

if err := mgr.AddReadyzCheck("tracker", tracker.CheckSatisfied); err != nil {
return nil, fmt.Errorf("registering readiness check: %w", err)
}

return tracker, nil
}
8 changes: 4 additions & 4 deletions pkg/target/target_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ func TestValidateConstraint(t *testing.T) {
"matchExpressions": [{
"key": "someKey",
"operator": "NotIn",
"values": ["some value"]
"values": ["some-value"]
}]
}
},
Expand Down Expand Up @@ -130,7 +130,7 @@ func TestValidateConstraint(t *testing.T) {
ErrorExpected: true,
},
{
Name: "Invalid LabelSelector MatchLels",
Name: "Invalid LabelSelector MatchLabels",
Constraint: `
{
"apiVersion": "constraints.gatekeeper.sh/v1beta1",
Expand Down Expand Up @@ -212,7 +212,7 @@ func TestValidateConstraint(t *testing.T) {
"matchExpressions": [{
"key": "someKey",
"operator": "In",
"values": ["some value"]
"values": ["some-value"]
}]
}
},
Expand Down Expand Up @@ -245,7 +245,7 @@ func TestValidateConstraint(t *testing.T) {
"matchExpressions": [{
"key": "someKey",
"operator": "In",
"values": ["some value"]
"values": ["some-value"]
}]
}
},
Expand Down

0 comments on commit 2ecccf2

Please sign in to comment.