Fix panic if preRun or postRun is null

bot/sequential.go

@@ -57,8 +57,8 @@ func NewSequentialBot(
 // Initialize initializes the bot
 func (b *SequentialBot) Initialize() error {
 	b.logger.Debug("Initializing bot")
-	pre := custom.GetPre(b.config, b.spec)
-	storage, err := pre.Run(b.spec.PreRun.Args)
+	pre, args := custom.GetPre(b.config, b.spec)
+	storage, err := pre.Run(args)
 	if err != nil {
 		return err
 	}
@@ -216,8 +216,8 @@ func (b *SequentialBot) runOperation(op *models.Operation) error {
 // Finalize finalizes the bot
 func (b *SequentialBot) Finalize() error {
 	b.logger.Debug("Finalizing bot")
-	post := custom.GetPost(b.config, b.spec)
-	if err := post.Run(b.spec.PostRun.Args, b.storage); err != nil {
+	post, args := custom.GetPost(b.config, b.spec)
+	if err := post.Run(args, b.storage); err != nil {
 		return err
 	}
 	b.logger.Debugf("Saved storage")

custom/custom.go

@@ -45,27 +45,27 @@ func (d *DummyPost) Run(args map[string]interface{}, store storage.Storage) erro
 }
 
 // GetPre returns the pre operation for the spec, if it exists
-func GetPre(config *viper.Viper, spec *models.Spec) PreOperation {
+func GetPre(config *viper.Viper, spec *models.Spec) (PreOperation, map[string]interface{}) {
 	if spec.PreRun == nil {
-		return &DummyPre{}
+		return &DummyPre{}, map[string]interface{}{}
 	}
 
 	switch spec.PreRun.Function {
 	case PreRunFunctionRedis:
-		return redis.GetPre(config)
+		return redis.GetPre(config), spec.PreRun.Args
 	}
-	return &DummyPre{}
+	return &DummyPre{}, map[string]interface{}{}
 }
 
 // GetPost returns the post operation for the spec, if it exists
-func GetPost(config *viper.Viper, spec *models.Spec) PostOperation {
+func GetPost(config *viper.Viper, spec *models.Spec) (PostOperation, map[string]interface{}) {
 	if spec.PostRun == nil {
-		return &DummyPost{}
+		return &DummyPost{}, map[string]interface{}{}
 	}
 
 	switch spec.PostRun.Function {
 	case PostRunFunctionRedis:
-		return redis.GetPost(config)
+		return redis.GetPost(config), spec.PostRun.Args
 	}
-	return &DummyPost{}
+	return &DummyPost{}, map[string]interface{}{}
 }