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.
79 lines
1.8 KiB
79 lines
1.8 KiB
package Auth_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
|
|
"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"
|
|
)
|
|
|
|
func Test_AddProfileImage(t *testing.T) {
|
|
client, ts, err := Tests.InitTestEnv()
|
|
defer ts.Close()
|
|
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
|
|
}
|
|
|
|
dat, err := os.ReadFile("./profile_picture_test.png")
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
encDat, err := key.AesEncrypt(dat)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
a := Models.Attachment{
|
|
Mimetype: "image/png",
|
|
Extension: "png",
|
|
Data: base64.StdEncoding.EncodeToString(encDat),
|
|
}
|
|
|
|
jsonStr, _ := json.Marshal(a)
|
|
req, _ := http.NewRequest("POST", ts.URL+"/api/v1/auth/image", 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
|
|
}
|
|
|
|
u, err := Database.GetUserByUsername("test")
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
|
|
if u.AttachmentID.IsNil() {
|
|
t.Errorf("Attachment not assigned to user")
|
|
}
|
|
|
|
err = os.Remove("/app/attachments/" + u.Attachment.FilePath)
|
|
if err != nil {
|
|
t.Errorf("Expected nil, recieved %s", err.Error())
|
|
return
|
|
}
|
|
}
|