1 files changed, 46 insertions, 0 deletions
diff --git a/internal/wireguard/wireguard_test.go b/internal/wireguard/wireguard_test.go
new file mode 100644
index 0000000..026658e
--- /dev/null
+++ b/internal/wireguard/wireguard_test.go
@@ -0,0 +1,46 @@
+package wireguard
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test_ConfigAddKey(t *testing.T) {
+	config := `
+[Interface]
+
+[interface]
+
+[interface2]
+
+interface
+
+ [Interface]
+
+[Interface]test
+`
+	wgKey, wgKeyErr := GenerateKey()
+
+	if wgKeyErr != nil {
+		t.Fatalf("WireGuard config add key, generate key error: %v", wgKeyErr)
+	}
+	expectedConfig := fmt.Sprintf(`
+[Interface]
+PrivateKey = %s
+
+[interface]
+
+[interface2]
+
+interface
+
+ [Interface]
+
+[Interface]test
+`, wgKey.String())
+	gotConfig := ConfigAddKey(config, wgKey)
+
+	if gotConfig != expectedConfig {
+		t.Fatalf("Got: %s, Want: %s", gotConfig, expectedConfig)
+	}
+}