Envelope/Backend/Api/Messages/CreateMessage_test.go

package Messages_test

import (
	"bytes"
	"encoding/base64"
	"encoding/json"
	"net/http"
	"testing"
	"time"

	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Database"
	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Database/Seeder"
	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Models"
	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Tests"
	"github.com/gofrs/uuid"
)

// TODO: Write test for message expiry

func Test_CreateMessage(t *testing.T) {
	client, ts, err := Tests.InitTestEnv()
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	u, err := Database.GetUserByUsername("test")

	key, err := Seeder.GenerateAesKey()
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	dataCiphertext, err := key.AesEncrypt([]byte("Test message..."))
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	senderIDCiphertext, err := key.AesEncrypt([]byte(u.ID.String()))
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	id, err := uuid.NewV4()
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	id2, err := uuid.NewV4()
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	d := []struct {
		MessageData struct {
			ID           uuid.UUID `json:"id"`
			Data         string    `json:"data"`
			SenderID     string    `json:"sender_id"`
			SymmetricKey string    `json:"symmetric_key"`
		} `json:"message_data"`
		Messages []struct {
			ID             uuid.UUID `json:"id"`
			MessageDataID  uuid.UUID `json:"message_data_id"`
			SymmetricKey   string    `json:"symmetric_key"`
			AssociationKey string    `json:"association_key"`
			Expiry         time.Time `json:"expiry"`
		} `json:"message"`
	}{
		{
			MessageData: struct {
				ID           uuid.UUID `json:"id"`
				Data         string    `json:"data"`
				SenderID     string    `json:"sender_id"`
				SymmetricKey string    `json:"symmetric_key"`
			}{
				ID:           id,
				Data:         base64.StdEncoding.EncodeToString(dataCiphertext),
				SenderID:     base64.StdEncoding.EncodeToString(senderIDCiphertext),
				SymmetricKey: "",
			},
			Messages: []struct {
				ID             uuid.UUID `json:"id"`
				MessageDataID  uuid.UUID `json:"message_data_id"`
				SymmetricKey   string    `json:"symmetric_key"`
				AssociationKey string    `json:"association_key"`
				Expiry         time.Time `json:"expiry"`
			}{
				{
					ID:             id2,
					MessageDataID:  id,
					SymmetricKey:   "",
					AssociationKey: "",
					Expiry:         time.Now(),
				},
			},
		},
	}

	jsonStr, _ := json.Marshal(d)
	req, _ := http.NewRequest("POST", ts.URL+"/api/v1/auth/message", bytes.NewBuffer(jsonStr))
	req.Header.Set("Content-Type", "application/json")

	resp, err := client.Do(req)
	if err != nil {
		t.Errorf("Expected nil, recieved %s", err.Error())
		return
	}

	if resp.StatusCode != http.StatusNoContent {
		t.Errorf("Expected %d, recieved %d", http.StatusNoContent, resp.StatusCode)
		return
	}

	var m Models.Message
	err = Database.DB.First(&m).Error
	if err != nil {
		t.Errorf("Expected conversation detail record, received %s", err.Error())
		return
	}

	var md Models.MessageData
	err = Database.DB.First(&md).Error
	if err != nil {
		t.Errorf("Expected conversation detail record, received %s", err.Error())
		return
	}

}