test code 수정

backend/internal/handler/auth_handler_login_test.go

@@ -11,7 +11,6 @@ import (
 	"bytes"
 	"context"
 	"encoding/json"
-	"errors"
 	"io"
 	"net/http"
 	"net/http/httptest"
@@ -81,15 +80,36 @@ func (m *MockIdentityProvider) UpdateUserPassword(loginID, newPassword string, r
 }
 
 type MockKratosAdminService struct {
-	// Simple mock for FindIdentityIDByIdentifier
+	mock.Mock
 }
 
 func (m *MockKratosAdminService) FindIdentityIDByIdentifier(ctx context.Context, identifier string) (string, error) {
-	// Always return a static ID for simplicity in this test
-	if identifier == "fail" {
-		return "", errors.New("not found")
+	args := m.Called(ctx, identifier)
+	return args.String(0), args.Error(1)
+}
+
+func (m *MockKratosAdminService) GetIdentity(ctx context.Context, id string) (*service.KratosIdentity, error) {
+	args := m.Called(ctx, id)
+	if args.Get(0) == nil {
+		return nil, args.Error(1)
 	}
-	return "kratos-identity-id", nil
+	return args.Get(0).(*service.KratosIdentity), args.Error(1)
+}
+
+func (m *MockKratosAdminService) ListIdentities(ctx context.Context) ([]service.KratosIdentity, error) {
+	return nil, nil
+}
+
+func (m *MockKratosAdminService) UpdateIdentity(ctx context.Context, identityID string, traits map[string]interface{}, state string) (*service.KratosIdentity, error) {
+	return nil, nil
+}
+
+func (m *MockKratosAdminService) UpdateIdentityPassword(ctx context.Context, identityID, newPassword string) error {
+	return nil
+}
+
+func (m *MockKratosAdminService) DeleteIdentity(ctx context.Context, identityID string) error {
+	return nil
 }
 
 // --- Helper ---
@@ -142,30 +162,17 @@ func TestPasswordLogin_OIDC_Success(t *testing.T) {
 		}
 	})
 
+	mockKratos := new(MockKratosAdminService)
+	mockKratos.On("FindIdentityIDByIdentifier", mock.Anything, "user@example.com").Return("kratos-identity-id", nil)
+
 	h := &AuthHandler{
 		IdpProvider: mockIdp,
-		KratosAdmin: service.NewKratosAdminService(), // We need to mock this better if resolveKratosIdentityIDFromLoginID calls real API
+		KratosAdmin: mockKratos,
 		Hydra: &service.HydraAdminService{
 			AdminURL:   "http://hydra.test",
 			HTTPClient: &http.Client{Transport: mockHydraTransport(hydraHandler)},
 		},
 	}
-	// Inject Mock Kratos (Hack: overwrite the service field if it was an interface, but it's a struct pointer)
-	// AuthHandler uses *service.KratosAdminService struct pointer.
-	// KratosAdminService methods are real. We need to mock HTTP client inside KratosAdminService too.
-
-	kratosHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		// Mock FindIdentityIDByIdentifier response
-		if strings.Contains(r.URL.Path, "/identities") {
-			json.NewEncoder(w).Encode([]map[string]interface{}{
-				{"id": "kratos-identity-id"},
-			})
-			return
-		}
-		http.NotFound(w, r)
-	})
-	h.KratosAdmin.HTTPClient = &http.Client{Transport: mockHydraTransport(kratosHandler)}
-	h.KratosAdmin.AdminURL = "http://kratos.test"
 
 	app := newAuthLoginTestApp(h)
 
@@ -215,21 +222,18 @@ func TestPasswordLogin_OIDC_InactiveClient(t *testing.T) {
 		http.NotFound(w, r)
 	})
 
+	mockKratos := new(MockKratosAdminService)
+	mockKratos.On("FindIdentityIDByIdentifier", mock.Anything, "user@example.com").Return("kratos-identity-id", nil)
+
 	h := &AuthHandler{
 		IdpProvider: mockIdp,
-		KratosAdmin: service.NewKratosAdminService(),
+		KratosAdmin: mockKratos,
 		Hydra: &service.HydraAdminService{
 			AdminURL:   "http://hydra.test",
 			HTTPClient: &http.Client{Transport: mockHydraTransport(hydraHandler)},
 		},
 	}
 
-	kratosHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		json.NewEncoder(w).Encode([]map[string]interface{}{{"id": "kratos-identity-id"}})
-	})
-	h.KratosAdmin.HTTPClient = &http.Client{Transport: mockHydraTransport(kratosHandler)}
-	h.KratosAdmin.AdminURL = "http://kratos.test"
-
 	app := newAuthLoginTestApp(h)
 
 	body, _ := json.Marshal(map[string]string{
@@ -259,18 +263,15 @@ func TestPasswordLogin_NoOIDC_Success(t *testing.T) {
 		Subject:      "kratos-identity-id",
 	}, nil)
 
+	mockKratos := new(MockKratosAdminService)
+	mockKratos.On("FindIdentityIDByIdentifier", mock.Anything, "user@example.com").Return("kratos-identity-id", nil)
+
 	h := &AuthHandler{
 		IdpProvider: mockIdp,
-		KratosAdmin: service.NewKratosAdminService(),
+		KratosAdmin: mockKratos,
 		Hydra:       service.NewHydraAdminService(),
 	}
 
-	kratosHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		json.NewEncoder(w).Encode([]map[string]interface{}{{"id": "kratos-identity-id"}})
-	})
-	h.KratosAdmin.HTTPClient = &http.Client{Transport: mockHydraTransport(kratosHandler)}
-	h.KratosAdmin.AdminURL = "http://kratos.test"
-
 	app := newAuthLoginTestApp(h)
 
 	body, _ := json.Marshal(map[string]string{