tsk-57: Change schema of scheduled_message_events (#65)
Closes #57 Reviewed-on: phoenix/textsender-api#65 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
This commit is contained in:
@@ -6,7 +6,7 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"git.kundeng.us/phoenix/textsender-models/tx0/message"
|
||||
"git.kundeng.us/phoenix/textsender-models/tx0/message/event"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
@@ -15,20 +15,20 @@ type MessageEventResponseKey struct {
|
||||
UserId uuid.UUID
|
||||
}
|
||||
type MockMessageEventResponseStore struct {
|
||||
MessageEventResponses map[uuid.UUID]*message.MessageEventResponse
|
||||
MessageEventResponsesByKey map[MessageEventResponseKey]*message.MessageEventResponse
|
||||
MessageEventResponses map[uuid.UUID]*event.MessageEventResponse
|
||||
MessageEventResponsesByKey map[MessageEventResponseKey]*event.MessageEventResponse
|
||||
mu sync.RWMutex
|
||||
Error error
|
||||
}
|
||||
|
||||
func NewMockMessageEventResponseStore() *MockMessageEventResponseStore {
|
||||
return &MockMessageEventResponseStore{
|
||||
MessageEventResponses: make(map[uuid.UUID]*message.MessageEventResponse),
|
||||
MessageEventResponsesByKey: make(map[MessageEventResponseKey]*message.MessageEventResponse),
|
||||
MessageEventResponses: make(map[uuid.UUID]*event.MessageEventResponse),
|
||||
MessageEventResponsesByKey: make(map[MessageEventResponseKey]*event.MessageEventResponse),
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockMessageEventResponseStore) Create(ctx context.Context, mer *message.MessageEventResponse) error {
|
||||
func (m *MockMessageEventResponseStore) Create(ctx context.Context, mer *event.MessageEventResponse) error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
@@ -62,7 +62,7 @@ func (m *MockMessageEventResponseStore) Create(ctx context.Context, mer *message
|
||||
}
|
||||
|
||||
// TODO: Add code to get mock MessageEventResponse
|
||||
func (m *MockMessageEventResponseStore) GetWithUserId(ctx context.Context, userId uuid.UUID) ([]*message.MessageEventResponse, error) {
|
||||
func (m *MockMessageEventResponseStore) GetWithUserId(ctx context.Context, userId uuid.UUID) ([]*event.MessageEventResponse, error) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
@@ -70,13 +70,12 @@ func (m *MockMessageEventResponseStore) GetWithUserId(ctx context.Context, userI
|
||||
return nil, m.Error
|
||||
}
|
||||
|
||||
var messages []*message.MessageEventResponse
|
||||
var messages []*event.MessageEventResponse
|
||||
for _, msg := range m.MessageEventResponses {
|
||||
if msg.UserId == userId {
|
||||
messages = append(messages, msg)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return messages, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user