mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 01:15:22 +00:00
76f8d59290
- Use pointers rather than null package types. - Use new internal ptr package. - Improved handling of changing user teams/roles.
99 lines
3.0 KiB
Go
99 lines
3.0 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/WatchBeam/clock"
|
|
"github.com/fleetdm/fleet/server/config"
|
|
"github.com/fleetdm/fleet/server/kolide"
|
|
"github.com/fleetdm/fleet/server/ptr"
|
|
kitlog "github.com/go-kit/kit/log"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newTestService(ds kolide.Datastore, rs kolide.QueryResultStore, lq kolide.LiveQueryStore) (kolide.Service, error) {
|
|
mailer := &mockMailService{SendEmailFn: func(e kolide.Email) error { return nil }}
|
|
license := kolide.LicenseInfo{Tier: "core"}
|
|
return NewService(ds, rs, kitlog.NewNopLogger(), config.TestConfig(), mailer, clock.C, nil, lq, ds, license)
|
|
}
|
|
|
|
func newTestServiceWithClock(ds kolide.Datastore, rs kolide.QueryResultStore, lq kolide.LiveQueryStore, c clock.Clock) (kolide.Service, error) {
|
|
mailer := &mockMailService{SendEmailFn: func(e kolide.Email) error { return nil }}
|
|
license := kolide.LicenseInfo{Tier: "core"}
|
|
return NewService(ds, rs, kitlog.NewNopLogger(), config.TestConfig(), mailer, c, nil, lq, ds, license)
|
|
}
|
|
|
|
func createTestAppConfig(t *testing.T, ds kolide.Datastore) *kolide.AppConfig {
|
|
config := &kolide.AppConfig{
|
|
OrgName: "Tyrell Corp",
|
|
OrgLogoURL: "https://tyrell.com/image.png",
|
|
KolideServerURL: "https://kolide.tyrell.com",
|
|
SMTPConfigured: true,
|
|
SMTPSenderAddress: "kolide@tyrell.com",
|
|
SMTPServer: "smtp.tyrell.com",
|
|
SMTPPort: 587,
|
|
SMTPAuthenticationType: kolide.AuthTypeUserNamePassword,
|
|
SMTPUserName: "deckard",
|
|
SMTPPassword: "replicant",
|
|
SMTPVerifySSLCerts: true,
|
|
SMTPEnableTLS: true,
|
|
}
|
|
result, err := ds.NewAppConfig(config)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, result)
|
|
return result
|
|
}
|
|
|
|
func createTestUsers(t *testing.T, ds kolide.Datastore) map[string]kolide.User {
|
|
users := make(map[string]kolide.User)
|
|
for _, u := range testUsers {
|
|
user := &kolide.User{
|
|
Name: "Test Name " + u.Username,
|
|
Username: u.Username,
|
|
Email: u.Email,
|
|
}
|
|
err := user.SetPassword(u.PlaintextPassword, 10, 10)
|
|
require.Nil(t, err)
|
|
user, err = ds.NewUser(user)
|
|
require.Nil(t, err)
|
|
users[user.Username] = *user
|
|
}
|
|
return users
|
|
}
|
|
|
|
var testUsers = map[string]struct {
|
|
Username string
|
|
Email string
|
|
PlaintextPassword string
|
|
GlobalRole *string
|
|
}{
|
|
"admin1": {
|
|
Username: "admin1",
|
|
PlaintextPassword: "foobarbaz1234!",
|
|
Email: "admin1@example.com",
|
|
GlobalRole: ptr.String(kolide.RoleAdmin),
|
|
},
|
|
"user1": {
|
|
Username: "user1",
|
|
PlaintextPassword: "foobarbaz1234!",
|
|
Email: "user1@example.com",
|
|
GlobalRole: ptr.String(kolide.RoleMaintainer),
|
|
},
|
|
"user2": {
|
|
Username: "user2",
|
|
PlaintextPassword: "bazfoo1234!",
|
|
Email: "user2@example.com",
|
|
GlobalRole: ptr.String(kolide.RoleObserver),
|
|
},
|
|
}
|
|
|
|
type mockMailService struct {
|
|
SendEmailFn func(e kolide.Email) error
|
|
Invoked bool
|
|
}
|
|
|
|
func (svc *mockMailService) SendEmail(e kolide.Email) error {
|
|
svc.Invoked = true
|
|
return svc.SendEmailFn(e)
|
|
}
|