added C chain config via env variables

Author: aleksandarknezevic

tests/antithesis/compose.go

@@ -4,6 +4,7 @@
 package antithesis
 
 import (
+	"encoding/base64"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -27,17 +28,23 @@ const (
 	imageTagEnvName   = "IMAGE_TAG"
 )
 
+type CChainConfig struct {
+	LogJSONFormat bool `json:"log-json-format"`
+}
+
+type ChainConfig struct {
+	C struct {
+		Config string `json:"Config"`
+	} `json:"C"`
+}
+
 var (
 	errTargetPathEnvVarNotSet = errors.New(targetPathEnvName + " environment variable not set")
 	errImageTagEnvVarNotSet   = errors.New(imageTagEnvName + " environment variable not set")
 	errAvalancheGoEvVarNotSet = errors.New(tmpnet.AvalancheGoPathEnvName + " environment variable not set")
 	errPluginDirEnvVarNotSet  = errors.New(tmpnet.AvalancheGoPluginDirEnvName + " environment variable not set")
 )
 
-type cChainConfig struct {
-	LogJSONFormat bool `json:"log-json-format"`
-}
-
 // Creates docker compose configuration for an antithesis test setup. Configuration is via env vars to
 // simplify usage by main entrypoints. If the provided network includes a subnet, the initial DB state for
 // the subnet will be created and written to the target path.
@@ -129,18 +136,6 @@ func initComposeConfig(
 			if err := os.MkdirAll(volumePath, perms.ReadWriteExecute); err != nil {
 				return fmt.Errorf("failed to create volume path %q: %w", volumePath, err)
 			}
-			// Write config.json if this is the C-Chain config directory
-			if isCChainConfigDir(volumePath) {
-				configFilePath := filepath.Join(volumePath, "config.json")
-				cfg := cChainConfig{LogJSONFormat: true}
-				configContent, err := json.MarshalIndent(cfg, "", "  ")
-				if err != nil {
-					return fmt.Errorf("failed to marshal config.json: %w", err)
-				}
-				if err := os.WriteFile(configFilePath, configContent, perms.ReadWrite); err != nil {
-					return fmt.Errorf("failed to write config.json to %q: %w", configFilePath, err)
-				}
-			}
 		}
 	}
 	return nil
@@ -158,6 +153,9 @@ func newComposeProject(network *tmpnet.Network, nodeImageName string, workloadIm
 		bootstrapIP  string
 		bootstrapIDs string
 	)
+
+	chainConfigB64, _ := encodeChainConfig()
+
 	for i, node := range network.Nodes {
 		address := fmt.Sprintf("%s.%d", baseNetworkAddress, 3+i)
 
@@ -177,6 +175,10 @@ func newComposeProject(network *tmpnet.Network, nodeImageName string, workloadIm
 			config.StakingSignerKeyContentKey: signerKey,
 		}
 
+		if chainConfigB64 != "" {
+			env[config.ChainConfigContentKey] = chainConfigB64
+		}
+
 		// Apply configuration appropriate to a test network
 		for k, v := range tmpnet.DefaultTmpnetFlags() {
 			env[k] = v
@@ -190,11 +192,6 @@ func newComposeProject(network *tmpnet.Network, nodeImageName string, workloadIm
 				Source: fmt.Sprintf("./volumes/%s/logs", serviceName),
 				Target: "/root/.avalanchego/logs",
 			},
-			{
-				Type:   types.VolumeTypeBind,
-				Source: fmt.Sprintf("./volumes/%s/configs/chains/C", serviceName),
-				Target: "/root/.avalanchego/configs/chains/C",
-			},
 		}
 
 		trackSubnets := node.Flags[config.TrackSubnetsKey]
@@ -305,10 +302,20 @@ func getServiceName(index int) string {
 	return fmt.Sprintf("%s-node-%d", baseName, index)
 }
 
-// isCChainConfigDir returns true if the given path matches the expected C-Chain config directory structure:
-// .../volumes/<serviceName>/configs/chains/C
-func isCChainConfigDir(path string) bool {
-	return filepath.Base(path) == "C" &&
-		filepath.Base(filepath.Dir(path)) == "chains" &&
-		filepath.Base(filepath.Dir(filepath.Dir(path))) == "configs"
+func encodeChainConfig() (string, error) {
+	cchainConfig := CChainConfig{LogJSONFormat: true}
+	cchainConfigBytes, err := json.Marshal(cchainConfig)
+	if err != nil {
+		return "", err
+	}
+	cchainConfigB64 := base64.StdEncoding.EncodeToString(cchainConfigBytes)
+
+	var chainConfig ChainConfig
+	chainConfig.C.Config = cchainConfigB64
+	chainConfigBytes, err := json.Marshal(chainConfig)
+	if err != nil {
+		return "", err
+	}
+	chainConfigB64 := base64.StdEncoding.EncodeToString(chainConfigBytes)
+	return chainConfigB64, nil
 }