Fix existing TCP and UDP tests.

Author: sbruens

service/tcp_test.go

@@ -224,7 +224,7 @@ type probeTestMetrics struct {
 
 var _ TCPMetrics = (*probeTestMetrics)(nil)
 
-func (m *probeTestMetrics) AddClosedTCPConnection(clientInfo ipinfo.IPInfo, accessKey, status string, data metrics.ProxyMetrics, duration time.Duration) {
+func (m *probeTestMetrics) AddClosedTCPConnection(addr net.Addr, accessKey string, status string, data metrics.ProxyMetrics, duration time.Duration) {
 	m.mu.Lock()
 	m.closeStatus = append(m.closeStatus, status)
 	m.mu.Unlock()
@@ -233,8 +233,11 @@ func (m *probeTestMetrics) AddClosedTCPConnection(clientInfo ipinfo.IPInfo, acce
 func (m *probeTestMetrics) GetIPInfo(net.IP) (ipinfo.IPInfo, error) {
 	return ipinfo.IPInfo{}, nil
 }
-func (m *probeTestMetrics) AddOpenTCPConnection(clientInfo ipinfo.IPInfo) {
+func (m *probeTestMetrics) AddOpenTCPConnection(addr net.Addr) {
 }
+
+func (m *probeTestMetrics) AddAuthenticatedTCPConnection(ip net.Addr, accessKey string) {}
+
 func (m *probeTestMetrics) AddTCPProbe(status, drainResult string, port int, clientProxyBytes int64) {
 	m.mu.Lock()
 	m.probeData = append(m.probeData, clientProxyBytes)

service/udp_test.go

@@ -113,10 +113,10 @@ func (m *natTestMetrics) AddUDPPacketFromClient(clientInfo ipinfo.IPInfo, access
 }
 func (m *natTestMetrics) AddUDPPacketFromTarget(clientInfo ipinfo.IPInfo, accessKey, status string, targetProxyBytes, proxyClientBytes int) {
 }
-func (m *natTestMetrics) AddUDPNatEntry() {
+func (m *natTestMetrics) AddUDPNatEntry(ip net.Addr, accessKey string) {
 	m.natEntriesAdded++
 }
-func (m *natTestMetrics) RemoveUDPNatEntry()                                                 {}
+func (m *natTestMetrics) RemoveUDPNatEntry(ip net.Addr, accessKey string)                    {}
 func (m *natTestMetrics) AddUDPCipherSearch(accessKeyFound bool, timeToCipher time.Duration) {}
 
 // Takes a validation policy, and returns the metrics it