Usage: Round messages to 1K, files to 100MB (#20603)

utils/utils_test.go

@@ -247,3 +247,132 @@ func TestRoundOffToZeroes(t *testing.T) {
 		})
 	}
 }
+
+func TestRoundOffToZeroesResolution(t *testing.T) {
+	messageGranularity := 3
+	storageGranularity := 8
+	testCases := []struct {
+		desc          string
+		n             float64
+		minResolution int
+		expected      int64
+	}{
+		{
+			desc:     "returns 0 when n is 0",
+			n:        0,
+			expected: 0,
+		},
+		{
+			desc:          "resolution of 0 does not round",
+			n:             12345,
+			expected:      12345,
+			minResolution: 0,
+		},
+		{
+			desc:          "resolution of 1 truncates to 10s: 9 -> 0",
+			n:             9,
+			expected:      0,
+			minResolution: 1,
+		},
+		{
+			desc:          "resolution of 1 truncates to 10s: 10 -> 10",
+			n:             10,
+			expected:      10,
+			minResolution: 1,
+		},
+		{
+			desc:          "resolution of 1 truncates to 10s: 11 -> 10",
+			n:             11,
+			expected:      10,
+			minResolution: 1,
+		},
+		{
+			desc:          "resolution of 1 truncates to 10s: 19 -> 10",
+			n:             19,
+			expected:      10,
+			minResolution: 1,
+		},
+		{
+			desc:          "supports message usage granularity (1000s): 9 -> 0",
+			n:             9,
+			expected:      0,
+			minResolution: messageGranularity,
+		},
+		{
+			desc:          "supports message usage granularity (1000s): 123 -> 100",
+			n:             9,
+			expected:      0,
+			minResolution: messageGranularity,
+		},
+		{
+			desc:          "supports message usage granularity (1000s): 1234 -> 1000",
+			n:             1234,
+			expected:      1000,
+			minResolution: messageGranularity,
+		},
+		{
+			desc:          "supports message usage granularity (1000s): 1500 -> 1000",
+			n:             1500,
+			expected:      1000,
+			minResolution: messageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (~100s of MiB): 13GiB -> 12.94GiB",
+			n:             13 * 1024 * 1024 * 1024,
+			expected:      13_900_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (~100s of MiB): 10GiB -> 9.965GiB",
+			n:             10 * 1024 * 1024 * 1024,
+			expected:      10_700_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			// first number at which usage reports as in excess of 10GiB.
+			// Evaluates to 10299.5992MiB
+			// Should be close enough for notifying of being over limit.
+			desc:          "supports file storage usage granularity (~100s of MiB): 10.0583GiB -> 10.0583GiB",
+			n:             10.0583 * 1024 * 1024 * 1024,
+			expected:      10_800_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (~100s of MiB): 953.67MiB -> 953.67MiB",
+			n:             1_000_000_000,
+			expected:      1_000_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (~100s of MiB): 1GiB -> 953.67MiB",
+			n:             1 * 1024 * 1024 * 1024,
+			expected:      1_000_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (~100s of MiB): 1049.04MiB -> 1049.04MiB",
+			n:             1_100_000_000,
+			expected:      1_100_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (smaller amounts): 104.904MiB -> 104.904MiB",
+			n:             100_000_000,
+			expected:      100_000_000,
+			minResolution: storageGranularity,
+		},
+		{
+			desc:          "supports file storage usage granularity (smaller amounts): 10.4904MiB -> 10.4904MiB",
+			n:             10_000_000,
+			expected:      10_000_000,
+			minResolution: storageGranularity,
+		},
+	}
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.desc, func(t *testing.T) {
+			res := RoundOffToZeroesResolution(tc.n, tc.minResolution)
+			assert.Equal(t, tc.expected, res)
+		})
+	}
+}