From d99fae647dc541c8c0db9bd696dfaaf5f760c206 Mon Sep 17 00:00:00 2001
From: Mayank Thakur <thakurmayank88@gmail.com>
Date: Mon, 6 Nov 2023 17:27:43 +0530
Subject: [PATCH] test: added more tests for network interceptor hook

---
 .../emailpassword/network_interceptor_test.go | 129 ++++++++++++++++++
 1 file changed, 129 insertions(+)

diff --git a/recipe/emailpassword/network_interceptor_test.go b/recipe/emailpassword/network_interceptor_test.go
index 09593217..fd719c04 100644
--- a/recipe/emailpassword/network_interceptor_test.go
+++ b/recipe/emailpassword/network_interceptor_test.go
@@ -1,11 +1,14 @@
 package emailpassword
 
 import (
+	"bytes"
 	"net/http"
 	"net/http/httptest"
+	"net/url"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/supertokens/supertokens-golang/recipe/userroles"
 	"github.com/supertokens/supertokens-golang/supertokens"
 	"github.com/supertokens/supertokens-golang/test/unittesting"
 )
@@ -96,3 +99,129 @@ func TestNetworkInterceptorNotSet(t *testing.T) {
 	assert.Equal(t, 200, res.StatusCode)
 	assert.Equal(t, false, isNetworkIntercepted)
 }
+
+func TestNetworkInterceptorIncorrectCoreURL(t *testing.T) {
+	isNetworkIntercepted = false
+	configValue := supertokens.TypeInput{
+		Supertokens: &supertokens.ConnectionInfo{
+			ConnectionURI: "http://localhost:8080",
+			NetworkInterceptor: func(request *http.Request, context supertokens.UserContext) *http.Request {
+				isNetworkIntercepted = true
+				newRequest := request
+				newRequest.URL.Path = "/public/recipe/incorrect/path"
+				return newRequest
+			},
+		},
+		AppInfo: supertokens.AppInfo{
+			AppName:       "SuperTokens",
+			APIDomain:     "api.supertokens.io",
+			WebsiteDomain: "supertokens.io",
+		},
+		RecipeList: []supertokens.Recipe{
+			Init(nil),
+		},
+	}
+	BeforeEach()
+
+	unittesting.StartUpST("localhost", "8080")
+
+	defer AfterEach()
+
+	err := supertokens.Init(configValue)
+
+	if err != nil {
+		t.Error(err.Error())
+	}
+
+	mux := http.NewServeMux()
+	testServer := httptest.NewServer(supertokens.Middleware(mux))
+	defer testServer.Close()
+
+	_, err = SignIn("public", "random@gmail.com", "validpass123")
+
+	assert.NotNil(t, err, "there should be an error")
+	assert.Contains(t, err.Error(), "status code: 404")
+	assert.Equal(t, true, isNetworkIntercepted)
+}
+
+func TestNetworkInterceptorIncorrectQueryParams(t *testing.T) {
+	isNetworkIntercepted = false
+	configValue := supertokens.TypeInput{
+		Supertokens: &supertokens.ConnectionInfo{
+			ConnectionURI: "http://localhost:8080",
+			NetworkInterceptor: func(r *http.Request, context supertokens.UserContext) *http.Request {
+				isNetworkIntercepted = true
+				newRequest := r
+				q := url.Values{}
+				newRequest.URL.RawQuery = q.Encode()
+				return newRequest
+			},
+		},
+		AppInfo: supertokens.AppInfo{
+			AppName:       "SuperTokens",
+			APIDomain:     "api.supertokens.io",
+			WebsiteDomain: "supertokens.io",
+		},
+		RecipeList: []supertokens.Recipe{
+			Init(nil),
+			userroles.Init(nil),
+		},
+	}
+	BeforeEach()
+
+	unittesting.StartUpST("localhost", "8080")
+
+	defer AfterEach()
+
+	supertokens.Init(configValue)
+
+	mux := http.NewServeMux()
+	testServer := httptest.NewServer(supertokens.Middleware(mux))
+	defer testServer.Close()
+
+	resp, _ := SignUp("public", "random@gmail.com", "validpass123")
+	_, err := userroles.GetRolesForUser("public", resp.OK.User.ID)
+	assert.NotNil(t, err, "should err, because userId is not passed")
+	assert.Contains(t, err.Error(), "status code: 400")
+	assert.Equal(t, true, isNetworkIntercepted)
+}
+
+func TestNetworkInterceptorRequestBody(t *testing.T) {
+	isNetworkIntercepted = false
+	configValue := supertokens.TypeInput{
+		Supertokens: &supertokens.ConnectionInfo{
+			ConnectionURI: "http://localhost:8080",
+			NetworkInterceptor: func(r *http.Request, context supertokens.UserContext) *http.Request {
+				isNetworkIntercepted = true
+				newBody := bytes.NewReader([]byte(`{"newKey": "newValue"}`))
+				req, _ := http.NewRequest(r.Method, r.URL.String(), newBody)
+				req.Header = r.Header
+				return req
+			},
+		},
+		AppInfo: supertokens.AppInfo{
+			AppName:       "SuperTokens",
+			APIDomain:     "api.supertokens.io",
+			WebsiteDomain: "supertokens.io",
+		},
+		RecipeList: []supertokens.Recipe{
+			Init(nil),
+		},
+	}
+	BeforeEach()
+
+	unittesting.StartUpST("localhost", "8080")
+
+	defer AfterEach()
+
+	supertokens.Init(configValue)
+
+	mux := http.NewServeMux()
+	testServer := httptest.NewServer(supertokens.Middleware(mux))
+	defer testServer.Close()
+
+	_, err := SignIn("public", "random@gmail.com", "validpass123")
+	assert.NotNil(t, err, "should err, because request body is incorrect")
+	assert.Contains(t, err.Error(), "status code: 400")
+	assert.Equal(t, true, isNetworkIntercepted)
+}