Rogeeandrogee
2b182f9956
* H-300: wire Captain Skills into Web runtime * H-300: enforce effective model and conservative skill budget * H-300: fix CI gosec step * ci: extend golangci-lint timeout * fix lint findings across backend * fix(push): resolve delivery protocol blockers * test(repository): close SQLite test databases * test(repository): reuse SQLite schema per package * H-307: restore backend Go cache in CI * H-307: prefetch modules before cold lint * H-307: resolve govulncheck security gate * H-307: build lint with patched Go toolchain * H-307: clear remaining security scan findings --------- Co-authored-by: Rogee <rogee@ipao.vip>
114 lines
4.0 KiB
Go
114 lines
4.0 KiB
Go
package v1
|
|
|
|
import (
|
|
"bytes"
|
|
"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 CaptainScenarioHandlerTestSuite struct {
|
|
suite.Suite
|
|
db *gorm.DB
|
|
handler *CaptainScenarioHandler
|
|
router *gin.Engine
|
|
account *model.Account
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) SetupSuite() {
|
|
var err error
|
|
s.db, err = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
s.Require().NoError(err)
|
|
s.Require().NoError(s.db.AutoMigrate(&model.Account{}, &model.CaptainAssistant{}, &model.CaptainScenario{}))
|
|
|
|
repo := repository.NewCaptainScenarioRepo(s.db)
|
|
svc := service.NewCaptainScenarioService(repo)
|
|
s.handler = NewCaptainScenarioHandler(svc)
|
|
|
|
gin.SetMode(gin.TestMode)
|
|
r := gin.New()
|
|
// Handler Create/Update/Delete read :id (account) and :assistant_id
|
|
// Handler Get reads :id (scenario ID, same key as account group — Gin conflict)
|
|
// Handler List reads :assistant_id
|
|
ag := r.Group("/api/v1/accounts/:id")
|
|
assistants := ag.Group("/captain_assistants/:assistant_id")
|
|
assistants.POST("/scenarios", s.handler.Create)
|
|
assistants.GET("/scenarios", s.handler.List)
|
|
assistants.PUT("/scenarios/:id", s.handler.Update)
|
|
assistants.DELETE("/scenarios/:id", s.handler.Delete)
|
|
// Separate route for Get since it reads c.Param("id") as scenario ID
|
|
r.GET("/api/v1/captain/scenarios/:id", s.handler.Get)
|
|
s.router = r
|
|
|
|
s.account = &model.Account{Name: "TestAccount"}
|
|
s.db.Create(s.account)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) SetupTest() {
|
|
s.db.Exec("DELETE FROM captain_scenarios")
|
|
}
|
|
|
|
func TestCaptainScenarioHandlerTestSuite(t *testing.T) {
|
|
suite.Run(t, new(CaptainScenarioHandlerTestSuite))
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestCreate_InvalidAccountID() {
|
|
w := httptest.NewRecorder()
|
|
body := bytes.NewBufferString(`{"name":"scenario1","description":"desc"}`)
|
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/abc/captain_assistants/1/scenarios", body)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestCreate_InvalidAssistantID() {
|
|
w := httptest.NewRecorder()
|
|
body := bytes.NewBufferString(`{"name":"scenario1","description":"desc"}`)
|
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("/api/v1/accounts/%d/captain_assistants/abc/scenarios", s.account.ID), body)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestGet_InvalidID() {
|
|
// Get reads c.Param("id") as scenario ID from separate route
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/captain/scenarios/abc", nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestList_InvalidAssistantID() {
|
|
// List reads c.Param("assistant_id") — not account_id
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/captain_assistants/abc/scenarios", s.account.ID), nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestUpdate_InvalidAccountID() {
|
|
w := httptest.NewRecorder()
|
|
body := bytes.NewBufferString(`{"name":"updated"}`)
|
|
req := httptest.NewRequest(http.MethodPut, "/api/v1/accounts/abc/captain_assistants/1/scenarios/1", body)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func (s *CaptainScenarioHandlerTestSuite) TestDelete_InvalidAccountID() {
|
|
w := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodDelete, "/api/v1/accounts/abc/captain_assistants/1/scenarios/1", nil)
|
|
s.router.ServeHTTP(w, req)
|
|
s.Equal(http.StatusBadRequest, w.Code)
|
|
}
|