54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
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))
|
||
|
}
|
||
|
}
|