97 lines
3.8 KiB
Go
97 lines
3.8 KiB
Go
package v1
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"github.com/gochat/gochat/internal/service"
|
|
"github.com/gochat/gochat/internal/worker"
|
|
)
|
|
|
|
func TestBulkActionHandler_ConversationEnqueuesChatwootPayload(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open("file:bulk-action-handler?mode=memory&cache=private"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&model.BackgroundJob{}))
|
|
sqlDB, _ := db.DB()
|
|
t.Cleanup(func() { _ = sqlDB.Close() })
|
|
|
|
wp := worker.NewWorkerPool(db)
|
|
handler := NewBulkActionHandler(nil, nil).WithWorkerPool(wp)
|
|
router := gin.New()
|
|
router.POST("/api/v1/accounts/:account_id/bulk_actions", handler.Create)
|
|
|
|
body := []byte(`{"type":"Conversation","ids":[101,102],"fields":{"status":"resolved"},"labels":{"add":["vip"],"remove":["old"]}}`)
|
|
request := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/7/bulk_actions", bytes.NewReader(body))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
request.Header.Set("X-User-ID", "42")
|
|
response := httptest.NewRecorder()
|
|
router.ServeHTTP(response, request)
|
|
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
require.Empty(t, response.Body.String())
|
|
|
|
var job model.BackgroundJob
|
|
require.NoError(t, db.Where("job_type = ? AND queue = ?", service.TaskTypeConversationBulkAction, "medium").First(&job).Error)
|
|
require.Contains(t, string(job.Payload), `"account_id":7`)
|
|
require.Contains(t, string(job.Payload), `"user_id":42`)
|
|
require.Contains(t, string(job.Payload), `"ids":[101,102]`)
|
|
require.Contains(t, string(job.Payload), `"status":"resolved"`)
|
|
require.Contains(t, string(job.Payload), `"add":["vip"]`)
|
|
}
|
|
|
|
func TestBulkActionHandler_ContactEnqueuesChatwootPayload(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open("file:bulk-action-contact-handler?mode=memory&cache=private"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&model.BackgroundJob{}))
|
|
sqlDB, _ := db.DB()
|
|
t.Cleanup(func() { _ = sqlDB.Close() })
|
|
|
|
wp := worker.NewWorkerPool(db)
|
|
handler := NewBulkActionHandler(nil, nil).WithWorkerPool(wp)
|
|
router := gin.New()
|
|
router.POST("/api/v1/accounts/:account_id/bulk_actions", handler.Create)
|
|
|
|
body := []byte(`{"type":"contact","ids":[11,12],"labels":{"add":["vip"],"remove":["old"]}}`)
|
|
request := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/7/bulk_actions", bytes.NewReader(body))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
request.Header.Set("X-User-ID", "42")
|
|
response := httptest.NewRecorder()
|
|
router.ServeHTTP(response, request)
|
|
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
require.Empty(t, response.Body.String())
|
|
|
|
var job model.BackgroundJob
|
|
require.NoError(t, db.Where("job_type = ? AND queue = ?", service.TaskTypeContactBulkAction, "medium").First(&job).Error)
|
|
require.Contains(t, string(job.Payload), `"account_id":7`)
|
|
require.Contains(t, string(job.Payload), `"user_id":42`)
|
|
require.Contains(t, string(job.Payload), `"type":"Contact"`)
|
|
require.Contains(t, string(job.Payload), `"ids":[11,12]`)
|
|
require.Contains(t, string(job.Payload), `"add":["vip"]`)
|
|
}
|
|
|
|
func TestBulkActionHandler_InvalidTypeMatchesChatwootPayload(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
handler := NewBulkActionHandler(nil, nil)
|
|
router := gin.New()
|
|
router.POST("/api/v1/accounts/:account_id/bulk_actions", handler.Create)
|
|
|
|
request := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/7/bulk_actions", bytes.NewReader([]byte(`{"type":"Ticket"}`)))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
response := httptest.NewRecorder()
|
|
router.ServeHTTP(response, request)
|
|
|
|
require.Equal(t, http.StatusUnprocessableEntity, response.Code)
|
|
require.JSONEq(t, `{"success":false}`, response.Body.String())
|
|
}
|