mirror of https://github.com/coder/coder.git
206 lines
6.3 KiB
Go
206 lines
6.3 KiB
Go
|
package coderd_test
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"net/http"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/google/uuid"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
|
||
|
"github.com/coder/coder/coderd/coderdtest"
|
||
|
"github.com/coder/coder/codersdk"
|
||
|
"github.com/coder/coder/provisioner/echo"
|
||
|
"github.com/coder/coder/provisionersdk/proto"
|
||
|
)
|
||
|
|
||
|
func TestProjectVersion(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
t.Run("Get", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, nil)
|
||
|
_, err := client.ProjectVersion(context.Background(), version.ID)
|
||
|
require.NoError(t, err)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestProjectVersionSchema(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
t.Run("ListRunning", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, nil)
|
||
|
_, err := client.ProjectVersionSchema(context.Background(), version.ID)
|
||
|
var apiErr *codersdk.Error
|
||
|
require.ErrorAs(t, err, &apiErr)
|
||
|
require.Equal(t, http.StatusPreconditionFailed, apiErr.StatusCode())
|
||
|
})
|
||
|
t.Run("List", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
coderdtest.NewProvisionerDaemon(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, &echo.Responses{
|
||
|
Parse: []*proto.Parse_Response{{
|
||
|
Type: &proto.Parse_Response_Complete{
|
||
|
Complete: &proto.Parse_Complete{
|
||
|
ParameterSchemas: []*proto.ParameterSchema{{
|
||
|
Name: "example",
|
||
|
DefaultDestination: &proto.ParameterDestination{
|
||
|
Scheme: proto.ParameterDestination_PROVISIONER_VARIABLE,
|
||
|
},
|
||
|
}},
|
||
|
},
|
||
|
},
|
||
|
}},
|
||
|
Provision: echo.ProvisionComplete,
|
||
|
})
|
||
|
coderdtest.AwaitProjectVersionJob(t, client, version.ID)
|
||
|
schemas, err := client.ProjectVersionSchema(context.Background(), version.ID)
|
||
|
require.NoError(t, err)
|
||
|
require.NotNil(t, schemas)
|
||
|
require.Len(t, schemas, 1)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestProjectVersionParameters(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
t.Run("ListRunning", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, nil)
|
||
|
_, err := client.ProjectVersionParameters(context.Background(), version.ID)
|
||
|
var apiErr *codersdk.Error
|
||
|
require.ErrorAs(t, err, &apiErr)
|
||
|
require.Equal(t, http.StatusPreconditionFailed, apiErr.StatusCode())
|
||
|
})
|
||
|
t.Run("List", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
coderdtest.NewProvisionerDaemon(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, &echo.Responses{
|
||
|
Parse: []*proto.Parse_Response{{
|
||
|
Type: &proto.Parse_Response_Complete{
|
||
|
Complete: &proto.Parse_Complete{
|
||
|
ParameterSchemas: []*proto.ParameterSchema{{
|
||
|
Name: "example",
|
||
|
RedisplayValue: true,
|
||
|
DefaultSource: &proto.ParameterSource{
|
||
|
Scheme: proto.ParameterSource_DATA,
|
||
|
Value: "hello",
|
||
|
},
|
||
|
DefaultDestination: &proto.ParameterDestination{
|
||
|
Scheme: proto.ParameterDestination_PROVISIONER_VARIABLE,
|
||
|
},
|
||
|
}},
|
||
|
},
|
||
|
},
|
||
|
}},
|
||
|
Provision: echo.ProvisionComplete,
|
||
|
})
|
||
|
coderdtest.AwaitProjectVersionJob(t, client, version.ID)
|
||
|
params, err := client.ProjectVersionParameters(context.Background(), version.ID)
|
||
|
require.NoError(t, err)
|
||
|
require.NotNil(t, params)
|
||
|
require.Len(t, params, 1)
|
||
|
require.Equal(t, "hello", params[0].SourceValue)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestProjectVersionResources(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
t.Run("ListRunning", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, nil)
|
||
|
_, err := client.ProjectVersionResources(context.Background(), version.ID)
|
||
|
var apiErr *codersdk.Error
|
||
|
require.ErrorAs(t, err, &apiErr)
|
||
|
require.Equal(t, http.StatusPreconditionFailed, apiErr.StatusCode())
|
||
|
})
|
||
|
t.Run("List", func(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
coderdtest.NewProvisionerDaemon(t, client)
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, &echo.Responses{
|
||
|
Parse: echo.ParseComplete,
|
||
|
Provision: []*proto.Provision_Response{{
|
||
|
Type: &proto.Provision_Response_Complete{
|
||
|
Complete: &proto.Provision_Complete{
|
||
|
Resources: []*proto.Resource{{
|
||
|
Name: "some",
|
||
|
Type: "example",
|
||
|
Agent: &proto.Agent{
|
||
|
Id: "something",
|
||
|
Auth: &proto.Agent_Token{},
|
||
|
},
|
||
|
}, {
|
||
|
Name: "another",
|
||
|
Type: "example",
|
||
|
}},
|
||
|
},
|
||
|
},
|
||
|
}},
|
||
|
})
|
||
|
coderdtest.AwaitProjectVersionJob(t, client, version.ID)
|
||
|
resources, err := client.ProjectVersionResources(context.Background(), version.ID)
|
||
|
require.NoError(t, err)
|
||
|
require.NotNil(t, resources)
|
||
|
require.Len(t, resources, 4)
|
||
|
require.Equal(t, "some", resources[0].Name)
|
||
|
require.Equal(t, "example", resources[0].Type)
|
||
|
require.NotNil(t, resources[0].Agent)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestProjectVersionLogs(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
client := coderdtest.New(t, nil)
|
||
|
user := coderdtest.CreateFirstUser(t, client)
|
||
|
coderdtest.NewProvisionerDaemon(t, client)
|
||
|
before := time.Now()
|
||
|
version := coderdtest.CreateProjectVersion(t, client, user.OrganizationID, &echo.Responses{
|
||
|
Parse: echo.ParseComplete,
|
||
|
Provision: []*proto.Provision_Response{{
|
||
|
Type: &proto.Provision_Response_Log{
|
||
|
Log: &proto.Log{
|
||
|
Level: proto.LogLevel_INFO,
|
||
|
Output: "example",
|
||
|
},
|
||
|
},
|
||
|
}, {
|
||
|
Type: &proto.Provision_Response_Complete{
|
||
|
Complete: &proto.Provision_Complete{
|
||
|
Resources: []*proto.Resource{{
|
||
|
Name: "some",
|
||
|
Type: "example",
|
||
|
Agent: &proto.Agent{
|
||
|
Id: "something",
|
||
|
Auth: &proto.Agent_Token{
|
||
|
Token: uuid.NewString(),
|
||
|
},
|
||
|
},
|
||
|
}, {
|
||
|
Name: "another",
|
||
|
Type: "example",
|
||
|
}},
|
||
|
},
|
||
|
},
|
||
|
}},
|
||
|
})
|
||
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
||
|
t.Cleanup(cancelFunc)
|
||
|
logs, err := client.ProjectVersionLogsAfter(ctx, version.ID, before)
|
||
|
require.NoError(t, err)
|
||
|
log := <-logs
|
||
|
require.Equal(t, "example", log.Output)
|
||
|
}
|