Skip to content

Commit 0c282fa

Browse files
committed
Set default values from package manifest earlier
1 parent d2e5303 commit 0c282fa

File tree

1 file changed

+7
-18
lines changed

1 file changed

+7
-18
lines changed

internal/benchrunner/runners/system/runner.go

Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -184,18 +184,19 @@ func (r *runner) setUp(ctx context.Context) error {
184184
return fmt.Errorf("reading data stream manifest failed: %w", err)
185185
}
186186

187+
// Set default values for scenario fields from package manifest if not set
187188
if r.scenario.Version == "" {
188189
r.scenario.Version = pkgManifest.Version
189-
} else {
190-
// If the scenario version is set, override the package manifest version
191-
// This is needed to create the policy with the correct version
192-
pkgManifest.Version = r.scenario.Version
193190
}
194191

195192
if r.scenario.Package == "" {
196193
r.scenario.Package = pkgManifest.Name
197194
}
198195

196+
if r.scenario.PolicyTemplate == "" {
197+
r.scenario.PolicyTemplate = pkgManifest.PolicyTemplates[0].Name
198+
}
199+
199200
policy, err := r.createBenchmarkPolicy(ctx, pkgManifest, defaultNamespace)
200201
if err != nil {
201202
return err
@@ -520,18 +521,6 @@ func (r *runner) createBenchmarkPolicy(ctx context.Context, pkgManifest *package
520521
func (r *runner) createPackagePolicy(ctx context.Context, pkgManifest *packages.PackageManifest, p *kibana.Policy) (*kibana.PackagePolicy, error) {
521522
logger.Debug("creating package policy...")
522523

523-
if r.scenario.Version == "" {
524-
r.scenario.Version = pkgManifest.Version
525-
}
526-
527-
if r.scenario.Package == "" {
528-
r.scenario.Package = pkgManifest.Name
529-
}
530-
531-
if r.scenario.PolicyTemplate == "" {
532-
r.scenario.PolicyTemplate = pkgManifest.PolicyTemplates[0].Name
533-
}
534-
535524
pp := kibana.PackagePolicy{
536525
Namespace: p.Namespace,
537526
PolicyIDs: []string{p.ID},
@@ -541,7 +530,7 @@ func (r *runner) createPackagePolicy(ctx context.Context, pkgManifest *packages.
541530
Enabled: true,
542531
Vars: r.scenario.Vars,
543532
Streams: map[string]kibana.PackagePolicyStream{
544-
fmt.Sprintf("%s.%s", pkgManifest.Name, r.scenario.DataStream.Name): {
533+
fmt.Sprintf("%s.%s", r.scenario.Package, r.scenario.DataStream.Name): {
545534
Enabled: true,
546535
Vars: r.scenario.DataStream.Vars,
547536
},
@@ -564,7 +553,7 @@ func (r *runner) createPackagePolicy(ctx context.Context, pkgManifest *packages.
564553
}
565554
}
566555

567-
pp.Package.Name = pkgManifest.Name
556+
pp.Package.Name = r.scenario.Package
568557
pp.Package.Version = r.scenario.Version
569558

570559
policy, err := r.options.KibanaClient.CreatePackagePolicy(ctx, pp)

0 commit comments

Comments
 (0)