You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
133 lines
3.3 KiB
133 lines
3.3 KiB
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
|
|
}
|
|
|
|
}
|