fix(handlers): restore bundle import test build #850

Merged
devops-engineer merged 3 commits from fix/main-red-sqlmock-import into main 2026-05-13 18:10:09 +00:00
2 changed files with 8 additions and 4 deletions

View File

@ -49,6 +49,10 @@ func (h *BundleHandler) Import(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid bundle"})
return
}
if b.Schema == "" || b.Name == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid bundle"})
return
}
ctx := c.Request.Context()
result := bundle.Import(ctx, &b, nil, h.broadcaster, h.provisioner, h.platformURL)

View File

@ -7,6 +7,7 @@ import (
"net/http/httptest"
"testing"
"github.com/DATA-DOG/go-sqlmock"
"github.com/gin-gonic/gin"
)
@ -52,6 +53,7 @@ func TestBundleImport_InvalidJSON(t *testing.T) {
func TestBundleImport_ValidJSON(t *testing.T) {
mock := setupTestDB(t)
_ = setupTestRedis(t)
broadcaster := newTestBroadcaster()
h := NewBundleHandler(broadcaster, nil, "http://localhost:8080", t.TempDir(), nil)
@ -59,12 +61,10 @@ func TestBundleImport_ValidJSON(t *testing.T) {
// bundle.Import recurses into SubWorkspaces (empty in this test bundle → no recursive INSERTs).
mock.ExpectExec("INSERT INTO workspaces").
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectExec("INSERT INTO structure_events").
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectExec("UPDATE workspaces SET runtime").
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectExec("INSERT INTO workspace_schedules").
WillReturnResult(sqlmock.NewResult(0, 1))
mock.ExpectExec("INSERT INTO workspace_secrets").
WillReturnResult(sqlmock.NewResult(0, 1))
body := `{"name": "test-workspace", "schema": "1.0", "tier": 3}`
w := httptest.NewRecorder()