Function TestGroupAIBudget

enterprise/coderd/aibridge_test.go:2611–2872

(t *testing.T)

}

func TestGroupAIBudget(t *testing.T) {
	t.Parallel()

	t.Run("Upsert", func(t *testing.T) {
		t.Parallel()

		adminClient, group := setupGroupAIBudgetTest(t)
		ctx := testutil.Context(t, testutil.WaitLong)

		// First upsert creates the budget.
		newBudget, err := adminClient.UpsertGroupAIBudget(ctx, group.ID, codersdk.UpsertGroupAIBudgetRequest{
			SpendLimitMicros: 500_000_000,
		})
		require.NoError(t, err)
		require.Equal(t, group.ID, newBudget.GroupID)
		require.EqualValues(t, 500_000_000, newBudget.SpendLimitMicros)

		// Second upsert updates the existing budget.
		updatedBudget, err := adminClient.UpsertGroupAIBudget(ctx, group.ID, codersdk.UpsertGroupAIBudgetRequest{
			SpendLimitMicros: 1_000_000_000,
		})
		require.NoError(t, err)
		require.EqualValues(t, 1_000_000_000, updatedBudget.SpendLimitMicros)

		// GET returns the latest value.
		currentBudget, err := adminClient.GroupAIBudget(ctx, group.ID)
		require.NoError(t, err)
		require.EqualValues(t, 1_000_000_000, currentBudget.SpendLimitMicros)
	})

	t.Run("GetWhenAbsent_404", func(t *testing.T) {
		t.Parallel()

		adminClient, group := setupGroupAIBudgetTest(t)
		ctx := testutil.Context(t, testutil.WaitLong)

		_, err := adminClient.GroupAIBudget(ctx, group.ID)
		var sdkErr *codersdk.Error
		require.ErrorAs(t, err, &sdkErr)
		require.Equal(t, http.StatusNotFound, sdkErr.StatusCode())
	})

	t.Run("DeleteWhenAbsent_404", func(t *testing.T) {
		t.Parallel()

		adminClient, group := setupGroupAIBudgetTest(t)
		ctx := testutil.Context(t, testutil.WaitLong)

		err := adminClient.DeleteGroupAIBudget(ctx, group.ID)
		var sdkErr *codersdk.Error
		require.ErrorAs(t, err, &sdkErr)
		require.Equal(t, http.StatusNotFound, sdkErr.StatusCode())
	})

	t.Run("DeleteWhenPresent", func(t *testing.T) {
		t.Parallel()

		adminClient, group := setupGroupAIBudgetTest(t)