From a0b7f4feebe977eb02e3bad60955ce8deb26d861 Mon Sep 17 00:00:00 2001
From: AhsanFarooqDev <ahsanfarooqdevtech@gmail.com>
Date: Mon, 24 Jun 2024 17:14:11 +0500
Subject: [PATCH] Refactor TestRefreshToken UT

---
 handlers/auth_test.go | 30 ++++++++++++++++++++----------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/handlers/auth_test.go b/handlers/auth_test.go
index 461adde96..f2e10ec9e 100644
--- a/handlers/auth_test.go
+++ b/handlers/auth_test.go
@@ -5,6 +5,9 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"github.com/google/uuid"
+	"github.com/lib/pq"
+	mocks "github.com/stakwork/sphinx-tribes/mocks"
 	"net/http"
 	"net/http/httptest"
 	"os"
@@ -16,7 +19,6 @@ import (
 	"github.com/stakwork/sphinx-tribes/auth"
 	"github.com/stakwork/sphinx-tribes/config"
 	"github.com/stakwork/sphinx-tribes/db"
-	mocks "github.com/stakwork/sphinx-tribes/mocks"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -203,22 +205,27 @@ func TestGetIsAdmin(t *testing.T) {
 }
 
 func TestRefreshToken(t *testing.T) {
-	mockDb := mocks.NewDatabase(t)
-	aHandler := NewAuthHandler(mockDb)
+	teardownSuite := SetupSuite(t)
+	defer teardownSuite(t)
+	aHandler := NewAuthHandler(db.TestDB)
 
 	t.Run("Should test that a user token can be refreshed", func(t *testing.T) {
 		mockToken := "mock_token"
-		mockUserPubkey := "mock_pubkey"
-		mockPerson := db.Person{
-			ID:          1,
-			OwnerPubKey: mockUserPubkey,
+		person := db.Person{
+			Uuid:         uuid.New().String(),
+			OwnerPubKey:  "your_pubkey",
+			OwnerAlias:   "your_owner",
+			UniqueName:   "your_user",
+			Description:  "your user",
+			Tags:         pq.StringArray{},
+			Extras:       db.PropertyMap{},
+			GithubIssues: db.PropertyMap{},
 		}
-		mockDb.On("GetLnUser", mockUserPubkey).Return(int64(1)).Once()
-		mockDb.On("GetPersonByPubkey", mockUserPubkey).Return(mockPerson).Once()
+		db.TestDB.CreateOrEditPerson(person)
 
 		// Mock JWT decoding
 		mockClaims := jwt.MapClaims{
-			"pubkey": mockUserPubkey,
+			"pubkey": person.OwnerPubKey,
 		}
 		mockDecodeJwt := func(token string) (jwt.MapClaims, error) {
 			return mockClaims, nil
@@ -239,6 +246,8 @@ func TestRefreshToken(t *testing.T) {
 		}
 		req.Header.Set("x-jwt", mockToken)
 
+		fetchedPerson := db.TestDB.GetPersonByUuid(person.Uuid)
+		person.ID = fetchedPerson.ID
 		// Serve request
 		rr := httptest.NewRecorder()
 		handler := http.HandlerFunc(aHandler.RefreshToken)
@@ -253,5 +262,6 @@ func TestRefreshToken(t *testing.T) {
 		}
 		assert.Equal(t, true, responseData["status"])
 		assert.Equal(t, mockEncodedToken, responseData["jwt"])
+		assert.EqualValues(t, person, fetchedPerson)
 	})
 }