mirror of
https://github.com/kyverno/kyverno.git
synced 2025-03-07 00:17:13 +00:00
87fa77fbcc
* store policy names cache to reduce lookup time * add validate audit handler * fix #958, remove auto-gen annotation on Pod * formatting code * update processTime to readable format * #586, add back unit test * update logging info * remove unused interface * handle generate policy in a single thread in weboook * resolve pr comments
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package policystatus
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"testing"
|
|
"time"
|
|
|
|
v1 "github.com/nirmata/kyverno/pkg/api/kyverno/v1"
|
|
)
|
|
|
|
type dummyStore struct {
|
|
}
|
|
|
|
func (d dummyStore) Get(policyName string) (*v1.ClusterPolicy, error) {
|
|
return &v1.ClusterPolicy{}, nil
|
|
}
|
|
|
|
type dummyStatusUpdater struct {
|
|
}
|
|
|
|
func (d dummyStatusUpdater) UpdateStatus(status v1.PolicyStatus) v1.PolicyStatus {
|
|
status.RulesAppliedCount++
|
|
return status
|
|
}
|
|
|
|
func (d dummyStatusUpdater) PolicyName() string {
|
|
return "policy1"
|
|
}
|
|
|
|
type dummyLister struct {
|
|
}
|
|
|
|
func (dl dummyLister) List(selector labels.Selector) (ret []*v1.ClusterPolicy, err error) {
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
func (dl dummyLister) Get(name string) (*v1.ClusterPolicy, error) {
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
func (dl dummyLister) GetPolicyForPolicyViolation(pv *v1.ClusterPolicyViolation) ([]*v1.ClusterPolicy, error) {
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
func (dl dummyLister) GetPolicyForNamespacedPolicyViolation(pv *v1.PolicyViolation) ([]*v1.ClusterPolicy, error) {
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
func (dl dummyLister) ListResources(selector labels.Selector) (ret []*v1.ClusterPolicy, err error) {
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
func TestKeyToMutex(t *testing.T) {
|
|
expectedCache := `{"policy1":{"rulesAppliedCount":100}}`
|
|
|
|
stopCh := make(chan struct{})
|
|
s := NewSync(nil, dummyLister{})
|
|
for i := 0; i < 100; i++ {
|
|
go s.updateStatusCache(stopCh)
|
|
}
|
|
|
|
for i := 0; i < 100; i++ {
|
|
go s.Listener.Send(dummyStatusUpdater{})
|
|
}
|
|
|
|
<-time.After(time.Second * 3)
|
|
stopCh <- struct{}{}
|
|
|
|
cacheRaw, _ := json.Marshal(s.cache.data)
|
|
if string(cacheRaw) != expectedCache {
|
|
t.Errorf("\nTestcase Failed\nGot:\n%v\nExpected:\n%v\n", string(cacheRaw), expectedCache)
|
|
}
|
|
}
|