Prechádzať zdrojové kódy

refactor(vpn): prevent sandbox violation when testing

Mustafa Arici 8 rokov pred
rodič
commit
174cd47329
2 zmenil súbory, kde vykonal 19 pridanie a 16 odobranie
  1. 15 15
      vpn.go
  2. 4 1
      vpn_test.go

+ 15 - 15
vpn.go

@@ -487,26 +487,26 @@ func emitCCD() error {
 	if err != nil {
 		return err
 	}
-
-	// Clean and then create and write rendered ccd data.
-	err = os.RemoveAll(_DefaultVPNCCDPath)
-	if err != nil {
-		if os.IsNotExist(err) {
-		} else {
-			return err
+	if !Testing {
+		// Clean and then create and write rendered ccd data.
+		err = os.RemoveAll(_DefaultVPNCCDPath)
+		if err != nil {
+			if os.IsNotExist(err) {
+			} else {
+				return err
+			}
 		}
-	}
 
-	if _, err := os.Stat(_DefaultVPNCCDPath); err != nil {
-	}
+		if _, err := os.Stat(_DefaultVPNCCDPath); err != nil {
+		}
 
-	err = os.Mkdir(_DefaultVPNCCDPath, 0755)
-	if err != nil {
-		if !os.IsExist(err) {
-			return err
+		err = os.Mkdir(_DefaultVPNCCDPath, 0755)
+		if err != nil {
+			if !os.IsExist(err) {
+				return err
+			}
 		}
 	}
-
 	for _, user := range users {
 		var result bytes.Buffer
 		params := struct {

+ 4 - 1
vpn_test.go

@@ -62,7 +62,10 @@ func TestVPNDeinit(t *testing.T) {
 	// Prepare:
 	// Initialize the server.
 	Init("localhost", "")
-	u, _ := CreateNewUser("user", "p", false)
+	u, err := CreateNewUser("user", "p", false)
+	if err != nil {
+		t.Fatal(err)
+	}
 	u.Delete()
 
 	// Test: