mirror of https://github.com/databricks/cli.git
67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
|
package python
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/databricks/databricks-sdk-go/service/jobs"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
type testCase struct {
|
||
|
|
Actual []string
|
||
|
|
Expected string
|
||
|
|
}
|
||
|
|
type NamedParams map[string]string
|
||
|
|
type testCaseNamed struct {
|
||
|
|
Actual NamedParams
|
||
|
|
Expected string
|
||
|
|
}
|
||
|
|
|
||
|
|
var paramsTestCases []testCase = []testCase{
|
||
|
|
{[]string{}, `"python"`},
|
||
|
|
{[]string{"a"}, `"python", "a"`},
|
||
|
|
{[]string{"a", "b"}, `"python", "a", "b"`},
|
||
|
|
{[]string{"123!@#$%^&*()-="}, `"python", "123!@#$%^&*()-="`},
|
||
|
|
{[]string{`{"a": 1}`}, `"python", "{\"a\": 1}"`},
|
||
|
|
}
|
||
|
|
|
||
|
|
var paramsTestCasesNamed []testCaseNamed = []testCaseNamed{
|
||
|
|
{NamedParams{}, `"python"`},
|
||
|
|
{NamedParams{"a": "1"}, `"python", "a=1"`},
|
||
|
|
{NamedParams{"a": "'1'"}, `"python", "a='1'"`},
|
||
|
|
{NamedParams{"a": `"1"`}, `"python", "a=\"1\""`},
|
||
|
|
{NamedParams{"a": "1", "b": "2"}, `"python", "a=1", "b=2"`},
|
||
|
|
{NamedParams{"data": `{"a": 1}`}, `"python", "data={\"a\": 1}"`},
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenerateParameters(t *testing.T) {
|
||
|
|
trampoline := pythonTrampoline{}
|
||
|
|
for _, c := range paramsTestCases {
|
||
|
|
task := &jobs.PythonWheelTask{Parameters: c.Actual}
|
||
|
|
result, err := trampoline.generateParameters(task)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, c.Expected, result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenerateNamedParameters(t *testing.T) {
|
||
|
|
trampoline := pythonTrampoline{}
|
||
|
|
for _, c := range paramsTestCasesNamed {
|
||
|
|
task := &jobs.PythonWheelTask{NamedParameters: c.Actual}
|
||
|
|
result, err := trampoline.generateParameters(task)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// parameters order can be undetermenistic, so just check that they exist as expected
|
||
|
|
require.ElementsMatch(t, strings.Split(c.Expected, ","), strings.Split(result, ","))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenerateBoth(t *testing.T) {
|
||
|
|
trampoline := pythonTrampoline{}
|
||
|
|
task := &jobs.PythonWheelTask{NamedParameters: map[string]string{"a": "1"}, Parameters: []string{"b"}}
|
||
|
|
_, err := trampoline.generateParameters(task)
|
||
|
|
require.Error(t, err)
|
||
|
|
require.ErrorContains(t, err, "not allowed to pass both paramaters and named_parameters")
|
||
|
|
}
|