@@ -2,6 +2,8 @@ package coderd_test
2
2
3
3
import (
4
4
"context"
5
+ "github.com/coder/coder/provisioner/echo"
6
+ "github.com/coder/coder/provisionersdk/proto"
5
7
"net/http"
6
8
"testing"
7
9
@@ -32,7 +34,8 @@ func TestPostParameter(t *testing.T) {
32
34
t .Parallel ()
33
35
client := coderdtest .New (t , nil )
34
36
user := coderdtest .CreateFirstUser (t , client )
35
- _ , err := client .CreateParameter (context .Background (), codersdk .ParameterOrganization , user .OrganizationID , codersdk.CreateParameterRequest {
37
+ template := createTemplate (t , client , user )
38
+ _ , err := client .CreateParameter (context .Background (), codersdk .ParameterTemplate , template .ID , codersdk.CreateParameterRequest {
36
39
Name : "example" ,
37
40
SourceValue : "tomato" ,
38
41
SourceScheme : codersdk .ParameterSourceSchemeData ,
@@ -45,15 +48,16 @@ func TestPostParameter(t *testing.T) {
45
48
t .Parallel ()
46
49
client := coderdtest .New (t , nil )
47
50
user := coderdtest .CreateFirstUser (t , client )
48
- _ , err := client .CreateParameter (context .Background (), codersdk .ParameterOrganization , user .OrganizationID , codersdk.CreateParameterRequest {
51
+ template := createTemplate (t , client , user )
52
+ _ , err := client .CreateParameter (context .Background (), codersdk .ParameterTemplate , template .ID , codersdk.CreateParameterRequest {
49
53
Name : "example" ,
50
54
SourceValue : "tomato" ,
51
55
SourceScheme : codersdk .ParameterSourceSchemeData ,
52
56
DestinationScheme : codersdk .ParameterDestinationSchemeProvisionerVariable ,
53
57
})
54
58
require .NoError (t , err )
55
59
56
- _ , err = client .CreateParameter (context .Background (), codersdk .ParameterOrganization , user . OrganizationID , codersdk.CreateParameterRequest {
60
+ _ , err = client .CreateParameter (context .Background (), codersdk .ParameterTemplate , template . ID , codersdk.CreateParameterRequest {
57
61
Name : "example" ,
58
62
SourceValue : "tomato" ,
59
63
SourceScheme : codersdk .ParameterSourceSchemeData ,
@@ -71,21 +75,23 @@ func TestParameters(t *testing.T) {
71
75
t .Parallel ()
72
76
client := coderdtest .New (t , nil )
73
77
user := coderdtest .CreateFirstUser (t , client )
74
- _ , err := client .Parameters (context .Background (), codersdk .ParameterOrganization , user .OrganizationID )
78
+ template := createTemplate (t , client , user )
79
+ _ , err := client .Parameters (context .Background (), codersdk .ParameterTemplate , template .ID )
75
80
require .NoError (t , err )
76
81
})
77
82
t .Run ("List" , func (t * testing.T ) {
78
83
t .Parallel ()
79
84
client := coderdtest .New (t , nil )
80
85
user := coderdtest .CreateFirstUser (t , client )
81
- _ , err := client .CreateParameter (context .Background (), codersdk .ParameterOrganization , user .OrganizationID , codersdk.CreateParameterRequest {
86
+ template := createTemplate (t , client , user )
87
+ _ , err := client .CreateParameter (context .Background (), codersdk .ParameterTemplate , template .ID , codersdk.CreateParameterRequest {
82
88
Name : "example" ,
83
89
SourceValue : "tomato" ,
84
90
SourceScheme : codersdk .ParameterSourceSchemeData ,
85
91
DestinationScheme : codersdk .ParameterDestinationSchemeProvisionerVariable ,
86
92
})
87
93
require .NoError (t , err )
88
- params , err := client .Parameters (context .Background (), codersdk .ParameterOrganization , user . OrganizationID )
94
+ params , err := client .Parameters (context .Background (), codersdk .ParameterTemplate , template . ID )
89
95
require .NoError (t , err )
90
96
require .Len (t , params , 1 )
91
97
})
@@ -97,7 +103,8 @@ func TestDeleteParameter(t *testing.T) {
97
103
t .Parallel ()
98
104
client := coderdtest .New (t , nil )
99
105
user := coderdtest .CreateFirstUser (t , client )
100
- err := client .DeleteParameter (context .Background (), codersdk .ParameterOrganization , user .OrganizationID , "something" )
106
+ template := createTemplate (t , client , user )
107
+ err := client .DeleteParameter (context .Background (), codersdk .ParameterTemplate , template .ID , "something" )
101
108
var apiErr * codersdk.Error
102
109
require .ErrorAs (t , err , & apiErr )
103
110
require .Equal (t , http .StatusNotFound , apiErr .StatusCode ())
@@ -106,14 +113,39 @@ func TestDeleteParameter(t *testing.T) {
106
113
t .Parallel ()
107
114
client := coderdtest .New (t , nil )
108
115
user := coderdtest .CreateFirstUser (t , client )
109
- param , err := client .CreateParameter (context .Background (), codersdk .ParameterOrganization , user .OrganizationID , codersdk.CreateParameterRequest {
116
+ template := createTemplate (t , client , user )
117
+ param , err := client .CreateParameter (context .Background (), codersdk .ParameterTemplate , template .ID , codersdk.CreateParameterRequest {
110
118
Name : "example" ,
111
119
SourceValue : "tomato" ,
112
120
SourceScheme : codersdk .ParameterSourceSchemeData ,
113
121
DestinationScheme : codersdk .ParameterDestinationSchemeProvisionerVariable ,
114
122
})
115
123
require .NoError (t , err )
116
- err = client .DeleteParameter (context .Background (), codersdk .ParameterOrganization , user . OrganizationID , param .Name )
124
+ err = client .DeleteParameter (context .Background (), codersdk .ParameterTemplate , template . ID , param .Name )
117
125
require .NoError (t , err )
118
126
})
119
127
}
128
+
129
+ func createTemplate (t * testing.T , client * codersdk.Client , user codersdk.CreateFirstUserResponse ) codersdk.Template {
130
+ instanceID := "instanceidentifier"
131
+ version := coderdtest .CreateTemplateVersion (t , client , user .OrganizationID , & echo.Responses {
132
+ Parse : echo .ParseComplete ,
133
+ Provision : []* proto.Provision_Response {{
134
+ Type : & proto.Provision_Response_Complete {
135
+ Complete : & proto.Provision_Complete {
136
+ Resources : []* proto.Resource {{
137
+ Name : "somename" ,
138
+ Type : "someinstance" ,
139
+ Agents : []* proto.Agent {{
140
+ Auth : & proto.Agent_InstanceId {
141
+ InstanceId : instanceID ,
142
+ },
143
+ }},
144
+ }},
145
+ },
146
+ },
147
+ }},
148
+ })
149
+ template := coderdtest .CreateTemplate (t , client , user .OrganizationID , version .ID )
150
+ return template
151
+ }
0 commit comments