databricks-cli/internal/init_test.go

package internal

import (
	"fmt"
	"os"
	"path/filepath"
	"testing"

	_ "github.com/databricks/cli/cmd/init"
	"github.com/databricks/cli/cmd/root"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func assertFileContains(t *testing.T, path string, substr string) {
	b, err := os.ReadFile(path)
	require.NoError(t, err)
	assert.Contains(t, string(b), substr)
}

func TestTemplateInitializationForDevConfig(t *testing.T) {
	// create target directory with the input config
	tmp := t.TempDir()
	f, err := os.Create(filepath.Join(tmp, "config.json"))
	require.NoError(t, err)
	defer f.Close()
	_, err = f.WriteString(`
	{
		"project_name": "development_project",
		"cloud_type": "AWS",
		"ci_type": "github",
		"is_production": false
	}
	`)
	require.NoError(t, err)

	// materialize the template
	cmd := root.RootCmd
	cmd.SetArgs([]string{"init", filepath.FromSlash("testdata/init/templateDefinition"), "--target-dir", tmp})
	err = cmd.Execute()
	require.NoError(t, err)

	// assert on materialized template
	assert.FileExists(t, filepath.Join(tmp, "development_project", "aws_file"))
	assert.FileExists(t, filepath.Join(tmp, "development_project", ".github"))
	assert.NoFileExists(t, filepath.Join(tmp, "development_project", "azure_file"))
	assertFileContains(t, filepath.Join(tmp, "development_project", "aws_file"), "This file should only be generated for AWS")
	assertFileContains(t, filepath.Join(tmp, "development_project", ".github"), "This is a development project")
}

func TestTemplateInitializationForProdConfig(t *testing.T) {
	// create target directory with the input config
	tmp := t.TempDir()
	f, err := os.Create(filepath.Join(tmp, "config.json"))
	require.NoError(t, err)
	defer f.Close()
	_, err = f.WriteString(`
	{
		"project_name": "production_project",
		"cloud_type": "Azure",
		"ci_type": "azure_devops",
		"is_production": true
	}
	`)
	require.NoError(t, err)

	// materialize the template
	cmd := root.RootCmd
	childCommands := cmd.Commands()
	fmt.Println(childCommands)
	cmd.SetArgs([]string{"init", filepath.FromSlash("testdata/init/templateDefinition"), "--target-dir", tmp})
	err = cmd.Execute()
	require.NoError(t, err)

	// assert on materialized template
	assert.FileExists(t, filepath.Join(tmp, "production_project", "azure_file"))
	assert.FileExists(t, filepath.Join(tmp, "production_project", ".azure_devops"))
	assert.NoFileExists(t, filepath.Join(tmp, "production_project", "aws_file"))
	assertFileContains(t, filepath.Join(tmp, "production_project", "azure_file"), "This file should only be generated for Azure")
	assertFileContains(t, filepath.Join(tmp, "production_project", ".azure_devops"), "This is a production project")
}