mirror of https://github.com/coder/coder.git
94 lines
1.5 KiB
Go
94 lines
1.5 KiB
Go
package metricscache
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestClosest(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testCases := []struct {
|
|
Name string
|
|
Keys []int
|
|
Input int
|
|
Expected int
|
|
NotFound bool
|
|
}{
|
|
{
|
|
Name: "Empty",
|
|
Input: 10,
|
|
NotFound: true,
|
|
},
|
|
{
|
|
Name: "Equal",
|
|
Keys: []int{1, 2, 3, 4, 5, 6, 10, 12, 15},
|
|
Input: 10,
|
|
Expected: 10,
|
|
},
|
|
{
|
|
Name: "ZeroOnly",
|
|
Keys: []int{0},
|
|
Input: 10,
|
|
Expected: 0,
|
|
},
|
|
{
|
|
Name: "NegativeOnly",
|
|
Keys: []int{-10, -5},
|
|
Input: 10,
|
|
Expected: -5,
|
|
},
|
|
{
|
|
Name: "CloseBothSides",
|
|
Keys: []int{-10, -5, 0, 5, 8, 12},
|
|
Input: 10,
|
|
Expected: 8,
|
|
},
|
|
{
|
|
Name: "CloseNoZero",
|
|
Keys: []int{-10, -5, 5, 8, 12},
|
|
Input: 0,
|
|
Expected: -5,
|
|
},
|
|
{
|
|
Name: "CloseLeft",
|
|
Keys: []int{-10, -5, 0, 5, 8, 12},
|
|
Input: 20,
|
|
Expected: 12,
|
|
},
|
|
{
|
|
Name: "CloseRight",
|
|
Keys: []int{-10, -5, 0, 5, 8, 12},
|
|
Input: -20,
|
|
Expected: -10,
|
|
},
|
|
{
|
|
Name: "ChooseZero",
|
|
Keys: []int{-10, -5, 0, 5, 8, 12},
|
|
Input: 2,
|
|
Expected: 0,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
tc := tc
|
|
t.Run(tc.Name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
m := make(map[int]int)
|
|
for _, k := range tc.Keys {
|
|
m[k] = k
|
|
}
|
|
|
|
found, _, ok := closest(m, tc.Input)
|
|
if tc.NotFound {
|
|
require.False(t, ok, "should not be found")
|
|
} else {
|
|
require.True(t, ok)
|
|
require.Equal(t, tc.Expected, found, "closest")
|
|
}
|
|
})
|
|
}
|
|
}
|