Revert "Add `--key` flag for generate commands to specify resource key (#1165)"

This reverts commit b28432afed.
This commit is contained in:
Shreyas Goenka 2024-02-01 11:23:29 +01:00
parent 06de08859d
commit eaf8066068
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
6 changed files with 8 additions and 23 deletions

View File

@ -6,8 +6,6 @@ import (
)
func newGenerateCommand() *cobra.Command {
var key string
cmd := &cobra.Command{
Use: "generate",
Short: "Generate bundle configuration",
@ -17,6 +15,5 @@ func newGenerateCommand() *cobra.Command {
cmd.AddCommand(generate.NewGenerateJobCommand())
cmd.AddCommand(generate.NewGeneratePipelineCommand())
cmd.PersistentFlags().StringVar(&key, "key", "", `resource key to use for the generated configuration`)
return cmd
}

View File

@ -72,18 +72,14 @@ func TestGeneratePipelineCommand(t *testing.T) {
srcDir := filepath.Join(root, "src")
cmd.Flag("source-dir").Value.Set(srcDir)
var key string
cmd.Flags().StringVar(&key, "key", "test_pipeline", "")
err := cmd.RunE(cmd, []string{})
require.NoError(t, err)
data, err := os.ReadFile(filepath.Join(configDir, "test_pipeline.yml"))
data, err := os.ReadFile(filepath.Join(configDir, "pipeline_test_pipeline.yml"))
require.NoError(t, err)
require.Equal(t, fmt.Sprintf(`resources:
pipelines:
test_pipeline:
pipeline_test_pipeline:
name: test-pipeline
libraries:
- notebook:

View File

@ -63,11 +63,7 @@ func NewGenerateJobCommand() *cobra.Command {
return err
}
jobKey := cmd.Flag("key").Value.String()
if jobKey == "" {
jobKey = textutil.NormalizeString(job.Settings.Name)
}
jobKey := fmt.Sprintf("job_%s", textutil.NormalizeString(job.Settings.Name))
result := map[string]dyn.Value{
"resources": dyn.V(map[string]dyn.Value{
"jobs": dyn.V(map[string]dyn.Value{

View File

@ -63,15 +63,11 @@ func NewGeneratePipelineCommand() *cobra.Command {
return err
}
pipelineKey := cmd.Flag("key").Value.String()
if pipelineKey == "" {
pipelineKey = textutil.NormalizeString(pipeline.Name)
}
jobKey := fmt.Sprintf("pipeline_%s", textutil.NormalizeString(pipeline.Name))
result := map[string]dyn.Value{
"resources": dyn.V(map[string]dyn.Value{
"pipelines": dyn.V(map[string]dyn.Value{
pipelineKey: v,
jobKey: v,
}),
}),
}
@ -81,7 +77,7 @@ func NewGeneratePipelineCommand() *cobra.Command {
return err
}
filename := filepath.Join(configDir, fmt.Sprintf("%s.yml", pipelineKey))
filename := filepath.Join(configDir, fmt.Sprintf("%s.yml", jobKey))
err = yamlsaver.SaveAsYAML(result, filename, force)
if err != nil {
return err

View File

@ -46,7 +46,7 @@ func TestAccGenerateFromExistingJobAndDeploy(t *testing.T) {
_, err = os.Stat(filepath.Join(bundleRoot, "src", "test.py"))
require.NoError(t, err)
matches, err := filepath.Glob(filepath.Join(bundleRoot, "resources", "generated_job_*.yml"))
matches, err := filepath.Glob(filepath.Join(bundleRoot, "resources", "job_generated_job_*.yml"))
require.NoError(t, err)
require.Len(t, matches, 1)

View File

@ -47,7 +47,7 @@ func TestAccGenerateFromExistingPipelineAndDeploy(t *testing.T) {
_, err = os.Stat(filepath.Join(bundleRoot, "src", "test.py"))
require.NoError(t, err)
matches, err := filepath.Glob(filepath.Join(bundleRoot, "resources", "generated_pipeline_*.yml"))
matches, err := filepath.Glob(filepath.Join(bundleRoot, "resources", "pipeline_generated_pipeline_*.yml"))
require.NoError(t, err)
require.Len(t, matches, 1)