mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 01:15:22 +00:00
9dba07395b
* check for errors when setting test logs on new services * fix tests that fail because a log file wasn't specified * remove unnecessary nil check
51 lines
1.9 KiB
Go
51 lines
1.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/contexts/viewer"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/fleetdm/fleet/v4/server/mock"
|
|
"github.com/fleetdm/fleet/v4/server/ptr"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestService_ListSoftware(t *testing.T) {
|
|
ds := new(mock.Store)
|
|
|
|
var calledWithTeamID *uint
|
|
var calledWithOpt fleet.SoftwareListOptions
|
|
ds.ListSoftwareFunc = func(ctx context.Context, opt fleet.SoftwareListOptions) ([]fleet.Software, error) {
|
|
calledWithTeamID = opt.TeamID
|
|
calledWithOpt = opt
|
|
return []fleet.Software{}, nil
|
|
}
|
|
|
|
user := &fleet.User{ID: 3, Email: "foo@bar.com", GlobalRole: ptr.String(fleet.RoleObserver)}
|
|
|
|
svc := newTestService(t, ds, nil, nil)
|
|
ctx := context.Background()
|
|
ctx = viewer.NewContext(ctx, viewer.Viewer{User: user})
|
|
|
|
_, err := svc.ListSoftware(ctx, fleet.SoftwareListOptions{TeamID: ptr.Uint(42), ListOptions: fleet.ListOptions{PerPage: 77, Page: 4}})
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ds.ListSoftwareFuncInvoked)
|
|
assert.Equal(t, ptr.Uint(42), calledWithTeamID)
|
|
// sort order defaults to hosts_count descending, automatically, if not explicitly provided
|
|
assert.Equal(t, fleet.ListOptions{PerPage: 77, Page: 4, OrderKey: "hosts_count", OrderDirection: fleet.OrderDescending}, calledWithOpt.ListOptions)
|
|
assert.True(t, calledWithOpt.WithHostCounts)
|
|
|
|
// call again, this time with an explicit sort
|
|
ds.ListSoftwareFuncInvoked = false
|
|
_, err = svc.ListSoftware(ctx, fleet.SoftwareListOptions{TeamID: nil, ListOptions: fleet.ListOptions{PerPage: 11, Page: 2, OrderKey: "id", OrderDirection: fleet.OrderAscending}})
|
|
require.NoError(t, err)
|
|
|
|
assert.True(t, ds.ListSoftwareFuncInvoked)
|
|
assert.Nil(t, calledWithTeamID)
|
|
assert.Equal(t, fleet.ListOptions{PerPage: 11, Page: 2, OrderKey: "id", OrderDirection: fleet.OrderAscending}, calledWithOpt.ListOptions)
|
|
assert.True(t, calledWithOpt.WithHostCounts)
|
|
}
|