mirror of
https://github.com/fleetdm/fleet
synced 2026-05-22 00:18:27 +00:00
25 lines
691 B
Go
25 lines
691 B
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func SetOrbitEnrollment(t *testing.T, h *fleet.Host, ds fleet.Datastore) string {
|
|
orbitKey := uuid.New().String()
|
|
_, err := ds.EnrollOrbit(context.Background(), false, fleet.OrbitHostInfo{
|
|
HardwareUUID: *h.OsqueryHostID,
|
|
HardwareSerial: h.HardwareSerial,
|
|
}, orbitKey, h.TeamID)
|
|
require.NoError(t, err)
|
|
err = ds.SetOrUpdateHostOrbitInfo(
|
|
context.Background(), h.ID, "1.22.0", sql.NullString{String: "42", Valid: true}, sql.NullBool{Bool: true, Valid: true},
|
|
)
|
|
require.NoError(t, err)
|
|
return orbitKey
|
|
}
|