ソースを参照

Merge pull request #311 from rod-hynes/master

Fixes for automated tests
Rod Hynes 9 年 前
コミット
26ed3b85eb
2 ファイル変更18 行追加3 行削除
  1. 2 1
      psiphon/remoteServerList_test.go
  2. 16 2
      psiphon/server/server_test.go

+ 2 - 1
psiphon/remoteServerList_test.go

@@ -146,7 +146,8 @@ func TestObfuscatedRemoteServerLists(t *testing.T) {
 			map[time.Time]string{
 				epoch: string(encodedServerEntry),
 			},
-		})
+		},
+		nil)
 	if err != nil {
 		t.Fatalf("error paving OSL files: %s", err)
 	}

+ 16 - 2
psiphon/server/server_test.go

@@ -580,12 +580,26 @@ func makeTunneledWebRequest(t *testing.T, localHTTPProxyPort int) error {
 
 func makeTunneledNTPRequest(t *testing.T, localSOCKSProxyPort int, udpgwServerAddress string) error {
 
-	testHostname := "pool.ntp.org"
 	timeout := 20 * time.Second
+	var err error
+
+	for _, testHostname := range []string{"time.google.com", "time.nist.gov", "pool.ntp.org"} {
+		err = makeTunneledNTPRequestAttempt(t, testHostname, timeout, localSOCKSProxyPort, udpgwServerAddress)
+		if err == nil {
+			break
+		}
+		t.Logf("makeTunneledNTPRequestAttempt failed: %s", err)
+	}
+
+	return err
+}
+
+func makeTunneledNTPRequestAttempt(
+	t *testing.T, testHostname string, timeout time.Duration, localSOCKSProxyPort int, udpgwServerAddress string) error {
 
 	localUDPProxyAddress, err := net.ResolveUDPAddr("udp", "127.0.0.1:7301")
 	if err != nil {
-		t.Fatalf("ResolveUDPAddr failed: %s", err)
+		return fmt.Errorf("ResolveUDPAddr failed: %s", err)
 	}
 
 	// Note: this proxy is intended for this test only -- it only accepts a single connection,