35 lines
645 B
Go
35 lines
645 B
Go
|
package netxmocks
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/tls"
|
||
|
|
"errors"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTLSConnConnectionState(t *testing.T) {
|
||
|
|
state := tls.ConnectionState{Version: tls.VersionTLS12}
|
||
|
|
c := &TLSConn{
|
||
|
|
MockConnectionState: func() tls.ConnectionState {
|
||
|
|
return state
|
||
|
|
},
|
||
|
|
}
|
||
|
|
out := c.ConnectionState()
|
||
|
|
if !reflect.DeepEqual(out, state) {
|
||
|
|
t.Fatal("not the result we expected")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTLSConnHandshake(t *testing.T) {
|
||
|
|
expected := errors.New("mocked error")
|
||
|
|
c := &TLSConn{
|
||
|
|
MockHandshake: func() error {
|
||
|
|
return expected
|
||
|
|
},
|
||
|
|
}
|
||
|
|
err := c.Handshake()
|
||
|
|
if !errors.Is(err, expected) {
|
||
|
|
t.Fatal("not the error we expected", err)
|
||
|
|
}
|
||
|
|
}
|