From f875f7138978fea56c888e2ea9d88ea46c8fbc28 Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Mon, 26 Aug 2024 13:31:18 +0200
Subject: Exports: Add tests for cleanup and profile id

---
 exports/exports_wrapper.go | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/exports/exports_wrapper.go b/exports/exports_wrapper.go
index 1dd0675..0e25232 100644
--- a/exports/exports_wrapper.go
+++ b/exports/exports_wrapper.go
@@ -172,6 +172,10 @@ func testServer(t *testing.T) *test.Server {
     }
 }`,
 		},
+		{
+			Method: http.MethodPost,
+			Path:   "/test-api-endpoint/disconnect",
+		},
 		{
 			Path: "/test-api-endpoint/connect",
 			ResponseHandler: func(w http.ResponseWriter, r *http.Request) {
@@ -265,6 +269,30 @@ func testServerList(t *testing.T) {
 	}
 }
 
+func testSetProfileID(t *testing.T) {
+	prfS := C.CString("idontexist")
+	defer FreeString(prfS)
+	pErr := getError(t, SetProfileID(prfS))
+	if pErr == "" {
+		t.Fatal("got empty error for non-existent profile")
+	}
+	prfS2 := C.CString("employees")
+	defer FreeString(prfS2)
+	pErr = getError(t, SetProfileID(prfS2))
+	if pErr != "" {
+		t.Fatal("got error setting existent profile")
+	}
+}
+
+func testCleanup(t *testing.T) {
+	ck := CookieNew()
+	defer CookieDelete(ck)
+	cErr := getError(t, Cleanup(ck))
+	if cErr != "" {
+		t.Fatalf("failed cleaning up connection: %v", cErr)
+	}
+}
+
 func testGetConfig(t *testing.T) {
 	mustRegister(t)
 	defer Deregister()
@@ -318,4 +346,7 @@ func testGetConfig(t *testing.T) {
 	if !ok {
 		t.Fatalf("VPN config does not match regex: %v", cfgS)
 	}
+
+	testSetProfileID(t)
+	testCleanup(t)
 }
-- 
cgit v1.2.3