package Friends_test import ( "bytes" "encoding/base64" "encoding/json" "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_CreateFriendRequest(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") if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } u2, err := Tests.InitTestCreateUser("test2") 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 } decodedPublicKey := Seeder.GetPubKey() encPublicKey, err := key.AesEncrypt([]byte(Seeder.PublicKey)) if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } friendReq := Models.FriendRequest{ UserID: u.ID, FriendID: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u2.ID.String()), decodedPublicKey, ), ), FriendUsername: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u2.Username), decodedPublicKey, ), ), FriendPublicAsymmetricKey: base64.StdEncoding.EncodeToString( encPublicKey, ), SymmetricKey: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey(key.Key, decodedPublicKey), ), } jsonStr, _ := json.Marshal(friendReq) req, _ := http.NewRequest("POST", ts.URL+"/api/v1/auth/friend_request", 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.StatusOK { t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode) return } var r Models.FriendRequest err = Database.DB.First(&r).Error if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } if r.AcceptedAt.Valid == true { t.Errorf("Expected false, recieved true") return } } func Test_CreateFriendRequestQrCode(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") if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } u2, err := Tests.InitTestCreateUser("test2") 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 } decodedPublicKey := Seeder.GetPubKey() encPublicKey, err := key.AesEncrypt([]byte(Seeder.PublicKey)) if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } friendReq := Models.FriendRequest{ UserID: u.ID, FriendID: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u2.ID.String()), decodedPublicKey, ), ), FriendUsername: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u2.Username), decodedPublicKey, ), ), FriendPublicAsymmetricKey: base64.StdEncoding.EncodeToString( encPublicKey, ), SymmetricKey: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey(key.Key, decodedPublicKey), ), } friendReq2 := Models.FriendRequest{ UserID: u2.ID, FriendID: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u.ID.String()), decodedPublicKey, ), ), FriendUsername: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey( []byte(u.Username), decodedPublicKey, ), ), FriendPublicAsymmetricKey: base64.StdEncoding.EncodeToString( encPublicKey, ), SymmetricKey: base64.StdEncoding.EncodeToString( Seeder.EncryptWithPublicKey(key.Key, decodedPublicKey), ), } jsonStr, _ := json.Marshal([]Models.FriendRequest{friendReq, friendReq2}) req, _ := http.NewRequest("POST", ts.URL+"/api/v1/auth/friend_request/qr_code", 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.StatusOK { t.Errorf("Expected %d, recieved %d", http.StatusOK, resp.StatusCode) return } var r Models.FriendRequest err = Database.DB.First(&r).Error if err != nil { t.Errorf("Expected nil, recieved %s", err.Error()) return } if r.AcceptedAt.Valid == false { t.Errorf("Expected true, recieved false") return } }