package assetcache

import "testing"

func TestDefaultBudget(t *testing.T) {
	const gib = int64(1) << 30
	cases := []struct {
		name  string
		total uint64
		want  int64
	}{
		{"20 GiB gives 2 GiB", uint64(20 * gib), 2 * gib},
		{"60 GiB gives 4 GiB", uint64(60 * gib), 4 * gib},
		{"40 GiB is exactly the cap", uint64(40 * gib), 4 * gib},
		{"zero gives zero", 0, 0},
	}
	for _, c := range cases {
		if got := DefaultBudget(c.total); got != c.want {
			t.Errorf("%s: DefaultBudget(%d) = %d, want %d", c.name, c.total, got, c.want)
		}
	}
}