package cli_test

import (

"context"

"path/filepath"

"testing"

"github.com/google/uuid"

"github.com/stretchr/testify/assert"

"github.com/stretchr/testify/require"

"github.com/coder/coder/cli/clitest"

"github.com/coder/coder/coderd/coderdtest"

"github.com/coder/coder/coderd/database"

"github.com/coder/coder/codersdk"

"github.com/coder/coder/provisioner/echo"

"github.com/coder/coder/pty/ptytest"

)

func TestTemplatePush(t *testing.T) {

t.Parallel()

// NewParameter will:

// 1. Create a template version with 0 params

// 2. Create a new version with 1 param

// 2a. Expects 1 param prompt, fills in value

// 3. Assert 1 param value in new version

// 4. Creates a new version with same param

// 4a. Expects 0 prompts as the param value is carried over

// 5. Assert 1 param value in new version

// 6. Creates a new version with 0 params

// 7. Asset 0 params in new version

t.Run("NewParameter", func(t *testing.T) {

t.Parallel()

client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})

user := coderdtest.CreateFirstUser(t, client)

// Create initial template version to update

lastActiveVersion := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)

_ = coderdtest.AwaitTemplateVersionJob(t, client, lastActiveVersion.ID)

template := coderdtest.CreateTemplate(t, client, user.OrganizationID, lastActiveVersion.ID)

// Create new template version with a new parameter

source := clitest.CreateTemplateVersionSource(t, &echo.Responses{

Parse: createTestParseResponse(),

ProvisionApply: echo.ProvisionComplete,

})

cmd, root := clitest.New(t, "templates", "push", template.Name, "-y", "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho))

clitest.SetupConfig(t, client, root)

pty := ptytest.New(t)

cmd.SetIn(pty.Input())

cmd.SetOut(pty.Output())

execDone := make(chan error)

go func() {

execDone <- cmd.Execute()

}()

matches := []struct {

match string

write string

}{

// Expect to be prompted for the new param

{match: "Enter a value:", write: "peter-pan"},

}

for _, m := range matches {

pty.ExpectMatch(m.match)

pty.WriteLine(m.write)

}

require.NoError(t, <-execDone)

// Assert template version changed and we have the new param

latestTV, latestParams := latestTemplateVersion(t, client, template.ID)

assert.NotEqual(t, lastActiveVersion.ID, latestTV.ID)

require.Len(t, latestParams, 1, "expect 1 param")

lastActiveVersion = latestTV

// Second update of the same source requires no prompt since the params

// are carried over.

cmd, root = clitest.New(t, "templates", "push", template.Name, "-y", "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho))

clitest.SetupConfig(t, client, root)

go func() {

execDone <- cmd.Execute()

}()

require.NoError(t, <-execDone)

// Assert template version changed and we have the carried over param

latestTV, latestParams = latestTemplateVersion(t, client, template.ID)

assert.NotEqual(t, lastActiveVersion.ID, latestTV.ID)

require.Len(t, latestParams, 1, "expect 1 param")

lastActiveVersion = latestTV

// Remove the param

source = clitest.CreateTemplateVersionSource(t, &echo.Responses{

Parse: echo.ParseComplete,

ProvisionApply: echo.ProvisionComplete,

})

cmd, root = clitest.New(t, "templates", "push", template.Name, "-y", "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho))

clitest.SetupConfig(t, client, root)

go func() {

execDone <- cmd.Execute()

}()

require.NoError(t, <-execDone)

// Assert template version changed and the param was removed

latestTV, latestParams = latestTemplateVersion(t, client, template.ID)

assert.NotEqual(t, lastActiveVersion.ID, latestTV.ID)

require.Len(t, latestParams, 0, "expect 0 param")

lastActiveVersion = latestTV

})

t.Run("OK", func(t *testing.T) {

t.Parallel()

client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})

user := coderdtest.CreateFirstUser(t, client)

version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)

_ = coderdtest.AwaitTemplateVersionJob(t, client, version.ID)

template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)

// Test the cli command.

source := clitest.CreateTemplateVersionSource(t, &echo.Responses{

Parse: echo.ParseComplete,

ProvisionApply: echo.ProvisionComplete,

})

cmd, root := clitest.New(t, "templates", "push", template.Name, "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho), "--name", "example")

clitest.SetupConfig(t, client, root)

pty := ptytest.New(t)

cmd.SetIn(pty.Input())

cmd.SetOut(pty.Output())

execDone := make(chan error)

go func() {

execDone <- cmd.Execute()

}()

matches := []struct {

match string

write string

}{

{match: "Upload", write: "yes"},

}

for _, m := range matches {

pty.ExpectMatch(m.match)

pty.WriteLine(m.write)

}

require.NoError(t, <-execDone)

// Assert that the template version changed.

templateVersions, err := client.TemplateVersionsByTemplate(context.Background(), codersdk.TemplateVersionsByTemplateRequest{

TemplateID: template.ID,

})

require.NoError(t, err)

assert.Len(t, templateVersions, 2)

assert.NotEqual(t, template.ActiveVersionID, templateVersions[1].ID)

require.Equal(t, "example", templateVersions[1].Name)

})

t.Run("UseWorkingDir", func(t *testing.T) {

t.Parallel()

client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})

user := coderdtest.CreateFirstUser(t, client)

version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)

_ = coderdtest.AwaitTemplateVersionJob(t, client, version.ID)

// Test the cli command.

source := clitest.CreateTemplateVersionSource(t, &echo.Responses{

Parse: echo.ParseComplete,

ProvisionApply: echo.ProvisionComplete,

})

template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID,

func(r *codersdk.CreateTemplateRequest) {

r.Name = filepath.Base(source)

})

// Don't pass the name of the template, it should use the

// directory of the source.

cmd, root := clitest.New(t, "templates", "push", "--directory", source, "--test.provisioner", string(database.ProvisionerTypeEcho))

clitest.SetupConfig(t, client, root)

pty := ptytest.New(t)

cmd.SetIn(pty.Input())

cmd.SetOut(pty.Output())

execDone := make(chan error)

go func() {

execDone <- cmd.Execute()

}()

matches := []struct {

match string

write string

}{

{match: "Upload", write: "yes"},

}

for _, m := range matches {

pty.ExpectMatch(m.match)

pty.WriteLine(m.write)

}

require.NoError(t, <-execDone)

// Assert that the template version changed.

templateVersions, err := client.TemplateVersionsByTemplate(context.Background(), codersdk.TemplateVersionsByTemplateRequest{

TemplateID: template.ID,

})

require.NoError(t, err)

assert.Len(t, templateVersions, 2)

assert.NotEqual(t, template.ActiveVersionID, templateVersions[1].ID)

})

}

func latestTemplateVersion(t *testing.T, client *codersdk.Client, templateID uuid.UUID) (codersdk.TemplateVersion, []codersdk.Parameter) {

t.Helper()

ctx := context.Background()

newTemplate, err := client.Template(ctx, templateID)

require.NoError(t, err)

tv, err := client.TemplateVersion(ctx, newTemplate.ActiveVersionID)

require.NoError(t, err)

params, err := client.Parameters(ctx, codersdk.ParameterImportJob, tv.Job.ID)

require.NoError(t, err)

return tv, params

}