51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
package controllers
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
actionsv1alpha1 "github.com/actions-runner-controller/actions-runner-controller/api/v1alpha1"
|
|
"github.com/google/go-cmp/cmp"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestGetValidCacheEntries(t *testing.T) {
|
|
now := time.Now()
|
|
|
|
hra := &actionsv1alpha1.HorizontalRunnerAutoscaler{
|
|
Status: actionsv1alpha1.HorizontalRunnerAutoscalerStatus{
|
|
CacheEntries: []actionsv1alpha1.CacheEntry{
|
|
{
|
|
Key: "foo",
|
|
Value: 1,
|
|
ExpirationTime: metav1.Time{Time: now.Add(-time.Second)},
|
|
},
|
|
{
|
|
Key: "foo",
|
|
Value: 2,
|
|
ExpirationTime: metav1.Time{Time: now},
|
|
},
|
|
{
|
|
Key: "foo",
|
|
Value: 3,
|
|
ExpirationTime: metav1.Time{Time: now.Add(time.Second)},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
revs := getValidCacheEntries(hra, now)
|
|
|
|
counts := map[string]int{}
|
|
|
|
for _, r := range revs {
|
|
counts[r.Key] += r.Value
|
|
}
|
|
|
|
want := map[string]int{"foo": 3}
|
|
|
|
if d := cmp.Diff(want, counts); d != "" {
|
|
t.Errorf("%s", d)
|
|
}
|
|
}
|