mirror of
https://github.com/fleetdm/fleet
synced 2026-05-24 09:28:54 +00:00
33396a5d91
For #27927 Refactoring to speed up fleetctl tests, no functional changes. Mostly changing test files. fleetctl is no longer the long pole in CI, the long pole is mysql, followed by vuln. <img width="389" alt="image" src="https://github.com/user-attachments/assets/9ada64e2-b5e8-42e3-b120-4eb36183ae38" />
120 lines
2.7 KiB
Go
120 lines
2.7 KiB
Go
package fleetctl
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fleetdm/fleet/v4/cmd/fleetctl/fleetctl/testing_utils"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDeleteLabel(t *testing.T) {
|
|
_, ds := testing_utils.RunServerWithMockedDS(t)
|
|
|
|
var deletedLabel string
|
|
ds.DeleteLabelFunc = func(ctx context.Context, name string) error {
|
|
deletedLabel = name
|
|
return nil
|
|
}
|
|
|
|
name := writeTmpYml(t, `---
|
|
apiVersion: v1
|
|
kind: label
|
|
spec:
|
|
name: pending_updates
|
|
query: select 1;
|
|
platforms:
|
|
- darwin
|
|
`)
|
|
|
|
assert.Equal(t, "", RunAppForTest(t, []string{"delete", "-f", name}))
|
|
assert.True(t, ds.DeleteLabelFuncInvoked)
|
|
assert.Equal(t, "pending_updates", deletedLabel)
|
|
}
|
|
|
|
func TestDeletePack(t *testing.T) {
|
|
_, ds := testing_utils.RunServerWithMockedDS(t)
|
|
|
|
var deletedPack string
|
|
ds.DeletePackFunc = func(ctx context.Context, name string) error {
|
|
deletedPack = name
|
|
return nil
|
|
}
|
|
ds.PackByNameFunc = func(ctx context.Context, name string, opts ...fleet.OptionalArg) (*fleet.Pack, bool, error) {
|
|
if name != "pack1" {
|
|
return nil, false, nil
|
|
}
|
|
return &fleet.Pack{
|
|
ID: 7,
|
|
Name: "pack1",
|
|
Description: "some desc",
|
|
Platform: "darwin",
|
|
Disabled: false,
|
|
}, true, nil
|
|
}
|
|
ds.NewActivityFunc = func(
|
|
ctx context.Context, user *fleet.User, activity fleet.ActivityDetails, details []byte, createdAt time.Time,
|
|
) error {
|
|
return nil
|
|
}
|
|
|
|
name := writeTmpYml(t, `---
|
|
apiVersion: v1
|
|
kind: pack
|
|
spec:
|
|
description: some desc
|
|
disabled: false
|
|
id: 7
|
|
name: pack1
|
|
platform: darwin
|
|
targets:
|
|
labels: null
|
|
`)
|
|
|
|
assert.Equal(t, "", RunAppForTest(t, []string{"delete", "-f", name}))
|
|
assert.True(t, ds.DeletePackFuncInvoked)
|
|
assert.Equal(t, "pack1", deletedPack)
|
|
}
|
|
|
|
func TestDeleteQuery(t *testing.T) {
|
|
_, ds := testing_utils.RunServerWithMockedDS(t)
|
|
|
|
var deletedQuery string
|
|
ds.DeleteQueryFunc = func(ctx context.Context, teamID *uint, name string) error {
|
|
deletedQuery = name
|
|
return nil
|
|
}
|
|
ds.QueryByNameFunc = func(ctx context.Context, teamID *uint, name string) (*fleet.Query, error) {
|
|
if name != "query1" {
|
|
return nil, nil
|
|
}
|
|
return &fleet.Query{
|
|
ID: 33,
|
|
Name: "query1",
|
|
Description: "some desc",
|
|
Query: "select 1;",
|
|
Saved: false,
|
|
ObserverCanRun: false,
|
|
}, nil
|
|
}
|
|
ds.NewActivityFunc = func(
|
|
ctx context.Context, user *fleet.User, activity fleet.ActivityDetails, details []byte, createdAt time.Time,
|
|
) error {
|
|
return nil
|
|
}
|
|
|
|
name := writeTmpYml(t, `---
|
|
apiVersion: v1
|
|
kind: query
|
|
spec:
|
|
description: some desc
|
|
name: query1
|
|
query: select 1;
|
|
`)
|
|
|
|
assert.Equal(t, "", RunAppForTest(t, []string{"delete", "-f", name}))
|
|
assert.True(t, ds.DeleteQueryFuncInvoked)
|
|
assert.Equal(t, "query1", deletedQuery)
|
|
}
|