second commit
This commit is contained in:
Rogee
@@ -0,0 +1,550 @@
|
||||
package v1
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"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"
|
||||
)
|
||||
|
||||
func uintPtr(v uint) *uint { return &v }
|
||||
|
||||
func setupNotificationDB(t *testing.T) *gorm.DB {
|
||||
t.Helper()
|
||||
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
||||
assert.NoError(t, err)
|
||||
err = db.AutoMigrate(&model.Notification{}, &model.User{}, &model.Account{}, &model.NotificationPreference{})
|
||||
assert.NoError(t, err)
|
||||
return db
|
||||
}
|
||||
|
||||
func setupNotificationHandler(t *testing.T, db *gorm.DB) *NotificationHandler {
|
||||
t.Helper()
|
||||
notifRepo := repository.NewNotificationRepo(db)
|
||||
prefRepo := repository.NewNotificationPreferenceRepo(db)
|
||||
ns := service.NewNotificationService(db, notifRepo, prefRepo)
|
||||
return NewNotificationHandler(ns)
|
||||
}
|
||||
|
||||
func setupNotificationRouter(handler *NotificationHandler) *gin.Engine {
|
||||
gin.SetMode(gin.TestMode)
|
||||
router := gin.New()
|
||||
// Middleware to set user_id in context — required by handlers that call getUserID(c)
|
||||
router.Use(func(c *gin.Context) {
|
||||
// Set a default user_id; individual tests can override via request header
|
||||
if uidStr := c.GetHeader("X-User-ID"); uidStr != "" {
|
||||
if uid, err := strconv.ParseUint(uidStr, 10, 64); err == nil {
|
||||
c.Set("user_id", uint(uid))
|
||||
}
|
||||
}
|
||||
c.Next()
|
||||
})
|
||||
router.GET("/api/v1/accounts/:account_id/notifications", handler.List)
|
||||
router.GET("/api/v1/accounts/:account_id/notifications/:notification_id", handler.Get)
|
||||
router.POST("/api/v1/accounts/:account_id/notifications/read_all", handler.MarkAllRead)
|
||||
router.PUT("/api/v1/accounts/:account_id/notifications/:notification_id", handler.Update)
|
||||
// G8 extension routes
|
||||
router.POST("/api/v1/accounts/:account_id/notifications/:notification_id/snooze", handler.Snooze)
|
||||
router.POST("/api/v1/accounts/:account_id/notifications/:notification_id/unread", handler.Unread)
|
||||
router.DELETE("/api/v1/accounts/:account_id/notifications/destroy_all", handler.DestroyAll)
|
||||
return router
|
||||
}
|
||||
|
||||
func TestNotificationList(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/1/notifications", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationListEmpty(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/1/notifications", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationGet(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/1/notifications/10", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusNotFound, w.Code) // notification doesn't exist
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationGetDifferentID(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/5/notifications/42", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusNotFound, w.Code) // notification doesn't exist
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationReadAll(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", "/api/v1/accounts/1/notifications/read_all", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationMarkRead(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("PATCH", "/api/v1/accounts/1/notifications/5/read", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusNotFound, w.Code) // notification doesn't exist
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationCRUDWithDB(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user := &model.User{Name: "Notif User", Email: "notif@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
// Create notification
|
||||
notif := &model.Notification{
|
||||
UserID: user.ID,
|
||||
AccountID: uintPtr(1),
|
||||
NotificationType: "conversation_assignment",
|
||||
PrimaryActorType: "conversation",
|
||||
PrimaryActorID: 5,
|
||||
SecondaryActorType: "user",
|
||||
SecondaryActorID: 3,
|
||||
}
|
||||
err := db.Create(notif).Error
|
||||
assert.NoError(t, err)
|
||||
assert.NotEqual(t, 0, notif.ID)
|
||||
|
||||
// Read notification
|
||||
var fetched model.Notification
|
||||
err = db.First(&fetched, notif.ID).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, "conversation_assignment", fetched.NotificationType)
|
||||
assert.Nil(t, fetched.ReadAt)
|
||||
|
||||
// Mark as read
|
||||
now := time.Now()
|
||||
err = db.Model(&fetched).Update("read_at", &now).Error
|
||||
assert.NoError(t, err)
|
||||
err = db.First(&fetched, notif.ID).Error
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, fetched.ReadAt)
|
||||
|
||||
// Delete notification
|
||||
err = db.Delete(&fetched).Error
|
||||
assert.NoError(t, err)
|
||||
err = db.First(&fetched, notif.ID).Error
|
||||
assert.Error(t, err)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationReadAllWithDB(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user := &model.User{Name: "Notif User2", Email: "notif2@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
// Create multiple unread notifications
|
||||
for i := 0; i < 5; i++ {
|
||||
notif := &model.Notification{
|
||||
UserID: user.ID,
|
||||
AccountID: uintPtr(1),
|
||||
NotificationType: "message_created",
|
||||
PrimaryActorType: "message",
|
||||
PrimaryActorID: uint(i + 1),
|
||||
}
|
||||
err := db.Create(notif).Error
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// Verify all are unread
|
||||
var unreadCount int64
|
||||
db.Model(&model.Notification{}).Where("user_id = ? AND read_at IS NULL", user.ID).Count(&unreadCount)
|
||||
assert.Equal(t, int64(5), unreadCount)
|
||||
|
||||
// Mark all as read
|
||||
now := time.Now()
|
||||
db.Model(&model.Notification{}).Where("user_id = ? AND read_at IS NULL", user.ID).Update("read_at", &now)
|
||||
|
||||
// Verify all are read
|
||||
db.Model(&model.Notification{}).Where("user_id = ? AND read_at IS NULL", user.ID).Count(&unreadCount)
|
||||
assert.Equal(t, int64(0), unreadCount)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationTypes(t *testing.T) {
|
||||
types := []string{
|
||||
"conversation_assignment",
|
||||
"conversation_created",
|
||||
"message_created",
|
||||
"assigned_conversation_new_message",
|
||||
"conversation_mention",
|
||||
}
|
||||
for _, nt := range types {
|
||||
notif := model.Notification{NotificationType: nt}
|
||||
assert.Equal(t, nt, notif.NotificationType)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNotificationModelFields(t *testing.T) {
|
||||
notif := model.Notification{
|
||||
UserID: 1,
|
||||
AccountID: uintPtr(2),
|
||||
NotificationType: "conversation_assignment",
|
||||
PrimaryActorType: "conversation",
|
||||
PrimaryActorID: 5,
|
||||
SecondaryActorType: "user",
|
||||
SecondaryActorID: 3,
|
||||
}
|
||||
assert.Equal(t, uint(1), notif.UserID)
|
||||
assert.Equal(t, uintPtr(2), notif.AccountID)
|
||||
assert.Equal(t, "conversation_assignment", notif.NotificationType)
|
||||
assert.Nil(t, notif.ReadAt)
|
||||
}
|
||||
|
||||
func TestNotificationPagination(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/1/notifications?page=1&per_page=20", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationListByAccount(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user1 := &model.User{Name: "User1", Email: "u1@example.com", Password: "pass", AccountID: 1}
|
||||
user2 := &model.User{Name: "User2", Email: "u2@example.com", Password: "pass", AccountID: 2}
|
||||
db.Create(user1)
|
||||
db.Create(user2)
|
||||
|
||||
for i := 0; i < 3; i++ {
|
||||
notif := &model.Notification{UserID: user1.ID, AccountID: uintPtr(1), NotificationType: "message_created", PrimaryActorType: "message", PrimaryActorID: uint(i + 1)}
|
||||
db.Create(notif)
|
||||
}
|
||||
for i := 0; i < 2; i++ {
|
||||
notif := &model.Notification{UserID: user2.ID, AccountID: uintPtr(2), NotificationType: "conversation_assignment", PrimaryActorType: "conversation", PrimaryActorID: uint(i + 1)}
|
||||
db.Create(notif)
|
||||
}
|
||||
|
||||
var account1Notifs []model.Notification
|
||||
err := db.Where("account_id = ?", 1).Find(&account1Notifs).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 3, len(account1Notifs))
|
||||
|
||||
var account2Notifs []model.Notification
|
||||
err = db.Where("account_id = ?", 2).Find(&account2Notifs).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 2, len(account2Notifs))
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationGetInvalidID(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("GET", "/api/v1/accounts/1/notifications/abc", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusBadRequest, w.Code) // invalid ID format
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationReadSpecificNotification(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user := &model.User{Name: "Specific User", Email: "specific@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: uintPtr(1), NotificationType: "message_created", PrimaryActorType: "message", PrimaryActorID: 1}
|
||||
db.Create(notif)
|
||||
|
||||
// Read specific notification
|
||||
err := db.Model(notif).Update("ReadAt", time.Now()).Error
|
||||
assert.NoError(t, err)
|
||||
|
||||
var fetched model.Notification
|
||||
db.First(&fetched, notif.ID)
|
||||
assert.NotNil(t, fetched.ReadAt)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationFilterByReadStatus(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user := &model.User{Name: "Filter User", Email: "filter@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
// Create mix of read and unread
|
||||
for i := 0; i < 5; i++ {
|
||||
var readAt *time.Time
|
||||
if i%2 == 0 {
|
||||
readAt = &time.Time{}
|
||||
}
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: uintPtr(1), NotificationType: "message_created", PrimaryActorType: "message", PrimaryActorID: uint(i + 1), ReadAt: readAt}
|
||||
db.Create(notif)
|
||||
}
|
||||
|
||||
var unread []model.Notification
|
||||
err := db.Where("user_id = ? AND read_at IS NULL", user.ID).Find(&unread).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 2, len(unread)) // indices 1,3 are unread
|
||||
|
||||
var read []model.Notification
|
||||
err = db.Where("user_id = ? AND read_at IS NOT NULL", user.ID).Find(&read).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 3, len(read)) // indices 0,2,4 are read
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationSoftDelete(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
|
||||
user := &model.User{Name: "SoftDel Notif User", Email: "softdel@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: uintPtr(1), NotificationType: "message_created", PrimaryActorType: "message", PrimaryActorID: 1}
|
||||
db.Create(notif)
|
||||
|
||||
err := db.Delete(notif).Error
|
||||
assert.NoError(t, err)
|
||||
|
||||
var found model.Notification
|
||||
err = db.First(&found, notif.ID).Error
|
||||
assert.Error(t, err)
|
||||
|
||||
err = db.Unscoped().First(&found, notif.ID).Error
|
||||
assert.NoError(t, err)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
// --- G8 extension handler tests: Snooze, Unread, DestroyAll ---
|
||||
// Reference: Chatwoot notifications_controller.rb#snooze, #unread, #destroy_all
|
||||
|
||||
func TestNotificationHandler_SnoozeWithDB(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
user := &model.User{Name: "Snooze Handler User", Email: "snooze-handler@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
accountID := uint(1)
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: &accountID, NotificationType: "conversation_assignment", PrimaryActorType: "conversation", PrimaryActorID: 1}
|
||||
db.Create(notif)
|
||||
|
||||
// Snooze the notification
|
||||
snoozeTime := time.Now().Add(2 * time.Hour).Format(time.RFC3339)
|
||||
body := fmt.Sprintf(`{"snoozed_until":"%s"}`, snoozeTime)
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", fmt.Sprintf("/api/v1/accounts/1/notifications/%d/snooze", notif.ID), strings.NewReader(body))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("X-User-ID", strconv.FormatUint(uint64(user.ID), 10))
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
|
||||
// Verify notification has snoozed_until set
|
||||
var updated model.Notification
|
||||
require.NoError(t, db.First(&updated, notif.ID).Error)
|
||||
assert.NotNil(t, updated.SnoozedUntil)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationHandler_Snooze_InvalidID(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
snoozeTime := time.Now().Add(2 * time.Hour).Format(time.RFC3339)
|
||||
body := fmt.Sprintf(`{"snoozed_until":"%s"}`, snoozeTime)
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", "/api/v1/accounts/1/notifications/abc/snooze", strings.NewReader(body))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationHandler_Snooze_MissingBody(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
user := &model.User{Name: "Snooze Missing User", Email: "snooze-missing@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
accountID := uint(1)
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: &accountID, NotificationType: "conversation_assignment", PrimaryActorType: "conversation", PrimaryActorID: 1}
|
||||
db.Create(notif)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", fmt.Sprintf("/api/v1/accounts/1/notifications/%d/snooze", notif.ID), nil)
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("X-User-ID", strconv.FormatUint(uint64(user.ID), 10))
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationHandler_UnreadWithDB(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
user := &model.User{Name: "Unread Handler User", Email: "unread-handler@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
accountID := uint(1)
|
||||
readTime := time.Now()
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: &accountID, NotificationType: "conversation_assignment", PrimaryActorType: "conversation", PrimaryActorID: 1, ReadAt: &readTime}
|
||||
db.Create(notif)
|
||||
|
||||
// Mark it unread
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", fmt.Sprintf("/api/v1/accounts/1/notifications/%d/unread", notif.ID), nil)
|
||||
req.Header.Set("X-User-ID", strconv.FormatUint(uint64(user.ID), 10))
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
|
||||
// Verify notification read_at is cleared
|
||||
var updated model.Notification
|
||||
require.NoError(t, db.First(&updated, notif.ID).Error)
|
||||
assert.Nil(t, updated.ReadAt)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationHandler_Unread_InvalidID(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", "/api/v1/accounts/1/notifications/abc/unread", nil)
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
|
||||
func TestNotificationHandler_DestroyAllWithDB(t *testing.T) {
|
||||
db := setupNotificationDB(t)
|
||||
handler := setupNotificationHandler(t, db)
|
||||
router := setupNotificationRouter(handler)
|
||||
|
||||
user := &model.User{Name: "DestroyAll Handler User", Email: "destroy-all-handler@example.com", Password: "pass", AccountID: 1}
|
||||
db.Create(user)
|
||||
|
||||
accountID := uint(1)
|
||||
// Create multiple notifications
|
||||
for i := 0; i < 5; i++ {
|
||||
notif := &model.Notification{UserID: user.ID, AccountID: &accountID, NotificationType: fmt.Sprintf("type_%d", i), PrimaryActorType: "conversation", PrimaryActorID: uint(i + 1)}
|
||||
db.Create(notif)
|
||||
}
|
||||
|
||||
// Verify they exist
|
||||
var count int64
|
||||
db.Model(&model.Notification{}).Where("user_id = ? AND account_id = ?", user.ID, accountID).Count(&count)
|
||||
assert.Equal(t, int64(5), count)
|
||||
|
||||
// Destroy all
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("DELETE", "/api/v1/accounts/1/notifications/destroy_all", nil)
|
||||
req.Header.Set("X-User-ID", strconv.FormatUint(uint64(user.ID), 10))
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
|
||||
// Verify all deleted
|
||||
db.Model(&model.Notification{}).Where("user_id = ? AND account_id = ?", user.ID, accountID).Count(&count)
|
||||
assert.Equal(t, int64(0), count)
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
sqlDB.Close()
|
||||
}
|
||||
Reference in New Issue
Block a user