Function TestManagedDeployWithPrimaryBranch

cli/cmd/deploy/deploy_test.go:96–136

(t *testing.T)

}

func TestManagedDeployWithPrimaryBranch(t *testing.T) {
	testmode.Expensive(t)
	adm := testadmin.New(t)

	_, c := adm.NewUser(t)
	u1 := testcli.New(t, adm, c.Token)

	result := u1.Run(t, "org", "create", "github-branch-test")
	require.Equal(t, 0, result.ExitCode)

	// set up a git repo with two branches having different content:
	// - main: models/model.sql contains SELECT 'main' AS env
	// - staging: models/model.sql contains SELECT 'staging' AS env
	// The repo is left on the staging branch for the deploy.
	tempDir := initRillProject(t)
	initGitWithTwoBranches(t, tempDir)

	result = u1.Run(t, "project", "deploy", "--interactive=false", "--org=github-branch-test", "--project=rill-mgd-branch", "--skip-deploy=true", "--primary-branch=staging", "--path="+tempDir)
	require.Equal(t, 0, result.ExitCode, result.Output)

	// verify the project is correctly created
	resp, err := c.GetProject(t.Context(), &adminv1.GetProjectRequest{
		Org:     "github-branch-test",
		Project: "rill-mgd-branch",
	})
	require.NoError(t, err)
	require.Equal(t, "rill-mgd-branch", resp.Project.Name)

	// verify the primary branch is set to "staging"
	require.Equal(t, "staging", resp.Project.PrimaryBranch)

	// get a github client
	installationID, err := adm.Admin.Github.ManagedOrgInstallationID()
	require.NoError(t, err)
	ghClient := adm.Admin.Github.InstallationClient(installationID, nil)

	// verify the model file is present on the "staging" branch
	verifyGithubRepoBranchContents(t, ghClient, resp.Project.GitRemote, "staging", map[string]string{
		"models/model.sql": `SELECT 'staging' AS env`,
	})
}

// This test require gh cli to be installed on the system.
// Alternatively a personal access token can be set via RILL_TEST_GH_TOKEN environment variable.