update tests to allow acting as a wallet with sig headers

Author: schottra

api/server_test.go

@@ -9,6 +9,7 @@ import (
 	"strings"
 	"testing"
 
+	"bridgerton.audius.co/api/testdata"
 	"bridgerton.audius.co/config"
 	"github.com/jackc/pgx/v5"
 	"github.com/stretchr/testify/assert"
@@ -87,12 +88,10 @@ func TestHome(t *testing.T) {
 	assert.True(t, strings.Contains(string(body), "uptime"))
 }
 
-func Test200(t *testing.T) {
+func Test200UnAuthed(t *testing.T) {
 	urls := []string{
 		"/v1/full/users?id=7eP5n&id=_some_invalid_hash_id",
 
-		"/v1/full/users/account/0x7d273271690538cf855e5b3002a0dd8c154bb060",
-
 		"/v1/full/users/7eP5n",
 		"/v1/full/users/7eP5n/followers",
 		"/v1/full/users/7eP5n/following",
@@ -148,6 +147,28 @@ func Test200(t *testing.T) {
 	}
 }
 
+func Test200Authed(t *testing.T) {
+	urls := []string{
+		"/v1/full/users/account/0x7d273271690538cf855e5b3002a0dd8c154bb060",
+	}
+
+	for _, u := range urls {
+		status, body := testGetWithWallet(t, u, "0x7d273271690538cf855e5b3002a0dd8c154bb060")
+		require.Equal(t, 200, status, u+" "+string(body))
+
+		// also test as a user
+		if strings.Contains(u, "?") {
+			u += "&user_id=7eP5n"
+		} else {
+			u += "?user_id=7eP5n"
+		}
+
+		status, _ = testGetWithWallet(t, u, "0x7d273271690538cf855e5b3002a0dd8c154bb060")
+		require.Equal(t, 200, status, u+" "+string(body))
+	}
+
+}
+
 func testGet(t *testing.T, path string, dest ...any) (int, []byte) {
 	req := httptest.NewRequest("GET", path, nil)
 	res, err := app.Test(req, -1)
@@ -161,3 +182,28 @@ func testGet(t *testing.T, path string, dest ...any) (int, []byte) {
 
 	return res.StatusCode, body
 }
+
+// testGetWithWallet makes a GET request with authentication headers for the given wallet address
+func testGetWithWallet(t *testing.T, path string, walletAddress string, dest ...any) (int, []byte) {
+	req := httptest.NewRequest("GET", path, nil)
+
+	// Add signature headers if wallet address is provided
+	if walletAddress != "" {
+		sigData, exists := testdata.GetSignatureData(walletAddress)
+		if exists {
+			req.Header.Set("Encoded-Data-Message", sigData.Message)
+			req.Header.Set("Encoded-Data-Signature", sigData.Signature)
+		}
+	}
+
+	res, err := app.Test(req, -1)
+	assert.NoError(t, err)
+	body, _ := io.ReadAll(res.Body)
+
+	if len(dest) > 0 {
+		err = json.Unmarshal(body, &dest[0])
+		assert.NoError(t, err)
+	}
+
+	return res.StatusCode, body
+}

api/testdata/signatures.go

@@ -0,0 +1,21 @@
+package testdata
+
+// SignatureData represents a message and its corresponding signature for a wallet
+type SignatureData struct {
+	Message   string
+	Signature string
+}
+
+// TestSignatures contains a mapping of wallet addresses to their corresponding message and signature pairs
+var TestSignatures = map[string]SignatureData{
+	"0x7d273271690538cf855e5b3002a0dd8c154bb060": {
+		Message:   "signature:1744763856446",
+		Signature: "0xbb202be3a7f3a0aa22c1458ef6a3f2f8360fb86791c7b137e8562df0707825c11fa1db01096efd2abc5e6613c4d1e8d4ae1e2b993abdd555fe270c1b17bff0d21c",
+	},
+}
+
+// GetSignatureData returns the message and signature for a given test wallet address
+func GetSignatureData(walletAddress string) (SignatureData, bool) {
+	data, exists := TestSignatures[walletAddress]
+	return data, exists
+}

api/testdata/user_fixtures.csv

@@ -1,5 +1,5 @@
-user_id,handle,handle_lc,is_deactivated,wallet
-1,rayjacobson,rayjacobson,f,0x7d273271690538cf855e5b3002a0dd8c154bb060
-2,stereosteve,stereosteve,f,
-3,someseller,someseller,f,
-91,badguy,badguy,t,
+user_id,handle,handle_lc,is_deactivated,wallet,playlist_library
+1,rayjacobson,rayjacobson,f,0x7d273271690538cf855e5b3002a0dd8c154bb060,"{""contents"":[{""type"":""playlist123"",""playlist_id"":""123""},{""type"":""explore_playlist"",""playlist_id"":""Audio NFTs""},{""type"":""folder"",""id"":""bbcae31a-7cd2-4a1a-8b54-fdc979a34435"",""name"":""My Nested Playlists"",""contents"":[{""type"":""playlist"",""playlist_id"":""345""}]}]}"
+2,stereosteve,stereosteve,f,0x1234567890abcdef,
+3,someseller,someseller,f,0x234567890abcdef1,
+91,badguy,badguy,t,0x4567890abcdef123,