From 757ea5b63cacd1d1a08459658a6ae0aa127b7b42 Mon Sep 17 00:00:00 2001
From: 9seconds <nineseconds@yandex.ru>
Date: Thu, 11 Mar 2021 05:00:56 +0300
Subject: [PATCH] Add PatchHTTPClient method

This method is mostly required for testing
---
 cli_access.go             | 2 ++
 mtglib/network/init.go    | 1 +
 mtglib/network/network.go | 3 +++
 3 files changed, 6 insertions(+)

diff --git a/cli_access.go b/cli_access.go
index d78ffca..e0eea15 100644
--- a/cli_access.go
+++ b/cli_access.go
@@ -80,6 +80,8 @@ func (c *cliCommandAccess) getIP(protocol string) net.IP {
 		},
 	}
 
+	c.network.PatchHTTPClient(client)
+
 	resp, err := client.Get("https://ifconfig.co") // nolint: bodyclose, noctx
 	if err != nil {
 		return nil
diff --git a/mtglib/network/init.go b/mtglib/network/init.go
index d3aa8dc..d922ce1 100644
--- a/mtglib/network/init.go
+++ b/mtglib/network/init.go
@@ -39,4 +39,5 @@ type Network interface {
 	DNSResolve(network, hostname string) (ips []string, err error)
 	MakeHTTPClient(timeout time.Duration) *http.Client
 	IdleTimeout() time.Duration
+	PatchHTTPClient(*http.Client)
 }
diff --git a/mtglib/network/network.go b/mtglib/network/network.go
index 7d82878..d256dd0 100644
--- a/mtglib/network/network.go
+++ b/mtglib/network/network.go
@@ -120,6 +120,9 @@ func (n *network) MakeHTTPClient(timeout time.Duration) *http.Client {
 	}
 }
 
+func (n *network) PatchHTTPClient(_ *http.Client) {
+}
+
 func NewNetwork(dialer Dialer, dohHostname string, idleTimeout time.Duration) (Network, error) {
 	switch {
 	case idleTimeout < 0: