fleet/server/mock/datastore_hosts.go
John Murphy d5f9fcaeb2 Added FIM support (#1548)
This PR adds support for file integrity monitoring. This is done by providing a simplified API that can be used to PATCH/GET FIM configurations. There is also code to build the FIM configuration to send back to osquery. Each PATCH request, if successful, replaces Fleet's existing FIM configuration. For example:

curl -X "PATCH" "https://localhost:8080/api/v1/kolide/fim" \
     -H "Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzZXNzaW9uX2tleSI6IkVhaFhvZWswMGtWSEdaTTNCWndIMnhpYWxkNWZpcVFDR2hEcW1HK2UySmRNOGVFVE1DeTNTaUlFWmhZNUxhdW1ueFZDV2JiR1Bwdm5TKzdyK3NJUzNnPT0ifQ.SDCHAUA1vTuWGjXtcQds2GZLM27HAAiOUhR4WvgvTNY" \
     -H "Content-Type: application/json; charset=utf-8" \
     -d $'{
  "interval": 500,
  "file_paths": {
    "etc": [
      "/etc/%%"
    ],
    "users": [
      "/Users/%/Library/%%",
      "/Users/%/Documents/%%"
    ],
    "usr": [
      "/usr/bin/%%"
    ]
  }
}'
2017-08-18 10:37:33 -05:00

124 lines
3.4 KiB
Go

// Automatically generated by mockimpl. DO NOT EDIT!
package mock
import (
"time"
"github.com/kolide/fleet/server/kolide"
)
var _ kolide.HostStore = (*HostStore)(nil)
type NewHostFunc func(host *kolide.Host) (*kolide.Host, error)
type SaveHostFunc func(host *kolide.Host) error
type DeleteHostFunc func(hid uint) error
type HostFunc func(id uint) (*kolide.Host, error)
type ListHostsFunc func(opt kolide.ListOptions) ([]*kolide.Host, error)
type EnrollHostFunc func(osqueryHostId string, nodeKeySize int) (*kolide.Host, error)
type AuthenticateHostFunc func(nodeKey string) (*kolide.Host, error)
type MarkHostSeenFunc func(host *kolide.Host, t time.Time) error
type SearchHostsFunc func(query string, omit ...uint) ([]*kolide.Host, error)
type GenerateHostStatusStatisticsFunc func(now time.Time) (online uint, offline uint, mia uint, new uint, err error)
type DistributedQueriesForHostFunc func(host *kolide.Host) (map[uint]string, error)
type HostStore struct {
NewHostFunc NewHostFunc
NewHostFuncInvoked bool
SaveHostFunc SaveHostFunc
SaveHostFuncInvoked bool
DeleteHostFunc DeleteHostFunc
DeleteHostFuncInvoked bool
HostFunc HostFunc
HostFuncInvoked bool
ListHostsFunc ListHostsFunc
ListHostsFuncInvoked bool
EnrollHostFunc EnrollHostFunc
EnrollHostFuncInvoked bool
AuthenticateHostFunc AuthenticateHostFunc
AuthenticateHostFuncInvoked bool
MarkHostSeenFunc MarkHostSeenFunc
MarkHostSeenFuncInvoked bool
SearchHostsFunc SearchHostsFunc
SearchHostsFuncInvoked bool
GenerateHostStatusStatisticsFunc GenerateHostStatusStatisticsFunc
GenerateHostStatusStatisticsFuncInvoked bool
DistributedQueriesForHostFunc DistributedQueriesForHostFunc
DistributedQueriesForHostFuncInvoked bool
}
func (s *HostStore) NewHost(host *kolide.Host) (*kolide.Host, error) {
s.NewHostFuncInvoked = true
return s.NewHostFunc(host)
}
func (s *HostStore) SaveHost(host *kolide.Host) error {
s.SaveHostFuncInvoked = true
return s.SaveHostFunc(host)
}
func (s *HostStore) DeleteHost(hid uint) error {
s.DeleteHostFuncInvoked = true
return s.DeleteHostFunc(hid)
}
func (s *HostStore) Host(id uint) (*kolide.Host, error) {
s.HostFuncInvoked = true
return s.HostFunc(id)
}
func (s *HostStore) ListHosts(opt kolide.ListOptions) ([]*kolide.Host, error) {
s.ListHostsFuncInvoked = true
return s.ListHostsFunc(opt)
}
func (s *HostStore) EnrollHost(osqueryHostId string, nodeKeySize int) (*kolide.Host, error) {
s.EnrollHostFuncInvoked = true
return s.EnrollHostFunc(osqueryHostId, nodeKeySize)
}
func (s *HostStore) AuthenticateHost(nodeKey string) (*kolide.Host, error) {
s.AuthenticateHostFuncInvoked = true
return s.AuthenticateHostFunc(nodeKey)
}
func (s *HostStore) MarkHostSeen(host *kolide.Host, t time.Time) error {
s.MarkHostSeenFuncInvoked = true
return s.MarkHostSeenFunc(host, t)
}
func (s *HostStore) SearchHosts(query string, omit ...uint) ([]*kolide.Host, error) {
s.SearchHostsFuncInvoked = true
return s.SearchHostsFunc(query, omit...)
}
func (s *HostStore) GenerateHostStatusStatistics(now time.Time) (online uint, offline uint, mia uint, new uint, err error) {
s.GenerateHostStatusStatisticsFuncInvoked = true
return s.GenerateHostStatusStatisticsFunc(now)
}
func (s *HostStore) DistributedQueriesForHost(host *kolide.Host) (map[uint]string, error) {
s.DistributedQueriesForHostFuncInvoked = true
return s.DistributedQueriesForHostFunc(host)
}