Закинул общий репозиторий

internal/services/decorators/NetDial.go

@@ -0,0 +1,40 @@
+package decorators
+
+import (
+	"net"
+	"strconv"
+	"time"
+)
+
+type Dialer interface {
+	DialTimeout(network, address string, timeout time.Duration) (net.Conn, error)
+}
+
+type NetDialer struct {
+}
+
+func (d NetDialer) DialTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
+	return net.DialTimeout(network, address, timeout)
+}
+
+type DialerDecorator interface {
+	DialPort(network, addr string, p int) int
+}
+
+type NetDecorator struct {
+	Dialer Dialer
+}
+
+func (d NetDecorator) DialPort(network, addr string, p int) int {
+	timeout := time.Second
+	c, err := d.Dialer.DialTimeout(network, addr+":"+strconv.Itoa(p), timeout)
+	if err != nil {
+		return 0
+	} else {
+		err = c.Close()
+		if err != nil {
+			panic(err)
+		}
+		return p
+	}
+}

internal/services/decorators/NetDial_test.go

@@ -0,0 +1,103 @@
+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)
+	})
+}