Add connection constructor (#26)

go-foreman · Sep 15, 2022 · 051e018 · 051e018
1 parent 134a7ed
commit 051e018
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/pubsub/transport/amqp/connection.go b/pubsub/transport/amqp/connection.go
@@ -74,6 +74,15 @@ type Connection struct {
 	chReconnectionDelay time.Duration
 }
 
+func NewReconnectConnection(logger log.Logger, underlyingConn UnderlyingConnection, chReconnectionDelay time.Duration) *Connection {
+
+	return &Connection{
+		logger:              logger,
+		underlyingConn:      underlyingConn,
+		chReconnectionDelay: chReconnectionDelay,
+	}
+}
+
 func (c *Connection) Close() error {
 	return c.underlyingConn.Close()
 }

diff --git a/pubsub/transport/amqp/connection_test.go b/pubsub/transport/amqp/connection_test.go
@@ -228,3 +228,15 @@ func TestConnection_Channel(t *testing.T) {
 	//	testLogger.AssertContainsSubstr(t, "channel recreation succeed")
 	//})
 }
+
+func TestReconnectConnection(t *testing.T) {
+	t.Run("basic constructor", func(t *testing.T) {
+		testLogger := log.NewNilLogger()
+		uc := &MockUnderlyingConnection{}
+		rc := NewReconnectConnection(testLogger, uc, time.Second)
+
+		assert.Equal(t, rc.underlyingConn, uc)
+		assert.Equal(t, rc.chReconnectionDelay, time.Second)
+		assert.NotEmpty(t, rc.logger)
+	})
+}