1
package mocks
2
 
3
import (
4
	"context"
5
	"crypto/tls"
6
 
7
	"go.miragespace.co/specter/spec/cipher"
8
 
9
	"github.com/stretchr/testify/mock"
10
)
11
 
12
type CertProvider struct {
13
	mock.Mock
14
}
15
 
16
func (c *CertProvider) GetCertificate(chi *tls.ClientHelloInfo) (*tls.Certificate, error) {
17
	args := c.Called(chi)
18
	r := args.Get(0)
19
	e := args.Error(1)
20
	if e != nil {
21
		return nil, e
22
	}
23
	return r.(*tls.Certificate), nil
24
}
25
 
26
func (c *CertProvider) GetCertificateWithContext(ctx context.Context, chi *tls.ClientHelloInfo) (*tls.Certificate, error) {
27
	args := c.Called(ctx, chi)
28
	r := args.Get(0)
29
	e := args.Error(1)
30
	if e != nil {
31
		return nil, e
32
	}
33
	return r.(*tls.Certificate), nil
34
}
35
 
36
func (c *CertProvider) Initialize(ctx context.Context) error {
37
	panic("unimplemented")
38
}
39
 
40
func (c *CertProvider) OnHandshake(cipher.OnHandshakeFunc) {
41
	panic("unimplemented")
42
}
43
 
44
var _ cipher.CertProvider = (*CertProvider)(nil)