Function TestGetWorkspaceAgentsByInstanceID

coderd/database/querier_test.go:7793–7908

(t *testing.T)

}

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

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

		db, _ := dbtestutil.NewDB(t)
		resources := setupWorkspaceAgentQueryResources(t, db, 2)
		authInstanceID := fmt.Sprintf("instance-%s-%d", t.Name(), time.Now().UnixNano())
		olderCreatedAt := dbtime.Now().Add(-time.Hour)
		newerCreatedAt := dbtime.Now()

		olderAgent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{
			ResourceID: resources[0].ID,
			CreatedAt:  olderCreatedAt,
			AuthInstanceID: sql.NullString{
				String: authInstanceID,
				Valid:  true,
			},
		})
		newerAgent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{
			ResourceID: resources[1].ID,
			CreatedAt:  newerCreatedAt,
			AuthInstanceID: sql.NullString{
				String: authInstanceID,
				Valid:  true,
			},
		})

		ctx := testutil.Context(t, testutil.WaitShort)

		agents, err := db.GetWorkspaceAgentsByInstanceID(ctx, authInstanceID)
		require.NoError(t, err)
		require.Len(t, agents, 2)
		assert.Equal(t, []uuid.UUID{newerAgent.ID, olderAgent.ID}, []uuid.UUID{agents[0].ID, agents[1].ID})
	})

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

		db, _, sqlDB := dbtestutil.NewDBWithSQLDB(t)
		resources := setupWorkspaceAgentQueryResources(t, db, 2)
		authInstanceID := fmt.Sprintf("instance-%s-%d", t.Name(), time.Now().UnixNano())
		baseCreatedAt := dbtime.Now()

		rootAgent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{
			ResourceID: resources[0].ID,
			CreatedAt:  baseCreatedAt.Add(-time.Hour),
			AuthInstanceID: sql.NullString{
				String: authInstanceID,
				Valid:  true,
			},
		})
		_ = dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{
			ParentID:   uuid.NullUUID{UUID: rootAgent.ID, Valid: true},
			ResourceID: resources[0].ID,
			CreatedAt:  baseCreatedAt,
			AuthInstanceID: sql.NullString{