chore: cleanup FGA server resources (#245)

internal/storetest/localstore.go

@@ -73,35 +73,42 @@ func initLocalStore(
 func getLocalServerModelAndTuples(
 	storeData *StoreData,
 	format authorizationmodel.ModelFormat,
-) (*server.Server, *authorizationmodel.AuthzModel, error) {
+) (*server.Server, *authorizationmodel.AuthzModel, func(), error) {
 	var fgaServer *server.Server
 
 	var authModel *authorizationmodel.AuthzModel
 
+	stopServerFn := func() {}
+
 	if storeData.Model == "" {
-		return fgaServer, authModel, nil
+		return fgaServer, authModel, stopServerFn, nil
 	}
 
 	// If we have at least one local test, initialize the local server
 	datastore := memory.New()
 
 	fgaServer, err := server.NewServerWithOpts(server.WithDatastore(datastore))
 	if err != nil {
-		return nil, nil, err //nolint:wrapcheck
+		return nil, nil, stopServerFn, err //nolint:wrapcheck
 	}
 
 	tempModel := authorizationmodel.AuthzModel{}
 	if format == authorizationmodel.ModelFormatJSON {
 		if err := tempModel.ReadFromJSONString(storeData.Model); err != nil {
-			return nil, nil, err //nolint:wrapcheck
+			return nil, nil, stopServerFn, err //nolint:wrapcheck
 		}
 	} else {
 		if err := tempModel.ReadFromDSLString(storeData.Model); err != nil {
-			return nil, nil, err //nolint:wrapcheck
+			return nil, nil, stopServerFn, err //nolint:wrapcheck
 		}
 	}
 
 	authModel = &tempModel
 
-	return fgaServer, authModel, nil
+	stopServerFn = func() {
+		datastore.Close()
+		fgaServer.Close()
+	}
+
+	return fgaServer, authModel, stopServerFn, nil
 }

internal/storetest/tests.go

@@ -36,11 +36,13 @@ func RunTests(
 ) (TestResults, error) {
 	test := TestResults{}
 
-	fgaServer, authModel, err := getLocalServerModelAndTuples(storeData, format)
+	fgaServer, authModel, stopServerFn, err := getLocalServerModelAndTuples(storeData, format)
 	if err != nil {
 		return test, err
 	}
 
+	defer stopServerFn()
+
 	for index := 0; index < len(storeData.Tests); index++ {
 		result, err := RunTest(
 			fgaClient,