schedtxt_api/internal/handler/scheduled_message_test.go

package handler

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"git.kundeng.us/phoenix/textsender-models/tx0/message/scheduling"
	"github.com/google/uuid"
	"github.com/stretchr/testify/assert"

	"git.kundeng.us/phoenix/textsender-api/internal/handler/endpoint"
	"git.kundeng.us/phoenix/textsender-api/internal/store/mock"
)

type CreateScheduledMessageRequest struct {
	Scheduled time.Time `json:"scheduled"`
	Status    string    `json:"status"`
	UserId    uuid.UUID `json:"user_id"`
}

func TestCreateScheduledMessageWithMock(t *testing.T) {
	now := time.Now()
	mockStore := mock.NewMockScheduledMessageStore()
	apiApp, err := GetApp()
	assert.NoError(t, err, "Error getting app")
	handler := NewScheduledMessageHandler(apiApp, mockStore)

	testUserId := uuid.New()
	testBody := testCreateScheduledMessageRequest(testUserId, now)
	jsonValue, _ := json.Marshal(testBody)

	req, _ := http.NewRequest("POST", endpoint.ScheduleMessageEndpoint, strings.NewReader(string(jsonValue)))
	rr := httptest.NewRecorder()

	handler.AddScheduledMessage(rr, req)

	assert.Equal(t, http.StatusCreated, rr.Code)

	var response AddScheduledMessageResponse
	err = json.Unmarshal(rr.Body.Bytes(), &response)
	assert.NoError(t, err, "Error Creating message %v", err)

	assert.NotEmpty(t, response.Data, "No Message created")

	assert.NotNil(t, response.Data[0].Id, "Id should not be nil")
}

func TestGetScheduledMessageWithMock(t *testing.T) {
	now := time.Now()
	mockStore := mock.NewMockScheduledMessageStore()

	testUserId := uuid.New()
	testBody := testCreateScheduledMessageRequest(testUserId, now)

	ctx := t.Context()
	schMsg := scheduling.ScheduledMessage{}
	schMsg.Scheduled = testBody.Scheduled
	schMsg.Status = testBody.Status
	schMsg.UserId = testBody.UserId
	if err := mockStore.CreateScheduledMessage(ctx, &schMsg); err != nil {
		assert.NoError(t, err, "Error Creating message %v", err)
	}

	url := fmt.Sprintf("%s?id=%s&user_id=%s", endpoint.GetScheduledMessageEndpoint, schMsg.Id, schMsg.UserId)
	req, _ := http.NewRequest("GET", url, nil)
	rr := httptest.NewRecorder()

	apiApp, err := GetApp()
	assert.NoError(t, err, "Error getting app")
	handler := NewScheduledMessageHandler(apiApp, mockStore)
	handler.GetScheduledMessage(rr, req)

	assert.Equal(t, http.StatusOK, rr.Code)

	var response GetScheduledMessageResponse
	err = json.Unmarshal(rr.Body.Bytes(), &response)
	assert.NoError(t, err, "Error parsing response %v", err)
}

func TestFetchScheduledMessageWithMock(t *testing.T) {
	now := time.Now()
	mockStore := mock.NewMockScheduledMessageStore()
	contactStore := mock.NewMockContactStore()
	messageStore := mock.NewMockMessageStore()
	schMsgEventStore := mock.NewMockScheduledMessageEventStore()

	contactId := uuid.New()
	messageId := uuid.New()
	scheduledMessageId := uuid.New()
	testUserId := uuid.New()

	con := TestContact(contactId, testUserId)
	msg := TestMessage(messageId, testUserId)
	schMsg := TestScheduledMessage(scheduledMessageId, testUserId, now)
	event := TestScheduledMessageEvent(msg.Id, *con.Id, schMsg.Id)

	ctx := t.Context()

	if err := contactStore.CreateContact(ctx, &con); err != nil {
		assert.NoError(t, err, "Error creating contact: %v", err)
	} else if err = messageStore.CreateMessage(ctx, &msg); err != nil {
		assert.NoError(t, err, "Error creating message: %v", err)
	} else if err = mockStore.CreateScheduledMessage(ctx, &schMsg); err != nil {
		assert.NoError(t, err, "Error creating scheduled message: %v", err)
	} else if err = schMsgEventStore.CreateScheduledMessageEvent(ctx, &event); err != nil {
		assert.NoError(t, err, "Error creating scheduled message event: %v", err)
	} else if newStatus, err := mockStore.UpdateStatus(ctx, schMsg.Id, scheduling.Ready); err != nil {
		assert.NoError(t, err, "Error updating status: %v", err)
	} else {
		assert.Equal(t, *newStatus, scheduling.Ready)
	}

	req, _ := http.NewRequest("GET", endpoint.FetchNextScheduledMessageEndpoint, nil)
	rr := httptest.NewRecorder()

	apiApp, err := GetApp()
	assert.NoError(t, err, "Error getting app")
	handler := NewScheduledMessageHandler(apiApp, mockStore)
	handler.FetchNextMessage(rr, req)

	assert.Equal(t, http.StatusOK, rr.Code)

	var response FetchNextMessageResponse
	err = json.Unmarshal(rr.Body.Bytes(), &response)
	assert.NoError(t, err, "Error fetching scheduled message: %v", err)

	assert.NotEmpty(t, response.Data, "No data")

	fetchedSchMsg := response.Data[0]
	assert.Equal(t, scheduling.Processing, fetchedSchMsg.Status, "The statuses do not match")
}

func testCreateScheduledMessageRequest(userId uuid.UUID, now time.Time) CreateScheduledMessageRequest {
	scheduled := now.Add(5 * time.Minute)
	return CreateScheduledMessageRequest{Status: scheduling.Pending, UserId: userId, Scheduled: scheduled}
}