Replicate most of drone-helm's config

internal/helm/plan_test.go

@@ -0,0 +1,53 @@
+package helm
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/suite"
+	"testing"
+
+	"github.com/pelotech/drone-helm3/internal/run"
+)
+
+type PlanTestSuite struct {
+	suite.Suite
+}
+
+func TestPlanTestSuite(t *testing.T) {
+	suite.Run(t, new(PlanTestSuite))
+}
+
+func (suite *PlanTestSuite) TestNewPlanUpgradeCommand() {
+	cfg := Config{
+		Command: "upgrade",
+		Chart:   "billboard_top_100",
+		Release: "post_malone_circles",
+	}
+
+	plan, err := NewPlan(cfg)
+	suite.Require().Nil(err)
+	suite.Equal(1, len(plan.steps))
+
+	switch step := plan.steps[0].(type) {
+	case *run.Upgrade:
+		suite.Equal("billboard_top_100", step.Chart)
+		suite.Equal("post_malone_circles", step.Release)
+	default:
+		suite.Failf("Wrong type for step 1", "Expected Upgrade, got %T", step)
+	}
+}
+
+func (suite *PlanTestSuite) TestNewPlanUpgradeFromDroneEvent() {
+	cfg := Config{
+		Chart:   "billboard_top_100",
+		Release: "lizzo_good_as_hell",
+	}
+
+	upgradeEvents := []string{"push", "tag", "deployment", "pull_request", "promote", "rollback"}
+	for _, event := range upgradeEvents {
+		cfg.DroneEvent = event
+		plan, err := NewPlan(cfg)
+		suite.Require().Nil(err)
+		suite.Require().Equal(1, len(plan.steps), fmt.Sprintf("for event type '%s'", event))
+		suite.IsType(&run.Upgrade{}, plan.steps[0], fmt.Sprintf("for event type '%s'", event))
+	}
+}