79 lines
3.0 KiB
Go
79 lines
3.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"github.com/gochat/gochat/internal/repository"
|
|
)
|
|
|
|
func TestContactInboxServiceCreate_GeneratesWhatsAppSourceID(t *testing.T) {
|
|
db := contactInboxServiceTestDB(t)
|
|
svc := NewContactInboxService(repository.NewContactInboxRepo(db))
|
|
contact := &model.Contact{AccountID: 1, Name: "Jane", PhoneNumber: "+15550101010"}
|
|
require.NoError(t, db.Create(contact).Error)
|
|
inbox := &model.Inbox{AccountID: 1, Name: "WA", ChannelType: "Channel::Whatsapp", ChannelID: 1}
|
|
require.NoError(t, db.Create(inbox).Error)
|
|
|
|
ci, err := svc.Create(context.Background(), CreateContactInboxRequest{
|
|
ContactID: contact.ID,
|
|
InboxID: inbox.ID,
|
|
Contact: contact,
|
|
Inbox: inbox,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "15550101010", ci.SourceID)
|
|
require.Equal(t, inbox.ID, ci.Inbox.ID)
|
|
}
|
|
|
|
func TestContactInboxServiceCreate_TwilioWhatsAppSourceID(t *testing.T) {
|
|
db := contactInboxServiceTestDB(t)
|
|
svc := NewContactInboxService(repository.NewContactInboxRepo(db))
|
|
contact := &model.Contact{AccountID: 1, Name: "Jane", PhoneNumber: "+15550101010"}
|
|
require.NoError(t, db.Create(contact).Error)
|
|
inbox := &model.Inbox{AccountID: 1, Name: "Twilio WA", ChannelType: "twilio_sms", ChannelID: 1, ChannelConfig: `{"medium":"whatsapp"}`}
|
|
require.NoError(t, db.Create(inbox).Error)
|
|
|
|
ci, err := svc.Create(context.Background(), CreateContactInboxRequest{
|
|
ContactID: contact.ID,
|
|
InboxID: inbox.ID,
|
|
Contact: contact,
|
|
Inbox: inbox,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "whatsapp:+15550101010", ci.SourceID)
|
|
}
|
|
|
|
func TestContactInboxServiceCreate_ReturnsExistingByContactInboxSource(t *testing.T) {
|
|
db := contactInboxServiceTestDB(t)
|
|
svc := NewContactInboxService(repository.NewContactInboxRepo(db))
|
|
contact := &model.Contact{AccountID: 1, Name: "Jane"}
|
|
require.NoError(t, db.Create(contact).Error)
|
|
inbox := &model.Inbox{AccountID: 1, Name: "API", ChannelType: "api", ChannelID: 1}
|
|
require.NoError(t, db.Create(inbox).Error)
|
|
|
|
first, err := svc.Create(context.Background(), CreateContactInboxRequest{ContactID: contact.ID, InboxID: inbox.ID, SourceID: "browser-1", HMACVerified: true, Contact: contact, Inbox: inbox})
|
|
require.NoError(t, err)
|
|
second, err := svc.Create(context.Background(), CreateContactInboxRequest{ContactID: contact.ID, InboxID: inbox.ID, SourceID: "browser-1", Contact: contact, Inbox: inbox})
|
|
require.NoError(t, err)
|
|
require.Equal(t, first.ID, second.ID)
|
|
require.True(t, second.HMACVerified)
|
|
|
|
var count int64
|
|
require.NoError(t, db.Model(&model.ContactInbox{}).Where("contact_id = ? AND inbox_id = ? AND source_id = ?", contact.ID, inbox.ID, "browser-1").Count(&count).Error)
|
|
require.Equal(t, int64(1), count)
|
|
}
|
|
|
|
func contactInboxServiceTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&model.Contact{}, &model.Inbox{}, &model.ContactInbox{}))
|
|
return db
|
|
}
|