25 lines
649 B
Go
25 lines
649 B
Go
package scheduler_test
|
|
|
|
import (
|
|
"github.com/cirruslabs/orchard/internal/controller/scheduler"
|
|
v1 "github.com/cirruslabs/orchard/pkg/resource/v1"
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
)
|
|
|
|
func TestWorkerToResources(t *testing.T) {
|
|
workerToResources := make(scheduler.WorkerToResources)
|
|
require.Len(t, workerToResources, 0)
|
|
|
|
workerToResources.Add("worker-name", v1.Resources{
|
|
"tart-vms": 1,
|
|
})
|
|
require.Len(t, workerToResources, 1)
|
|
|
|
workerToResources.Add("worker-name", v1.Resources{
|
|
"tart-vms": 1,
|
|
})
|
|
require.Len(t, workerToResources, 1)
|
|
require.Equal(t, v1.Resources{"tart-vms": 2}, workerToResources.Get("worker-name"))
|
|
}
|