104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
|
package decorators
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ConnStub struct {
|
||
|
|
isClosed bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c ConnStub) Read(b []byte) (n int, err error) {
|
||
|
|
return 0, nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) Write(b []byte) (n int, err error) {
|
||
|
|
return 0, nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) Close() error {
|
||
|
|
if c.isClosed {
|
||
|
|
return errors.New("failed close connection")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) LocalAddr() net.Addr {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) RemoteAddr() net.Addr {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) SetDeadline(t time.Time) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) SetReadDeadline(t time.Time) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func (c ConnStub) SetWriteDeadline(t time.Time) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type TestDialer struct {
|
||
|
|
isOpen bool
|
||
|
|
conn ConnStub
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d TestDialer) DialTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
|
||
|
|
if !d.isOpen {
|
||
|
|
return d.conn, errors.New("failed connection")
|
||
|
|
}
|
||
|
|
return d.conn, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type dialAllDataProvider struct {
|
||
|
|
dialer TestDialer
|
||
|
|
network string
|
||
|
|
address string
|
||
|
|
port int
|
||
|
|
openPort int
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNetDial_DialPort(t *testing.T) {
|
||
|
|
provider := []dialAllDataProvider{
|
||
|
|
{
|
||
|
|
TestDialer{isOpen: true, conn: ConnStub{isClosed: false}},
|
||
|
|
"tcp",
|
||
|
|
"scanme.nmap.org",
|
||
|
|
80,
|
||
|
|
80,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
TestDialer{isOpen: false, conn: ConnStub{isClosed: false}},
|
||
|
|
"tcp",
|
||
|
|
"scanme.nmap.org",
|
||
|
|
443,
|
||
|
|
0,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, p := range provider {
|
||
|
|
ps := NetDecorator{p.dialer}.DialPort(p.network, p.address, p.port)
|
||
|
|
assert.Equal(t, p.openPort, ps)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type PanicDataProvider struct {
|
||
|
|
dialer TestDialer
|
||
|
|
network string
|
||
|
|
address string
|
||
|
|
port int
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNetDial_DialPortWhenPanic(t *testing.T) {
|
||
|
|
p := PanicDataProvider{
|
||
|
|
TestDialer{isOpen: true, conn: ConnStub{isClosed: true}},
|
||
|
|
"tcp",
|
||
|
|
"scanme.nmap.org",
|
||
|
|
664,
|
||
|
|
}
|
||
|
|
assert.Panics(t, func() {
|
||
|
|
NetDecorator{p.dialer}.DialPort(p.network, p.address, p.port)
|
||
|
|
})
|
||
|
|
}
|