smart-search-back/tests/auth_handler_test.go

package tests

import (
	"context"

	authpb "git.techease.ru/Smart-search/smart-search-back/pkg/pb/auth"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

func (s *IntegrationSuite) TestAuthHandler_LoginWithNonExistentUser() {
	req := &authpb.LoginRequest{
		Email:     "nonexistent@example.com",
		Password:  "password123",
		Ip:        "127.0.0.1",
		UserAgent: "test-agent",
	}

	resp, err := s.authClient.Login(context.Background(), req)

	s.Error(err)
	s.Nil(resp)

	st, ok := status.FromError(err)
	s.True(ok)
	s.Equal(codes.NotFound, st.Code())
}

func (s *IntegrationSuite) TestAuthHandler_ValidateWithInvalidToken() {
	req := &authpb.ValidateRequest{
		AccessToken: "invalid-token",
	}

	resp, err := s.authClient.Validate(context.Background(), req)

	s.NoError(err)
	s.NotNil(resp)
	s.False(resp.Valid)
	s.Equal(int64(0), resp.UserId)
}

func (s *IntegrationSuite) TestAuthHandler_ValidateWithEmptyToken() {
	req := &authpb.ValidateRequest{
		AccessToken: "",
	}

	resp, err := s.authClient.Validate(context.Background(), req)

	s.NoError(err)
	s.NotNil(resp)
	s.False(resp.Valid)
	s.Equal(int64(0), resp.UserId)
}

func (s *IntegrationSuite) TestAuthHandler_RefreshWithInvalidToken() {
	req := &authpb.RefreshRequest{
		RefreshToken: "invalid-refresh-token",
	}

	resp, err := s.authClient.Refresh(context.Background(), req)

	s.Error(err)
	s.Nil(resp)

	st, ok := status.FromError(err)
	s.True(ok)
	s.Equal(codes.Unauthenticated, st.Code())
}

func (s *IntegrationSuite) TestAuthHandler_LogoutWithInvalidToken() {
	req := &authpb.LogoutRequest{
		AccessToken: "invalid-token",
	}

	resp, err := s.authClient.Logout(context.Background(), req)

	s.NoError(err)
	s.NotNil(resp)
	s.True(resp.Success)
}

func (s *IntegrationSuite) TestAuthHandler_RefreshTokenFlow() {
	ctx := context.Background()

	loginReq := &authpb.LoginRequest{
		Email:     "test@example.com",
		Password:  "testpassword",
		Ip:        "127.0.0.1",
		UserAgent: "integration-test",
	}

	loginResp, err := s.authClient.Login(ctx, loginReq)
	s.NoError(err)
	s.NotNil(loginResp)
	s.NotEmpty(loginResp.AccessToken)
	s.NotEmpty(loginResp.RefreshToken)

	validateReq := &authpb.ValidateRequest{
		AccessToken: loginResp.AccessToken,
	}

	validateResp, err := s.authClient.Validate(ctx, validateReq)
	s.NoError(err)
	s.NotNil(validateResp)
	s.True(validateResp.Valid)

	refreshReq := &authpb.RefreshRequest{
		RefreshToken: loginResp.RefreshToken,
		Ip:           "127.0.0.1",
		UserAgent:    "integration-test",
	}

	refreshResp, err := s.authClient.Refresh(ctx, refreshReq)
	s.NoError(err)
	s.NotNil(refreshResp)
	s.NotEmpty(refreshResp.AccessToken)

	logoutReq := &authpb.LogoutRequest{
		AccessToken: refreshResp.AccessToken,
	}

	logoutResp, err := s.authClient.Logout(ctx, logoutReq)
	s.NoError(err)
	s.NotNil(logoutResp)
	s.True(logoutResp.Success)
}

func (s *IntegrationSuite) TestAuthHandler_LogoutInvalidatesSession() {
	ctx := context.Background()

	loginReq := &authpb.LoginRequest{
		Email:     "test@example.com",
		Password:  "testpassword",
		Ip:        "127.0.0.1",
		UserAgent: "integration-test",
	}

	loginResp, err := s.authClient.Login(ctx, loginReq)
	s.NoError(err)
	s.NotEmpty(loginResp.AccessToken)

	logoutReq := &authpb.LogoutRequest{
		AccessToken: loginResp.AccessToken,
	}

	logoutResp, err := s.authClient.Logout(ctx, logoutReq)
	s.NoError(err)
	s.True(logoutResp.Success)

	validateReq := &authpb.ValidateRequest{
		AccessToken: loginResp.AccessToken,
	}

	validateResp, err := s.authClient.Validate(ctx, validateReq)
	s.NoError(err)
	s.NotNil(validateResp)
	s.False(validateResp.Valid)
}