tsk-26: Retrieve scheduled message event endpoint (#28)
Closes #26 Reviewed-on: phoenix/textsender-api#28 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
This commit is contained in:
@@ -3,9 +3,7 @@ package mock
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
|
||||
@@ -240,62 +238,3 @@ func (m *MockScheduledMessageStore) CreateScheduledMessage(ctx context.Context,
|
||||
m.ScheduledMessagesByKey[key] = schedMsg
|
||||
return nil
|
||||
}
|
||||
|
||||
type ScheduledMessageEventKey struct {
|
||||
RecipientId uuid.UUID
|
||||
MessageId uuid.UUID
|
||||
ScheduledMessageId uuid.UUID
|
||||
}
|
||||
|
||||
type MockScheduledMessageEventStore struct {
|
||||
ScheduledMessageEvents map[uuid.UUID]*scheduling.ScheduledMessageEvent
|
||||
ScheduledMessageEventsByKey map[ScheduledMessageEventKey]*scheduling.ScheduledMessageEvent
|
||||
mu sync.RWMutex
|
||||
Error error // Optional: simulate errors
|
||||
}
|
||||
|
||||
func NewMockScheduledMessageEventStore() *MockScheduledMessageEventStore {
|
||||
return &MockScheduledMessageEventStore{
|
||||
ScheduledMessageEvents: make(map[uuid.UUID]*scheduling.ScheduledMessageEvent),
|
||||
ScheduledMessageEventsByKey: make(map[ScheduledMessageEventKey]*scheduling.ScheduledMessageEvent),
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockScheduledMessageEventStore) CreateScheduledMessageEvent(ctx context.Context, event *scheduling.ScheduledMessageEvent) error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
if m.Error != nil {
|
||||
return m.Error
|
||||
}
|
||||
|
||||
if event.Id == uuid.Nil {
|
||||
event.Id = uuid.New()
|
||||
}
|
||||
|
||||
key := ScheduledMessageEventKey{RecipientId: event.RecipientId, MessageId: event.MessageId, ScheduledMessageId: event.ScheduledMessageId}
|
||||
|
||||
if _, exists := m.ScheduledMessageEventsByKey[key]; exists {
|
||||
return fmt.Errorf("Already exists")
|
||||
} else {
|
||||
event.Created = time.Now()
|
||||
m.ScheduledMessageEvents[event.Id] = event
|
||||
m.ScheduledMessageEventsByKey[key] = event
|
||||
return nil
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (m *MockScheduledMessageEventStore) Exists(ctx context.Context, event *scheduling.ScheduledMessageEvent) (bool, error) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
if m.Error != nil {
|
||||
return false, m.Error
|
||||
}
|
||||
|
||||
key := ScheduledMessageEventKey{RecipientId: event.RecipientId, MessageId: event.MessageId, ScheduledMessageId: event.ScheduledMessageId}
|
||||
_, exists := m.ScheduledMessageEventsByKey[key]
|
||||
|
||||
return exists, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user