package cmd import ( "context" "github.com/edgelesssys/constellation/internal/atls" ) type stubRecoveryClient struct { conn bool connectErr error closeErr error pushStateDiskKeyErr error pushStateDiskKeyKey []byte } func (c *stubRecoveryClient) Connect(_ string, _ []atls.Validator) error { c.conn = true return c.connectErr } func (c *stubRecoveryClient) Close() error { c.conn = false return c.closeErr } func (c *stubRecoveryClient) PushStateDiskKey(_ context.Context, stateDiskKey []byte) error { c.pushStateDiskKeyKey = stateDiskKey return c.pushStateDiskKeyErr }