Elgato_dark/fleet
1
0
Fork 0
mirror of https://github.com/fleetdm/fleet synced 2026-05-23 08:58:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 111

Add test for fleetctl query --hosts (#20276)

Browse source
This commit is contained in:
jacobshandling 2024-07-09 14:13:09 -07:00 committed by GitHub
parent 12c3d76b24
commit d54ac3a35b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cmd/fleetctl/query_test.go
View file

@ -49,7 +49,10 @@ func TestSavedLiveQuery(t *testing.T) {
}
ds.HostIDsByIdentifierFunc = func(ctx context.Context, filter fleet.TeamFilter, hostIdentifiers []string) ([]uint, error) {
return []uint{1234}, nil
if len(hostIdentifiers) == 1 && hostIdentifiers[0] == "1234" {
return []uint{1234}, nil
}
return nil, nil
}
ds.LabelIDsByNameFunc = func(ctx context.Context, labels []string) (map[string]uint, error) {
return nil, nil
@ -70,7 +73,11 @@ func TestSavedLiveQuery(t *testing.T) {
ds.NewDistributedQueryCampaignTargetFunc = func(ctx context.Context, target *fleet.DistributedQueryCampaignTarget) (*fleet.DistributedQueryCampaignTarget, error) {
return target, nil
}
noHostsTargeted := false
ds.HostIDsInTargetsFunc = func(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets) ([]uint, error) {
if noHostsTargeted {
return nil, nil
}
return []uint{1}, nil
}
ds.CountHostsInTargetsFunc = func(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets, now time.Time) (fleet.TargetMetrics, error) {
@ -176,6 +183,12 @@ func TestSavedLiveQuery(t *testing.T) {
)
case <-c: // All good
}
// Test targeting no hosts (e.g. host does exist)
noHostsTargeted = true
_, err = runAppNoChecks([]string{"query", "--hosts", "foobar", "--query-name", queryName})
assert.Error(t, err)
assert.Contains(t, err.Error(), "No hosts targeted")
}
func TestAdHocLiveQuery(t *testing.T) {