mirror of
https://github.com/hanzoai/base.git
synced 2026-10-21 03:41:50 +00:00
zeekayandHanzo Dev
566eba8994
NOTICE keeps the MIT copyright line and points at LICENSE.md. The realtime handshake is CONNECT on both sides, and the docs now say so. Co-authored-by: Hanzo Dev <[email protected]>
124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package core
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSendSystemAlert(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testDataDir, err := os.MkdirTemp("", "sendSystemAlert_data")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(testDataDir)
|
|
|
|
testApp := NewBaseApp(BaseAppConfig{
|
|
DataDir: testDataDir,
|
|
})
|
|
defer testApp.ResetBootstrapState()
|
|
|
|
if err := testApp.Bootstrap(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := createTestSuperusers(testApp, 3); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
superuser, err := testApp.FindAuthRecordByEmail(CollectionNameSuperusers, "[email protected]")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var sendCalls int
|
|
testApp.OnMailerSend().BindFunc(func(e *MailerEvent) error {
|
|
sendCalls++
|
|
|
|
if !strings.Contains(e.Message.Subject, "test_subject") {
|
|
t.Fatalf("Missing %q in Message.Subject:\n%s", "test_subject", e.Message.Subject)
|
|
}
|
|
|
|
if !strings.Contains(e.Message.HTML, "test_details") {
|
|
t.Fatalf("Missing %q in Message.HTML:\n%s", "test_details", e.Message.HTML)
|
|
}
|
|
|
|
if len(e.Message.To) != 1 || e.Message.To[0].Address != "[email protected]" {
|
|
t.Fatalf("Expected To address %q, got %v", "[email protected]", e.Message.To)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
sendSystemAlert(testApp, superuser, "test_subject", "test_details")
|
|
|
|
if sendCalls != 1 {
|
|
t.Fatalf("Expected 1 mail send call, got %d", sendCalls)
|
|
}
|
|
}
|
|
|
|
func TestSendSystemAlertToAllSuperusers(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testDataDir, err := os.MkdirTemp("", "sendSystemAlertToAllSuperusers_data")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(testDataDir)
|
|
|
|
testApp := NewBaseApp(BaseAppConfig{
|
|
DataDir: testDataDir,
|
|
})
|
|
defer testApp.ResetBootstrapState()
|
|
|
|
if err := testApp.Bootstrap(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := createTestSuperusers(testApp, 3); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var sendCalls int
|
|
testApp.OnMailerSend().BindFunc(func(e *MailerEvent) error {
|
|
sendCalls++
|
|
|
|
if !strings.Contains(e.Message.Subject, "test_subject") {
|
|
t.Fatalf("Missing %q in Message.Subject:\n%s", "test_subject", e.Message.Subject)
|
|
}
|
|
|
|
if !strings.Contains(e.Message.HTML, "test_details") {
|
|
t.Fatalf("Missing %q in Message.HTML:\n%s", "test_details", e.Message.HTML)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
sendSystemAlertToAllSuperusers(testApp, "test_subject", "test_details")
|
|
|
|
if sendCalls != 3 {
|
|
t.Fatalf("Expected 3 mail send calls, got %d", sendCalls)
|
|
}
|
|
}
|
|
|
|
func createTestSuperusers(app App, total int) error {
|
|
superusersCollection, err := app.FindCollectionByNameOrId(CollectionNameSuperusers)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := range total {
|
|
superuser := NewRecord(superusersCollection)
|
|
superuser.SetEmail("test" + strconv.Itoa(i+1) + "@example.com")
|
|
|
|
if err := app.Save(superuser); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|