From 9cd81930e9480ba41a4645188b51dd6b927015da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maximilian=20Pa=C3=9F?=
 <22845248+mpass99@users.noreply.github.com>
Date: Fri, 10 Dec 2021 11:30:56 +0100
Subject: [PATCH] Add API Querier test

---
 internal/nomad/api_querier_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 internal/nomad/api_querier_test.go

diff --git a/internal/nomad/api_querier_test.go b/internal/nomad/api_querier_test.go
new file mode 100644
index 0000000..c8e5929
--- /dev/null
+++ b/internal/nomad/api_querier_test.go
@@ -0,0 +1,18 @@
+package nomad
+
+import (
+	"errors"
+	"fmt"
+	"github.com/gorilla/websocket"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestWebsocketErrorNeedsToBeUnwrapped(t *testing.T) {
+	rawError := &websocket.CloseError{Code: websocket.CloseNormalClosure}
+	err := fmt.Errorf("websocket closed before receiving exit code: %w", rawError)
+
+	assert.False(t, websocket.IsCloseError(err, websocket.CloseNormalClosure))
+	rootCause := errors.Unwrap(err)
+	assert.True(t, websocket.IsCloseError(rootCause, websocket.CloseNormalClosure))
+}