Introduce support for MSP load generation

Signed-off-by: Liran Funaro <[email protected]>

Author: liran-funaro

cmd/config/app_config_test.go

@@ -379,20 +379,20 @@ func TestReadConfigLoadGen(t *testing.T) {
 				Block: workload.BlockProfile{Size: 500},
 				Transaction: workload.TransactionProfile{
 					ReadWriteCount: workload.NewConstantDistribution(2),
-					Policy: &workload.PolicyProfile{
-						ChannelID: "mychannel",
-						NamespacePolicies: map[string]*workload.Policy{
-							workload.DefaultGeneratedNamespaceID: {
-								Scheme: signature.Ecdsa, Seed: 10,
-							},
-							committerpb.MetaNamespaceID: {
-								Scheme: signature.Ecdsa, Seed: 11,
-							},
+				},
+				Policy: workload.PolicyProfile{
+					ChannelID: "mychannel",
+					NamespacePolicies: map[string]*workload.Policy{
+						workload.DefaultGeneratedNamespaceID: {
+							Scheme: signature.Ecdsa, Seed: 10,
 						},
-						OrdererEndpoints: []*commontypes.OrdererEndpoint{
-							newOrdererEndpoint("org", "orderer"),
+						committerpb.MetaNamespaceID: {
+							Scheme: signature.Ecdsa, Seed: 11,
 						},
 					},
+					OrdererEndpoints: []*commontypes.OrdererEndpoint{
+						newOrdererEndpoint("org", "orderer"),
+					},
 				},
 				Conflicts: workload.ConflictProfile{
 					InvalidSignatures: 0.1,

cmd/config/samples/loadgen.yaml

@@ -48,17 +48,17 @@ load-profile:
   transaction:
     read-write-count:
       const: 2
-    policy:
-      channel-id: mychannel
-      namespace-policies:
-        0:
-          scheme: ECDSA
-          seed: 10
-        _meta:
-          scheme: ECDSA
-          seed: 11
-      orderer-endpoints:
-        - id=0,msp-id=org,broadcast,deliver,orderer:7050
+  policy:
+    channel-id: mychannel
+    namespace-policies:
+      0:
+        scheme: ECDSA
+        seed: 10
+      _meta:
+        scheme: ECDSA
+        seed: 11
+    orderer-endpoints:
+      - id=0,msp-id=org,broadcast,deliver,orderer:7050
   conflicts:
     invalid-signatures: 0.1
   seed: 12345

cmd/config/templates/loadgen_common.yaml

@@ -38,15 +38,15 @@ load-profile:
     read-only-count:
     read-write-count:
       const: 2
-    policy:
-      channel-id: {{ .Policy.ChannelID }}
-      namespace-policies:
-        {{ range $nsID, $element := .Policy.NamespacePolicies }}
-        {{ $nsID }}:
-          scheme: {{ $element.Scheme }}
-          seed: {{ $element.Seed }}
-        {{ end }}
-      config-block-path: {{ .ConfigBlockPath }}
+  policy:
+    channel-id: {{ .Policy.ChannelID }}
+    namespace-policies:
+      {{ range $nsID, $element := .Policy.NamespacePolicies }}
+      {{ $nsID }}:
+        scheme: {{ $element.Scheme }}
+        seed: {{ $element.Seed }}
+      {{ end }}
+    config-block-path: {{ .ConfigBlockPath }}
   query:
     query-size:
     min-invalid-keys-portion:

cmd/loadgen/main.go

@@ -100,7 +100,7 @@ func loadGenGenesisBlock() *cobra.Command {
 			}
 			cmd.SilenceUsage = true
 
-			block, err := workload.CreateConfigBlock(conf.LoadProfile.Transaction.Policy)
+			block, err := workload.CreateConfigBlock(&conf.LoadProfile.Policy)
 			if err != nil {
 				return err
 			}

docker/images/test_node/Dockerfile

@@ -20,7 +20,7 @@ ENV SC_SIDECAR_COMMITTER_ENDPOINT=localhost:9001
 ENV SC_QUERY_DATABASE_ENDPOINTS=localhost:5433
 ENV SC_VC_DATABASE_ENDPOINTS=localhost:5433
 ENV SC_LOADGEN_ORDERER_CLIENT_ORDERER_CONNECTION_ENDPOINTS=localhost:7050
-ENV SC_LOADGEN_LOAD_PROFILE_TRANSACTION_POLICY_ORDERER_ENDPOINTS="id=0,msp-id=org,broadcast,deliver,localhost:7050"
+ENV SC_LOADGEN_LOAD_PROFILE_POLICY_ORDERER_ENDPOINTS="id=0,msp-id=org,broadcast,deliver,localhost:7050"
 ENV SC_LOADGEN_ORDERER_CLIENT_SIDECAR_CLIENT_ENDPOINT=localhost:4001
 
 # Disable TLS usage for db.

docker/test/container_release_image_test.go

@@ -66,7 +66,8 @@ func TestCommitterReleaseImagesWithTLS(t *testing.T) {
 	v := config.NewViperWithLoadGenDefaults()
 	c, err := config.ReadLoadGenYamlAndSetupLogging(v, filepath.Join(localConfigPath, "loadgen.yaml"))
 	require.NoError(t, err)
-	configBlock, err := workload.CreateConfigBlock(c.LoadProfile.Transaction.Policy)
+	c.LoadProfile.Policy.CryptoMaterialPath = t.TempDir()
+	configBlock, err := workload.CreateConfigBlock(&c.LoadProfile.Policy)
 	require.NoError(t, err)
 	require.NoError(t, configtxgen.WriteOutputBlock(configBlock, configBlockPath))
 

docker/test/container_test.go

@@ -76,7 +76,7 @@ func TestStartTestNodeWithTLSModesAndRemoteConnection(t *testing.T) {
 			c, err := config.ReadLoadGenYamlAndSetupLogging(v, filepath.Join(localConfigPath, "loadgen.yaml"))
 			require.NoError(t, err)
 			ordererEp := mustGetEndpoint(ctx, t, containerName, mockOrdererPort)
-			c.LoadProfile.Transaction.Policy.OrdererEndpoints = []*commontypes.OrdererEndpoint{
+			c.LoadProfile.Policy.OrdererEndpoints = []*commontypes.OrdererEndpoint{
 				{
 					Host: ordererEp.Host, Port: ordererEp.Port, ID: 0, MspID: "org",
 					API: []string{commontypes.Broadcast, commontypes.Deliver},
@@ -98,7 +98,7 @@ func TestStartTestNodeWithTLSModesAndRemoteConnection(t *testing.T) {
 							Server: mustGetEndpoint(ctx, t, containerName, coordinatorServicePort),
 						},
 					},
-					Policy: c.LoadProfile.Transaction.Policy,
+					Policy: &c.LoadProfile.Policy,
 				},
 				DBEnv: vc.NewDatabaseTestEnvFromConnection(
 					t,

integration/runner/runtime.go

@@ -135,8 +135,9 @@ func NewRuntime(t *testing.T, conf *Config) *CommitterRuntime {
 			LoadGenBlockLimit: conf.LoadgenBlockLimit,
 			LoadGenWorkers:    1,
 			Policy: &workload.PolicyProfile{
-				ChannelID:         TestChannelName,
-				NamespacePolicies: make(map[string]*workload.Policy),
+				ChannelID:          TestChannelName,
+				NamespacePolicies:  make(map[string]*workload.Policy),
+				CryptoMaterialPath: t.TempDir(),
 			},
 			Logging: &logging.DefaultConfig,
 		},

integration/test/config_update_test.go

@@ -86,7 +86,8 @@ func TestConfigUpdate(t *testing.T) {
 	lgMetaTx := c.TxBuilder.MakeTx(metaTx)
 
 	c.AddOrUpdateNamespaces(t, committerpb.MetaNamespaceID)
-	metaPolicy := c.TxBuilder.TxEndorser.Policy(committerpb.MetaNamespaceID).VerificationPolicy()
+	verPolicies := c.TxBuilder.TxEndorser.VerificationPolicies()
+	metaPolicy := verPolicies[committerpb.MetaNamespaceID]
 	submitConfigBlock := func(endpoints []*commontypes.OrdererEndpoint) {
 		ordererEnv.SubmitConfigBlock(t, &workload.ConfigBlock{
 			ChannelID:                    c.SystemConfig.Policy.ChannelID,

integration/test/namespace_lifecycle_test.go

@@ -29,9 +29,10 @@ func TestCreateUpdateNamespace(t *testing.T) {
 	})
 	c.Start(t, runner.FullTxPath)
 
-	policyBytesNs1, err := proto.Marshal(c.TxBuilder.TxEndorser.Policy("1").VerificationPolicy())
+	verPolicies := c.TxBuilder.TxEndorser.VerificationPolicies()
+	policyBytesNs1, err := proto.Marshal(verPolicies["1"])
 	require.NoError(t, err)
-	policyBytesNs2, err := proto.Marshal(c.TxBuilder.TxEndorser.Policy("2").VerificationPolicy())
+	policyBytesNs2, err := proto.Marshal(verPolicies["2"])
 	require.NoError(t, err)
 
 	tests := []struct {