associate project asset

server/e2e/common.go

@@ -8,6 +8,7 @@ import (
 	"io"
 	"net"
 	"net/http"
+	"reflect"
 	"regexp"
 	"strings"
 	"testing"
@@ -31,13 +32,15 @@ import (
 	"golang.org/x/text/language"
 )
 
-type Seeder func(ctx context.Context, r *repo.Container) error
+var fr *gateway.File
+
+type Seeder func(ctx context.Context, r *repo.Container, f gateway.File) error
 
 func init() {
 	mongotest.Env = "REEARTH_DB"
 }
 
-func initRepos(t *testing.T, useMongo bool, seeder Seeder) (repos *repo.Container) {
+func initRepos(t *testing.T, useMongo bool, seeder Seeder) (repos *repo.Container, file gateway.File) {
 	ctx := context.Background()
 
 	if useMongo {
@@ -48,18 +51,25 @@ func initRepos(t *testing.T, useMongo bool, seeder Seeder) (repos *repo.Containe
 		repos = memory.New()
 	}
 
+	file = lo.Must(fs.NewFile(afero.NewMemMapFs(), "https://example.com/"))
+	fr = &file
 	if seeder != nil {
-		if err := seeder(ctx, repos); err != nil {
+		if err := seeder(ctx, repos, file); err != nil {
 			t.Fatalf("failed to seed the db: %s", err)
 		}
 	}
 
-	return repos
+	return repos, file
 }
 
 func initGateway() *gateway.Container {
+	if fr == nil {
+		return &gateway.Container{
+			File: lo.Must(fs.NewFile(afero.NewMemMapFs(), "https://example.com/")),
+		}
+	}
 	return &gateway.Container{
-		File: lo.Must(fs.NewFile(afero.NewMemMapFs(), "https://example.com")),
+		File: *fr,
 	}
 }
 
@@ -123,7 +133,7 @@ func StartGQLServerAndRepos(t *testing.T, seeder Seeder) (*httpexpect.Expect, *a
 			Disabled: true,
 		},
 	}
-	repos := initRepos(t, true, seeder)
+	repos, _ := initRepos(t, true, seeder)
 	e, _, _ := StartGQLServerWithRepos(t, cfg, repos)
 	return e, repos.AccountRepos()
 }
@@ -175,7 +185,7 @@ func StartServerWithRepos(t *testing.T, cfg *config.Config, repos *repo.Containe
 }
 
 func StartServerAndRepos(t *testing.T, cfg *config.Config, useMongo bool, seeder Seeder) (*httpexpect.Expect, *repo.Container, *gateway.Container) {
-	repos := initRepos(t, useMongo, seeder)
+	repos, _ := initRepos(t, useMongo, seeder)
 	e, gateways := StartServerWithRepos(t, cfg, repos)
 	return e, repos, gateways
 }
@@ -203,8 +213,8 @@ func ServerLanguage(t *testing.T, lang language.Tag) *httpexpect.Expect {
 		},
 	}
 	return StartServer(t, c, true,
-		func(ctx context.Context, r *repo.Container) error {
-			return baseSeederWithLang(ctx, r, lang)
+		func(ctx context.Context, r *repo.Container, f gateway.File) error {
+			return baseSeederWithLang(ctx, r, f, lang)
 		},
 	)
 }
@@ -275,6 +285,12 @@ func RegexpJSONEReadCloser(t *testing.T, actual io.ReadCloser, expected string)
 	actualBuf := new(bytes.Buffer)
 	_, err := actualBuf.ReadFrom(actual)
 	assert.NoError(t, err)
+	// var data map[string]interface{}
+	// err = json.Unmarshal(actualBuf.Bytes(), &data)
+	// assert.NoError(t, err)
+	// if text, err := json.MarshalIndent(data, "", "  "); err == nil {
+	// 	fmt.Println(string(text))
+	// }
 	return JSONEqRegexp(t, actualBuf.String(), expected)
 }
 
@@ -284,6 +300,13 @@ func JSONEqRegexpInterface(t *testing.T, actual interface{}, expected string) bo
 	return JSONEqRegexp(t, string(actualBytes), expected)
 }
 
+func JSONEqRegexpValue(t *testing.T, actual *httpexpect.Value, expected string) bool {
+	if actualData, ok := actual.Raw().(map[string]interface{}); ok {
+		return JSONEqRegexpInterface(t, actualData, expected)
+	}
+	return false
+}
+
 func aligningJSON(t *testing.T, str string) string {
 	// Unmarshal and Marshal to make the JSON format consistent
 	var obj interface{}
@@ -295,9 +318,17 @@ func aligningJSON(t *testing.T, str string) string {
 }
 
 func ValueDump(val *httpexpect.Value) {
-	if data, ok := val.Raw().(map[string]interface{}); ok {
+	raw := val.Raw()
+	switch data := raw.(type) {
+	case map[string]interface{}:
+		if text, err := json.MarshalIndent(data, "", "  "); err == nil {
+			fmt.Println(string(text))
+		}
+	case []interface{}:
 		if text, err := json.MarshalIndent(data, "", "  "); err == nil {
 			fmt.Println(string(text))
 		}
+	default:
+		fmt.Println("Unsupported type:", reflect.TypeOf(raw))
 	}
 }