183 lines
6.3 KiB
Go
183 lines
6.3 KiB
Go
package v1
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/suite"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"github.com/gochat/gochat/internal/repository"
|
|
"github.com/gochat/gochat/internal/service"
|
|
)
|
|
|
|
type PortalMemberHandlerTestSuite struct {
|
|
suite.Suite
|
|
db *gorm.DB
|
|
handler *PortalMemberHandler
|
|
router *gin.Engine
|
|
account *model.Account
|
|
portal *model.Portal
|
|
user *model.User
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) SetupSuite() {
|
|
s.db, _ = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
s.db.AutoMigrate(&model.PortalMember{}, &model.Portal{}, &model.Account{}, &model.User{})
|
|
|
|
repo := repository.NewPortalMemberRepo(s.db)
|
|
svc := service.NewPortalMemberService(repo)
|
|
s.handler = NewPortalMemberHandler(svc)
|
|
|
|
gin.SetMode(gin.TestMode)
|
|
r := gin.New()
|
|
|
|
accountGroup := r.Group("/api/v1/accounts/:account_id")
|
|
portalGroup := accountGroup.Group("/portals/:portal_id/members")
|
|
portalGroup.POST("", s.handler.Create)
|
|
portalGroup.GET("/:member_id", s.handler.Get)
|
|
portalGroup.PUT("/:member_id", s.handler.Update)
|
|
portalGroup.DELETE("/:member_id", s.handler.Delete)
|
|
portalGroup.GET("", s.handler.List)
|
|
|
|
s.router = r
|
|
|
|
s.account = &model.Account{Name: "TestAccount"}
|
|
s.db.Create(s.account)
|
|
|
|
s.portal = &model.Portal{AccountID: s.account.ID, Name: "TestPortal", Slug: "test-portal"}
|
|
s.db.Create(s.portal)
|
|
|
|
s.user = &model.User{Name: "TestUser", Email: "test@test.com"}
|
|
s.db.Create(s.user)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TearDownSuite() {
|
|
s.db.Exec("DELETE FROM portal_members")
|
|
s.db.Exec("DELETE FROM portals")
|
|
s.db.Exec("DELETE FROM accounts")
|
|
s.db.Exec("DELETE FROM users")
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) SetupTest() {
|
|
s.db.Exec("DELETE FROM portal_members")
|
|
}
|
|
|
|
func TestPortalMemberHandlerTestSuite(t *testing.T) {
|
|
suite.Run(t, new(PortalMemberHandlerTestSuite))
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestCreate_Success() {
|
|
body := fmt.Sprintf(`{"user_id":%d,"role":"reader"}`, s.user.ID)
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), bytes.NewBufferString(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusCreated, w.Code)
|
|
|
|
var resp map[string]interface{}
|
|
s.NoError(json.Unmarshal(w.Body.Bytes(), &resp))
|
|
data := resp["data"].(map[string]interface{})
|
|
s.Equal("reader", data["role"])
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestCreate_InvalidPortalID() {
|
|
body := fmt.Sprintf(`{"user_id":%d,"role":"reader"}`, s.user.ID)
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/1/portals/abc/members", bytes.NewBufferString(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestCreate_InvalidJSON() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), bytes.NewBufferString(`{invalid`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestGet_Success() {
|
|
pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "editor"}
|
|
s.db.Create(pm)
|
|
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusOK, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestGet_InvalidMemberID() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/accounts/1/portals/1/members/abc", nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestGet_NotFound() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/99999", s.account.ID, s.portal.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusNotFound, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestUpdate_Success() {
|
|
pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"}
|
|
s.db.Create(pm)
|
|
|
|
body := `{"role":"editor"}`
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPut, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), bytes.NewBufferString(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusOK, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestUpdate_InvalidMemberID() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPut, "/api/v1/accounts/1/portals/1/members/abc", bytes.NewBufferString(`{"role":"editor"}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestDelete_Success() {
|
|
pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"}
|
|
s.db.Create(pm)
|
|
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusNoContent, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestDelete_InvalidMemberID() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodDelete, "/api/v1/accounts/1/portals/1/members/abc", nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestList_Success() {
|
|
s.db.Create(&model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"})
|
|
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusOK, w.Code)
|
|
}
|
|
|
|
func (s *PortalMemberHandlerTestSuite) TestList_Empty() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusOK, w.Code)
|
|
} |