Skip to content

Commit

Permalink
added newDefault function for TCPAddrConfig struct also added test fo…
Browse files Browse the repository at this point in the history
…r the same
  • Loading branch information
Sanket-0510 committed Mar 6, 2024
1 parent ae3bd69 commit 093a842
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 0 deletions.
5 changes: 5 additions & 0 deletions config/confignet/confignet.go
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,11 @@ type TCPAddrConfig struct {
DialerConfig DialerConfig `mapstructure:"dialer"`
}

// set default values for the TCP Address configuration
func NewDefaultTCPAddrConfig() TCPAddrConfig {
return TCPAddrConfig{}
}

// Dial equivalent with net.Dialer's DialContext for this address.
func (na *TCPAddrConfig) Dial(ctx context.Context) (net.Conn, error) {
d := net.Dialer{Timeout: na.DialerConfig.Timeout}
Expand Down
6 changes: 6 additions & 0 deletions config/confignet/confignet_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,12 @@ func TestNewDefaultDialerConfig(t *testing.T) {
require.Equal(t, expectedDialerConfig, dialerConfig)
}

func TestNewDefaultTCPAddrConfig(t *testing.T) {
expectedTCPAddrConfig := TCPAddrConfig{}
tcpAddrconfig := NewDefaultTCPAddrConfig()
require.Equal(t, expectedTCPAddrConfig, tcpAddrconfig)
}

func TestNewDefaultAddrConfig(t *testing.T) {
expectedAddrConfig := AddrConfig{}
addrConfig := NewDefaultAddrConfig()
Expand Down

0 comments on commit 093a842

Please sign in to comment.