From 987876b1c6e7bee0f3b6996731e47188bcfe10a9 Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Mon, 12 Sep 2022 09:04:28 +0200
Subject: WireGuard: Add config replace test

---
 internal/wireguard/wireguard_test.go | 46 ++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 internal/wireguard/wireguard_test.go

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)
+	}
+}
-- 
cgit v1.2.3