Fix timeout tests on linux and windows

Author: iksaif

statsd/uds_test.go

@@ -203,8 +203,13 @@ func TestUDSStreamPartialWrite(t *testing.T) {
 
 	// Force a connection
 	w.ensureConnection()
+	conn, err := listener.Accept()
+	defer conn.Close()
+
 	// Set a very low buffer size to force a partial write, but still enough to write the header
-	w.conn.(*net.UnixConn).SetWriteBuffer(8)
+	w.conn.(*net.UnixConn).SetWriteBuffer(1)
+	// On linux we need to force a timeout this way
+	w.connectTimeout = -1 * time.Millisecond
 
 	msg := []byte("some data")
 	n, err := w.Write(msg)

statsd/uds_windows.go

@@ -10,6 +10,6 @@ import (
 
 // newUDSWriter is disabled on Windows, SOCK_DGRAM  are still unavailable but
 // SOCK_STREAM should work once implemented in the agent (https://devblogs.microsoft.com/commandline/af_unix-comes-to-windows/)
-func newUDSWriter(_ string, _ time.Duration, _ string) (Transport, error) {
+func newUDSWriter(_ string, _ time.Duration, _ time.Duration, _ string) (Transport, error) {
 	return nil, fmt.Errorf("Unix socket is not available on Windows")
 }