github.com/Uptycs/basequery-go@v0.8.0/mock_manager.go (about)

     1  // Code generated by mockimpl. DO NOT EDIT.
     2  
     3  package osquery
     4  
     5  import "github.com/Uptycs/basequery-go/gen/osquery"
     6  
     7  var _ ExtensionManager = (*MockExtensionManager)(nil)
     8  
     9  type CloseFunc func()
    10  
    11  type PingFunc func() (*osquery.ExtensionStatus, error)
    12  
    13  type CallFunc func(registry string, item string, req osquery.ExtensionPluginRequest) (*osquery.ExtensionResponse, error)
    14  
    15  type ExtensionsFunc func() (osquery.InternalExtensionList, error)
    16  
    17  type RegisterExtensionFunc func(info *osquery.InternalExtensionInfo, registry osquery.ExtensionRegistry) (*osquery.ExtensionStatus, error)
    18  
    19  type OptionsFunc func() (osquery.InternalOptionList, error)
    20  
    21  type QueryFunc func(sql string) (*osquery.ExtensionResponse, error)
    22  
    23  type GetQueryColumnsFunc func(sql string) (*osquery.ExtensionResponse, error)
    24  
    25  type StreamEventsFunc func(name string, events osquery.ExtensionPluginResponse) (*osquery.ExtensionStatus, error)
    26  
    27  type GetNodeKeyFunc func() (string, error)
    28  
    29  type MockExtensionManager struct {
    30  	CloseFunc        CloseFunc
    31  	CloseFuncInvoked bool
    32  
    33  	PingFunc        PingFunc
    34  	PingFuncInvoked bool
    35  
    36  	CallFunc        CallFunc
    37  	CallFuncInvoked bool
    38  
    39  	ExtensionsFunc        ExtensionsFunc
    40  	ExtensionsFuncInvoked bool
    41  
    42  	RegisterExtensionFunc        RegisterExtensionFunc
    43  	RegisterExtensionFuncInvoked bool
    44  
    45  	OptionsFunc        OptionsFunc
    46  	OptionsFuncInvoked bool
    47  
    48  	QueryFunc        QueryFunc
    49  	QueryFuncInvoked bool
    50  
    51  	GetQueryColumnsFunc        GetQueryColumnsFunc
    52  	GetQueryColumnsFuncInvoked bool
    53  
    54  	StreamEventsFunc        StreamEventsFunc
    55  	StreamEventsFuncInvoked bool
    56  
    57  	GetNodeKeyFunc        GetNodeKeyFunc
    58  	GetNodeKeyFuncInvoked bool
    59  }
    60  
    61  func (m *MockExtensionManager) Close() {
    62  	m.CloseFuncInvoked = true
    63  	m.CloseFunc()
    64  }
    65  
    66  func (m *MockExtensionManager) Ping() (*osquery.ExtensionStatus, error) {
    67  	m.PingFuncInvoked = true
    68  	return m.PingFunc()
    69  }
    70  
    71  func (m *MockExtensionManager) Call(registry string, item string, req osquery.ExtensionPluginRequest) (*osquery.ExtensionResponse, error) {
    72  	m.CallFuncInvoked = true
    73  	return m.CallFunc(registry, item, req)
    74  }
    75  
    76  func (m *MockExtensionManager) Extensions() (osquery.InternalExtensionList, error) {
    77  	m.ExtensionsFuncInvoked = true
    78  	return m.ExtensionsFunc()
    79  }
    80  
    81  func (m *MockExtensionManager) RegisterExtension(info *osquery.InternalExtensionInfo, registry osquery.ExtensionRegistry) (*osquery.ExtensionStatus, error) {
    82  	m.RegisterExtensionFuncInvoked = true
    83  	return m.RegisterExtensionFunc(info, registry)
    84  }
    85  
    86  func (m *MockExtensionManager) Options() (osquery.InternalOptionList, error) {
    87  	m.OptionsFuncInvoked = true
    88  	return m.OptionsFunc()
    89  }
    90  
    91  func (m *MockExtensionManager) Query(sql string) (*osquery.ExtensionResponse, error) {
    92  	m.QueryFuncInvoked = true
    93  	return m.QueryFunc(sql)
    94  }
    95  
    96  func (m *MockExtensionManager) GetQueryColumns(sql string) (*osquery.ExtensionResponse, error) {
    97  	m.GetQueryColumnsFuncInvoked = true
    98  	return m.GetQueryColumnsFunc(sql)
    99  }
   100  
   101  func (m *MockExtensionManager) StreamEvents(name string, events osquery.ExtensionPluginResponse) (*osquery.ExtensionStatus, error) {
   102  	m.StreamEventsFuncInvoked = true
   103  	return m.StreamEventsFunc(name, events)
   104  }
   105  
   106  func (m *MockExtensionManager) GetNodeKey() (string, error) {
   107  	m.GetNodeKeyFuncInvoked = true
   108  	return m.GetNodeKey()
   109  }