github.com/anth0d/nomad@v0.0.0-20221214183521-ae3a0a2cad06/client/pluginmanager/testing.go (about)

     1  package pluginmanager
     2  
     3  import "context"
     4  
     5  type MockPluginManager struct {
     6  	RunF                      func()
     7  	ShutdownF                 func()
     8  	WaitForFirstFingerprintCh <-chan struct{}
     9  }
    10  
    11  func (m *MockPluginManager) Run()               { m.RunF() }
    12  func (m *MockPluginManager) Shutdown()          { m.ShutdownF() }
    13  func (m *MockPluginManager) PluginType() string { return "mock" }
    14  func (m *MockPluginManager) WaitForFirstFingerprint(ctx context.Context) <-chan struct{} {
    15  	if m.WaitForFirstFingerprintCh != nil {
    16  		return m.WaitForFirstFingerprintCh
    17  	}
    18  
    19  	ch := make(chan struct{})
    20  	close(ch)
    21  	return ch
    22  }