@@ -15,6 +15,7 @@ import (
1515 "github.com/cloudquery/cloudquery/cli/v6/internal/hub"
1616 "github.com/google/go-cmp/cmp"
1717 "github.com/google/uuid"
18+ "github.com/stretchr/testify/require"
1819)
1920
2021func TestPluginPublish (t * testing.T ) {
@@ -62,32 +63,39 @@ func TestPluginPublish(t *testing.T) {
6263 case "/plugins/cloudquery/source/test/versions/v1.2.3" :
6364 checkAuthHeader (t , r )
6465 w .WriteHeader (http .StatusCreated )
65- w .Write ([]byte (`{"name": "v1.2.3"}` ))
66+ _ , err := w .Write ([]byte (`{"name": "v1.2.3"}` ))
67+ require .NoError (t , err )
6668 checkCreatePluginVersionRequest (t , r )
6769 case "/plugins/cloudquery/source/test/versions/v1.2.3/tables" :
6870 checkAuthHeader (t , r )
6971 w .WriteHeader (http .StatusCreated )
70- w .Write ([]byte (`{}` ))
72+ _ , err := w .Write ([]byte (`{}` ))
73+ require .NoError (t , err )
7174 checkCreateTablesRequest (t , r )
7275 case "/plugins/cloudquery/source/test/versions/v1.2.3/docs" :
7376 checkAuthHeader (t , r )
7477 w .WriteHeader (http .StatusCreated )
75- w .Write ([]byte (`{}` ))
78+ _ , err := w .Write ([]byte (`{}` ))
79+ require .NoError (t , err )
7680 checkCreateDocsRequest (t , r , tc .distDir )
7781 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/linux_amd64" :
7882 checkAuthHeader (t , r )
7983 w .WriteHeader (http .StatusCreated )
80- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
84+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
85+ require .NoError (t , err )
8186 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/darwin_amd64" :
8287 checkAuthHeader (t , r )
8388 w .WriteHeader (http .StatusCreated )
84- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
89+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
90+ require .NoError (t , err )
8591 case "/upload-linux" :
8692 w .WriteHeader (http .StatusOK )
87- w .Write ([]byte (`{}` ))
93+ _ , err := w .Write ([]byte (`{}` ))
94+ require .NoError (t , err )
8895 case "/upload-darwin" :
8996 w .WriteHeader (http .StatusOK )
90- w .Write ([]byte (`{}` ))
97+ _ , err := w .Write ([]byte (`{}` ))
98+ require .NoError (t , err )
9199 }
92100 }))
93101 defer ts .Close ()
@@ -139,32 +147,39 @@ func TestPluginPublishFinalize(t *testing.T) {
139147 checkCreatePluginVersionRequest (t , r )
140148 w .WriteHeader (http .StatusCreated )
141149 }
142- w .Write ([]byte (`{"name": "v1.2.3"}` ))
150+ _ , err := w .Write ([]byte (`{"name": "v1.2.3"}` ))
151+ require .NoError (t , err )
143152 case "/plugins/cloudquery/source/test/versions/v1.2.3/tables" :
144153 checkAuthHeader (t , r )
145154 w .WriteHeader (http .StatusCreated )
146- w .Write ([]byte (`{}` ))
155+ _ , err := w .Write ([]byte (`{}` ))
156+ require .NoError (t , err )
147157 checkCreateTablesRequest (t , r )
148158 case "/plugins/cloudquery/source/test/versions/v1.2.3/docs" :
149159 checkAuthHeader (t , r )
150160 w .WriteHeader (http .StatusCreated )
151- w .Write ([]byte (`{}` ))
161+ _ , err := w .Write ([]byte (`{}` ))
162+ require .NoError (t , err )
152163 checkCreateDocsRequest (t , r , "testdata/dist-v1-with-team-package-json" )
153164 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/linux_amd64" :
154165 checkAuthHeader (t , r )
155166 w .WriteHeader (http .StatusCreated )
156- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
167+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
168+ require .NoError (t , err )
157169 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/darwin_amd64" :
158170 checkAuthHeader (t , r )
159171 w .WriteHeader (http .StatusCreated )
160- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
172+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
173+ require .NoError (t , err )
161174 case "/upload-linux" :
162175 w .WriteHeader (http .StatusOK )
163- w .Write ([]byte (`{}` ))
176+ _ , err := w .Write ([]byte (`{}` ))
177+ require .NoError (t , err )
164178 gotUploads ++
165179 case "/upload-darwin" :
166180 w .WriteHeader (http .StatusOK )
167- w .Write ([]byte (`{}` ))
181+ _ , err := w .Write ([]byte (`{}` ))
182+ require .NoError (t , err )
168183 gotUploads ++
169184 }
170185 }))
@@ -213,35 +228,43 @@ func TestPluginPublishWithUI(t *testing.T) {
213228 case "/plugins/cloudquery/source/test/versions/v1.2.3" :
214229 checkAuthHeader (t , r )
215230 w .WriteHeader (http .StatusCreated )
216- w .Write ([]byte (`{"name": "v1.2.3"}` ))
231+ _ , err := w .Write ([]byte (`{"name": "v1.2.3"}` ))
232+ require .NoError (t , err )
217233 checkCreatePluginVersionRequest (t , r )
218234 case "/plugins/cloudquery/source/test/versions/v1.2.3/tables" :
219235 checkAuthHeader (t , r )
220236 w .WriteHeader (http .StatusCreated )
221- w .Write ([]byte (`{}` ))
237+ _ , err := w .Write ([]byte (`{}` ))
238+ require .NoError (t , err )
222239 checkCreateTablesRequest (t , r )
223240 case "/plugins/cloudquery/source/test/versions/v1.2.3/docs" :
224241 checkAuthHeader (t , r )
225242 w .WriteHeader (http .StatusCreated )
226- w .Write ([]byte (`{}` ))
243+ _ , err := w .Write ([]byte (`{}` ))
244+ require .NoError (t , err )
227245 checkCreateDocsRequest (t , r , distDir )
228246 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/linux_amd64" :
229247 checkAuthHeader (t , r )
230248 w .WriteHeader (http .StatusCreated )
231- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
249+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-linux" )))
250+ require .NoError (t , err )
232251 case "/plugins/cloudquery/source/test/versions/v1.2.3/assets/darwin_amd64" :
233252 checkAuthHeader (t , r )
234253 w .WriteHeader (http .StatusCreated )
235- w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
254+ _ , err := w .Write ([]byte (fmt .Sprintf (`{"url": "%s"}` , "http://" + r .Host + "/upload-darwin" )))
255+ require .NoError (t , err )
236256 case "/upload-linux" :
237257 w .WriteHeader (http .StatusOK )
238- w .Write ([]byte (`{}` ))
258+ _ , err := w .Write ([]byte (`{}` ))
259+ require .NoError (t , err )
239260 case "/upload-darwin" :
240261 w .WriteHeader (http .StatusOK )
241- w .Write ([]byte (`{}` ))
262+ _ , err := w .Write ([]byte (`{}` ))
263+ require .NoError (t , err )
242264 case "/upload-uiasset" :
243265 w .WriteHeader (http .StatusOK )
244- w .Write ([]byte (`{}` ))
266+ _ , err := w .Write ([]byte (`{}` ))
267+ require .NoError (t , err )
245268 case "/plugins/cloudquery/source/test/versions/v1.2.3/uiassets" :
246269 checkAuthHeader (t , r )
247270 switch r .Method {
@@ -307,7 +330,8 @@ func TestPluginPublish_Unauthorized(t *testing.T) {
307330 ts := httptest .NewServer (http .HandlerFunc (func (w http.ResponseWriter , r * http.Request ) {
308331 w .Header ().Set ("Content-Type" , "application/json" )
309332 w .WriteHeader (http .StatusUnauthorized )
310- w .Write ([]byte (`{"message": "unauthorized"}` ))
333+ _ , err := w .Write ([]byte (`{"message": "unauthorized"}` ))
334+ require .NoError (t , err )
311335 }))
312336 defer ts .Close ()
313337
0 commit comments