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.
205 lines
4.9 KiB
205 lines
4.9 KiB
package Messages_test
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"git.tovijaeschke.xyz/tovi/Envelope/Backend/Database"
|
|
"git.tovijaeschke.xyz/tovi/Envelope/Backend/Database/Seeder"
|
|
"git.tovijaeschke.xyz/tovi/Envelope/Backend/Models"
|
|
"git.tovijaeschke.xyz/tovi/Envelope/Backend/Tests"
|
|
)
|
|
|
|
func Test_Messages(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")
|
|
|
|
userKey, err := Seeder.GenerateAesKey()
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
keyCiphertext, err := userKey.AesEncrypt(
|
|
[]byte(base64.StdEncoding.EncodeToString(key.Key)),
|
|
)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
pubKey := Seeder.GetPubKey()
|
|
|
|
message := Models.Message{
|
|
MessageData: Models.MessageData{
|
|
Data: base64.StdEncoding.EncodeToString(dataCiphertext),
|
|
SenderID: base64.StdEncoding.EncodeToString(senderIDCiphertext),
|
|
SymmetricKey: base64.StdEncoding.EncodeToString(keyCiphertext),
|
|
},
|
|
SymmetricKey: base64.StdEncoding.EncodeToString(
|
|
Seeder.EncryptWithPublicKey(userKey.Key, pubKey),
|
|
),
|
|
AssociationKey: "AssociationKey",
|
|
}
|
|
|
|
err = Database.CreateMessage(&message)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
resp, err := client.Get(ts.URL + "/api/v1/auth/messages/AssociationKey")
|
|
if err != nil {
|
|
t.Errorf("Expected user record, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode)
|
|
return
|
|
}
|
|
|
|
requestBody, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
var m []Models.Message
|
|
err = json.Unmarshal(requestBody, &m)
|
|
|
|
if len(m) != 1 {
|
|
t.Errorf("Expected %d, recieved %d", 1, len(m))
|
|
}
|
|
|
|
msg := m[0]
|
|
|
|
decodedData, err := base64.StdEncoding.DecodeString(msg.MessageData.Data)
|
|
if err != nil {
|
|
t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode)
|
|
return
|
|
}
|
|
decrypedData, err := key.AesDecrypt(decodedData)
|
|
if err != nil {
|
|
t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode)
|
|
return
|
|
}
|
|
|
|
if string(decrypedData) != "Test message" {
|
|
t.Errorf("Expected %s, recieved %s", "Test converation", string(decrypedData))
|
|
}
|
|
}
|
|
|
|
func Test_MessagesPagination(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")
|
|
|
|
userKey, err := Seeder.GenerateAesKey()
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
keyCiphertext, err := userKey.AesEncrypt(
|
|
[]byte(base64.StdEncoding.EncodeToString(key.Key)),
|
|
)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
pubKey := Seeder.GetPubKey()
|
|
|
|
for i := 0; i < 50; i++ {
|
|
message := Models.Message{
|
|
MessageData: Models.MessageData{
|
|
Data: base64.StdEncoding.EncodeToString(dataCiphertext),
|
|
SenderID: base64.StdEncoding.EncodeToString(senderIDCiphertext),
|
|
SymmetricKey: base64.StdEncoding.EncodeToString(keyCiphertext),
|
|
},
|
|
SymmetricKey: base64.StdEncoding.EncodeToString(
|
|
Seeder.EncryptWithPublicKey(userKey.Key, pubKey),
|
|
),
|
|
AssociationKey: "AssociationKey",
|
|
}
|
|
|
|
err = Database.CreateMessage(&message)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
resp, err := client.Get(ts.URL + "/api/v1/auth/messages/AssociationKey")
|
|
if err != nil {
|
|
t.Errorf("Expected user record, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode)
|
|
return
|
|
}
|
|
|
|
requestBody, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
var m []Models.Message
|
|
err = json.Unmarshal(requestBody, &m)
|
|
|
|
if len(m) != 20 {
|
|
t.Errorf("Expected %d, recieved %d", 20, len(m))
|
|
}
|
|
}
|