Function TestUpsertAISeats

coderd/database/querier_test.go:11432–11467

TestUpsertAISeats verifies 'UpsertAISeatState' only returns true when a new row is inserted.

(t *testing.T)

// TestUpsertAISeats verifies 'UpsertAISeatState' only returns true when a new
// row is inserted.
func TestUpsertAISeats(t *testing.T) {
	t.Parallel()

	sqlDB := testSQLDB(t)
	err := migrations.Up(sqlDB)
	require.NoError(t, err)
	db := database.New(sqlDB)
	ctx := testutil.Context(t, testutil.WaitShort)

	now := dbtime.Now()

	user := dbgen.User(t, db, database.User{})
	newRow, err := db.UpsertAISeatState(ctx, database.UpsertAISeatStateParams{
		UserID:        user.ID,
		FirstUsedAt:   now.Add(time.Hour * -24),
		LastEventType: database.AiSeatUsageReasonTask,
	})
	require.NoError(t, err)
	require.True(t, newRow)

	alreadyExists, err := db.UpsertAISeatState(ctx, database.UpsertAISeatStateParams{
		UserID:        user.ID,
		FirstUsedAt:   now.Add(time.Hour * -23),
		LastEventType: database.AiSeatUsageReasonTask,
	})
	require.NoError(t, err)
	require.False(t, alreadyExists)

	alreadyExists, err = db.UpsertAISeatState(ctx, database.UpsertAISeatStateParams{
		UserID:        user.ID,
		FirstUsedAt:   now,
		LastEventType: database.AiSeatUsageReasonTask,
	})
	require.NoError(t, err)
	require.False(t, alreadyExists)
}

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