2022-03-23 15:03:28 +00:00
|
|
|
package cli_test
|
|
|
|
|
|
|
|
import (
|
2022-05-19 12:49:40 +00:00
|
|
|
"fmt"
|
2022-03-23 15:03:28 +00:00
|
|
|
"testing"
|
|
|
|
|
2022-03-24 01:12:40 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
2022-03-23 15:03:28 +00:00
|
|
|
"github.com/coder/coder/cli/clitest"
|
|
|
|
"github.com/coder/coder/coderd/coderdtest"
|
2022-04-11 23:54:30 +00:00
|
|
|
"github.com/coder/coder/provisioner/echo"
|
|
|
|
"github.com/coder/coder/provisionersdk/proto"
|
2022-03-23 15:03:28 +00:00
|
|
|
"github.com/coder/coder/pty/ptytest"
|
|
|
|
)
|
|
|
|
|
2022-05-02 16:08:52 +00:00
|
|
|
func TestCreate(t *testing.T) {
|
2022-03-23 15:03:28 +00:00
|
|
|
t.Parallel()
|
|
|
|
t.Run("Create", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2022-05-19 22:47:45 +00:00
|
|
|
client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
|
2022-03-23 15:03:28 +00:00
|
|
|
user := coderdtest.CreateFirstUser(t, client)
|
2022-04-06 17:42:40 +00:00
|
|
|
version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)
|
|
|
|
coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
|
|
|
|
template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
|
2022-05-02 16:08:52 +00:00
|
|
|
cmd, root := clitest.New(t, "create", "my-workspace", "--template", template.Name)
|
2022-03-23 15:03:28 +00:00
|
|
|
clitest.SetupConfig(t, client, root)
|
|
|
|
doneChan := make(chan struct{})
|
|
|
|
pty := ptytest.New(t)
|
|
|
|
cmd.SetIn(pty.Input())
|
|
|
|
cmd.SetOut(pty.Output())
|
|
|
|
go func() {
|
|
|
|
defer close(doneChan)
|
|
|
|
err := cmd.Execute()
|
|
|
|
require.NoError(t, err)
|
|
|
|
}()
|
|
|
|
matches := []string{
|
2022-04-11 23:54:30 +00:00
|
|
|
"Confirm create", "yes",
|
|
|
|
}
|
|
|
|
for i := 0; i < len(matches); i += 2 {
|
|
|
|
match := matches[i]
|
|
|
|
value := matches[i+1]
|
|
|
|
pty.ExpectMatch(match)
|
|
|
|
pty.WriteLine(value)
|
|
|
|
}
|
|
|
|
<-doneChan
|
|
|
|
})
|
2022-05-19 12:49:40 +00:00
|
|
|
|
2022-04-11 23:54:30 +00:00
|
|
|
t.Run("CreateFromList", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2022-05-19 22:47:45 +00:00
|
|
|
client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
|
2022-04-11 23:54:30 +00:00
|
|
|
user := coderdtest.CreateFirstUser(t, client)
|
|
|
|
version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)
|
|
|
|
coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
|
|
|
|
_ = coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
|
2022-05-02 16:08:52 +00:00
|
|
|
cmd, root := clitest.New(t, "create", "my-workspace")
|
2022-04-11 23:54:30 +00:00
|
|
|
clitest.SetupConfig(t, client, root)
|
|
|
|
doneChan := make(chan struct{})
|
|
|
|
pty := ptytest.New(t)
|
|
|
|
cmd.SetIn(pty.Input())
|
|
|
|
cmd.SetOut(pty.Output())
|
|
|
|
go func() {
|
|
|
|
defer close(doneChan)
|
|
|
|
err := cmd.Execute()
|
|
|
|
require.NoError(t, err)
|
|
|
|
}()
|
|
|
|
matches := []string{
|
|
|
|
"Confirm create", "yes",
|
|
|
|
}
|
|
|
|
for i := 0; i < len(matches); i += 2 {
|
|
|
|
match := matches[i]
|
|
|
|
value := matches[i+1]
|
|
|
|
pty.ExpectMatch(match)
|
|
|
|
pty.WriteLine(value)
|
|
|
|
}
|
|
|
|
<-doneChan
|
|
|
|
})
|
2022-05-19 12:49:40 +00:00
|
|
|
|
2022-04-11 23:54:30 +00:00
|
|
|
t.Run("FromNothing", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2022-05-19 22:47:45 +00:00
|
|
|
client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
|
2022-04-11 23:54:30 +00:00
|
|
|
user := coderdtest.CreateFirstUser(t, client)
|
|
|
|
version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)
|
|
|
|
coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
|
|
|
|
_ = coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
|
2022-05-02 16:08:52 +00:00
|
|
|
cmd, root := clitest.New(t, "create", "")
|
2022-04-11 23:54:30 +00:00
|
|
|
clitest.SetupConfig(t, client, root)
|
|
|
|
doneChan := make(chan struct{})
|
|
|
|
pty := ptytest.New(t)
|
|
|
|
cmd.SetIn(pty.Input())
|
|
|
|
cmd.SetOut(pty.Output())
|
|
|
|
go func() {
|
|
|
|
defer close(doneChan)
|
|
|
|
err := cmd.Execute()
|
|
|
|
require.NoError(t, err)
|
|
|
|
}()
|
|
|
|
matches := []string{
|
|
|
|
"Specify a name", "my-workspace",
|
|
|
|
"Confirm create?", "yes",
|
|
|
|
}
|
|
|
|
for i := 0; i < len(matches); i += 2 {
|
|
|
|
match := matches[i]
|
|
|
|
value := matches[i+1]
|
|
|
|
pty.ExpectMatch(match)
|
|
|
|
pty.WriteLine(value)
|
|
|
|
}
|
|
|
|
<-doneChan
|
|
|
|
})
|
2022-05-19 12:49:40 +00:00
|
|
|
|
2022-04-11 23:54:30 +00:00
|
|
|
t.Run("WithParameter", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
2022-05-19 22:47:45 +00:00
|
|
|
client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
|
2022-04-11 23:54:30 +00:00
|
|
|
user := coderdtest.CreateFirstUser(t, client)
|
2022-05-19 12:49:40 +00:00
|
|
|
|
|
|
|
defaultValue := "something"
|
2022-04-11 23:54:30 +00:00
|
|
|
version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
|
|
|
|
Parse: []*proto.Parse_Response{{
|
|
|
|
Type: &proto.Parse_Response_Complete{
|
|
|
|
Complete: &proto.Parse_Complete{
|
2022-05-19 12:49:40 +00:00
|
|
|
ParameterSchemas: []*proto.ParameterSchema{
|
|
|
|
{
|
|
|
|
AllowOverrideSource: true,
|
|
|
|
Name: "region",
|
|
|
|
Description: "description 1",
|
|
|
|
DefaultSource: &proto.ParameterSource{
|
|
|
|
Scheme: proto.ParameterSource_DATA,
|
|
|
|
Value: defaultValue,
|
|
|
|
},
|
|
|
|
DefaultDestination: &proto.ParameterDestination{
|
|
|
|
Scheme: proto.ParameterDestination_PROVISIONER_VARIABLE,
|
|
|
|
},
|
2022-04-11 23:54:30 +00:00
|
|
|
},
|
2022-05-19 12:49:40 +00:00
|
|
|
{
|
|
|
|
AllowOverrideSource: true,
|
|
|
|
Name: "username",
|
|
|
|
Description: "description 2",
|
|
|
|
DefaultSource: &proto.ParameterSource{
|
|
|
|
Scheme: proto.ParameterSource_DATA,
|
|
|
|
// No default value
|
|
|
|
Value: "",
|
|
|
|
},
|
|
|
|
DefaultDestination: &proto.ParameterDestination{
|
|
|
|
Scheme: proto.ParameterDestination_PROVISIONER_VARIABLE,
|
|
|
|
},
|
2022-04-11 23:54:30 +00:00
|
|
|
},
|
2022-05-19 12:49:40 +00:00
|
|
|
},
|
2022-04-11 23:54:30 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}},
|
|
|
|
Provision: echo.ProvisionComplete,
|
|
|
|
ProvisionDryRun: echo.ProvisionComplete,
|
|
|
|
})
|
2022-05-19 12:49:40 +00:00
|
|
|
|
2022-04-11 23:54:30 +00:00
|
|
|
coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
|
|
|
|
_ = coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
|
2022-05-02 16:08:52 +00:00
|
|
|
cmd, root := clitest.New(t, "create", "")
|
2022-04-11 23:54:30 +00:00
|
|
|
clitest.SetupConfig(t, client, root)
|
|
|
|
doneChan := make(chan struct{})
|
|
|
|
pty := ptytest.New(t)
|
|
|
|
cmd.SetIn(pty.Input())
|
|
|
|
cmd.SetOut(pty.Output())
|
|
|
|
go func() {
|
|
|
|
defer close(doneChan)
|
|
|
|
err := cmd.Execute()
|
|
|
|
require.NoError(t, err)
|
|
|
|
}()
|
2022-05-19 12:49:40 +00:00
|
|
|
|
2022-04-11 23:54:30 +00:00
|
|
|
matches := []string{
|
|
|
|
"Specify a name", "my-workspace",
|
2022-05-19 12:49:40 +00:00
|
|
|
fmt.Sprintf("Enter a value (default: %q):", defaultValue), "bingo",
|
|
|
|
"Enter a value:", "boingo",
|
2022-04-11 23:54:30 +00:00
|
|
|
"Confirm create?", "yes",
|
2022-03-23 15:03:28 +00:00
|
|
|
}
|
|
|
|
for i := 0; i < len(matches); i += 2 {
|
|
|
|
match := matches[i]
|
|
|
|
value := matches[i+1]
|
|
|
|
pty.ExpectMatch(match)
|
|
|
|
pty.WriteLine(value)
|
|
|
|
}
|
|
|
|
<-doneChan
|
|
|
|
})
|
|
|
|
}
|