mirror of
https://github.com/fleetdm/fleet
synced 2026-05-24 09:28:54 +00:00
0b7747bef0
Replaces (and appropriately refactors) a number of endpoints that were removed long ago when we decided to kill the UI with the fleetctl release. We turned out not to do this, and now need to restore these missing endpoints. This is not a straight up replacement of the existing code because of refactoring to the DB schemas that was also done in the migration. Most of the replaced code was removed in #1670 and #1686. Fixes #1811, fixes #1810
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/kolide/fleet/server/kolide"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func NewQuery(t *testing.T, ds kolide.Datastore, name, q string, authorID uint, saved bool) *kolide.Query {
|
|
query, err := ds.NewQuery(&kolide.Query{
|
|
Name: name,
|
|
Query: q,
|
|
AuthorID: authorID,
|
|
Saved: saved,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
// Loading gives us the timestamps
|
|
query, err = ds.Query(query.ID)
|
|
require.Nil(t, err)
|
|
|
|
return query
|
|
}
|
|
|
|
func NewPack(t *testing.T, ds kolide.Datastore, name string) *kolide.Pack {
|
|
err := ds.ApplyPackSpecs([]*kolide.PackSpec{&kolide.PackSpec{Name: name}})
|
|
require.Nil(t, err)
|
|
|
|
// Loading gives us the timestamps
|
|
pack, ok, err := ds.PackByName(name)
|
|
require.True(t, ok)
|
|
require.Nil(t, err)
|
|
|
|
return pack
|
|
}
|
|
|
|
func NewCampaign(t *testing.T, ds kolide.Datastore, queryID uint, status kolide.DistributedQueryStatus, now time.Time) *kolide.DistributedQueryCampaign {
|
|
campaign, err := ds.NewDistributedQueryCampaign(&kolide.DistributedQueryCampaign{
|
|
UpdateCreateTimestamps: kolide.UpdateCreateTimestamps{
|
|
CreateTimestamp: kolide.CreateTimestamp{
|
|
CreatedAt: now,
|
|
},
|
|
},
|
|
QueryID: queryID,
|
|
Status: status,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
// Loading gives us the timestamps
|
|
campaign, err = ds.DistributedQueryCampaign(campaign.ID)
|
|
require.Nil(t, err)
|
|
|
|
return campaign
|
|
}
|
|
|
|
func AddHostToCampaign(t *testing.T, ds kolide.Datastore, campaignID, hostID uint) {
|
|
_, err := ds.NewDistributedQueryCampaignTarget(
|
|
&kolide.DistributedQueryCampaignTarget{
|
|
Type: kolide.TargetHost,
|
|
TargetID: hostID,
|
|
DistributedQueryCampaignID: campaignID,
|
|
})
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func AddLabelToCampaign(t *testing.T, ds kolide.Datastore, campaignID, labelID uint) {
|
|
_, err := ds.NewDistributedQueryCampaignTarget(
|
|
&kolide.DistributedQueryCampaignTarget{
|
|
Type: kolide.TargetLabel,
|
|
TargetID: labelID,
|
|
DistributedQueryCampaignID: campaignID,
|
|
})
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func NewExecution(t *testing.T, ds kolide.Datastore, campaignID uint, hostID uint) *kolide.DistributedQueryExecution {
|
|
execution, err := ds.NewDistributedQueryExecution(&kolide.DistributedQueryExecution{
|
|
HostID: hostID,
|
|
DistributedQueryCampaignID: campaignID,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
return execution
|
|
}
|
|
|
|
func NewHost(t *testing.T, ds kolide.Datastore, name, ip, key, uuid string, now time.Time) *kolide.Host {
|
|
osqueryHostID, _ := kolide.RandomText(10)
|
|
h, err := ds.NewHost(&kolide.Host{
|
|
HostName: name,
|
|
NodeKey: key,
|
|
UUID: uuid,
|
|
DetailUpdateTime: now,
|
|
SeenTime: now,
|
|
OsqueryHostID: osqueryHostID,
|
|
})
|
|
|
|
require.Nil(t, err)
|
|
require.NotZero(t, h.ID)
|
|
require.Nil(t, ds.MarkHostSeen(h, now))
|
|
|
|
return h
|
|
}
|
|
|
|
func NewUser(t *testing.T, ds kolide.Datastore, name, username, email string, admin bool) *kolide.User {
|
|
u, err := ds.NewUser(&kolide.User{
|
|
Password: []byte("garbage"),
|
|
Salt: "garbage",
|
|
Name: name,
|
|
Username: username,
|
|
Email: email,
|
|
Admin: admin,
|
|
})
|
|
|
|
require.Nil(t, err)
|
|
require.NotZero(t, u.ID)
|
|
|
|
return u
|
|
}
|
|
|
|
func NewScheduledQuery(t *testing.T, ds kolide.Datastore, pid, qid, interval uint, snapshot, removed bool) *kolide.ScheduledQuery {
|
|
sq, err := ds.NewScheduledQuery(&kolide.ScheduledQuery{
|
|
PackID: pid,
|
|
QueryID: qid,
|
|
Interval: interval,
|
|
Snapshot: &snapshot,
|
|
Removed: &removed,
|
|
})
|
|
require.Nil(t, err)
|
|
require.NotZero(t, sq.ID)
|
|
|
|
return sq
|
|
}
|