steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add RequireSuccessfulRun helper (#125)

This commit is contained in:
Pieter Noordhuis 2022-12-09 15:41:04 +01:00 committed by GitHub
parent 94f884f0a7
commit cb16ad1184
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/api_test.go
View File

@ -15,13 +15,11 @@ import (
func TestAccApiGet(t *testing.T) {
t.Log(GetEnvOrSkipTest(t, "CLOUD_ENV"))
c := NewCobraTestRunner(t, "api", "get", "/api/2.0/preview/scim/v2/Me")
stdout, _, err := c.Run()
require.NoError(t, err)
stdout, _ := RequireSuccessfulRun(t, "api", "get", "/api/2.0/preview/scim/v2/Me")
// Deserialize SCIM API response.
var out map[string]any
err = json.Unmarshal(stdout.Bytes(), &out)
err := json.Unmarshal(stdout.Bytes(), &out)
require.NoError(t, err)
// Assert that the output somewhat makes sense for the SCIM API.
@ -43,15 +41,11 @@ func TestAccApiPost(t *testing.T) {
// Post to mkdir
{
c := NewCobraTestRunner(t, "api", "post", "--body=@"+requestPath, "/api/2.0/dbfs/mkdirs")
_, _, err := c.Run()
require.NoError(t, err)
RequireSuccessfulRun(t, "api", "post", "--body=@"+requestPath, "/api/2.0/dbfs/mkdirs")
}
// Post to delete
{
c := NewCobraTestRunner(t, "api", "post", "--body=@"+requestPath, "/api/2.0/dbfs/delete")
_, _, err := c.Run()
require.NoError(t, err)
RequireSuccessfulRun(t, "api", "post", "--body=@"+requestPath, "/api/2.0/dbfs/delete")
}
}

7
internal/helpers.go
View File

@ -151,6 +151,13 @@ func NewCobraTestRunner(t *testing.T, args ...string) *cobraTestRunner {
}
}
func RequireSuccessfulRun(t *testing.T, args ...string) (bytes.Buffer, bytes.Buffer) {
c := NewCobraTestRunner(t, args...)
stdout, stderr, err := c.Run()
require.NoError(t, err)
return stdout, stderr
}
func writeFile(t *testing.T, name string, body string) string {
f, err := os.Create(filepath.Join(t.TempDir(), name))
require.NoError(t, err)