chore: fix cmd test

Author: tianzhou

cmd/dump/dump_integration_test.go

@@ -114,7 +114,7 @@ func runTenantSchemaTest(t *testing.T, testDataDir string) {
 	defer containerInfo.Terminate(ctx, t)
 
 	// Load public schema types first
-	publicSQL, err := os.ReadFile(fmt.Sprintf("../../testdata/%s/public.sql", testDataDir))
+	publicSQL, err := os.ReadFile(fmt.Sprintf("../../testdata/dump/%s/public.sql", testDataDir))
 	if err != nil {
 		t.Fatalf("Failed to read public.sql: %v", err)
 	}
@@ -134,7 +134,7 @@ func runTenantSchemaTest(t *testing.T, testDataDir string) {
 	}
 
 	// Read the tenant SQL
-	tenantSQL, err := os.ReadFile(fmt.Sprintf("../../testdata/%s/pgschema.sql", testDataDir))
+	tenantSQL, err := os.ReadFile(fmt.Sprintf("../../testdata/dump/%s/pgschema.sql", testDataDir))
 	if err != nil {
 		t.Fatalf("Failed to read tenant.sql: %v", err)
 	}
@@ -187,7 +187,7 @@ func runTenantSchemaTest(t *testing.T, testDataDir string) {
 	}
 
 	// Read expected output
-	expectedBytes, err := os.ReadFile(fmt.Sprintf("../../testdata/%s/pgschema.sql", testDataDir))
+	expectedBytes, err := os.ReadFile(fmt.Sprintf("../../testdata/dump/%s/pgschema.sql", testDataDir))
 	if err != nil {
 		t.Fatalf("Failed to read expected output: %v", err)
 	}

cmd/root_test.go

@@ -18,7 +18,7 @@ func TestRootCommand(t *testing.T) {
 	}
 
 	output := buf.String()
-	if !strings.Contains(output, "A simple CLI tool with version information") {
+	if !strings.Contains(output, "Declarative schema migration for Postgres") {
 		t.Errorf("expected help output to contain description, got: %s", output)
 	}
 }
@@ -35,15 +35,15 @@ func TestRootCommandWithoutArgs(t *testing.T) {
 	}
 
 	output := buf.String()
-	if !strings.Contains(output, "A simple CLI tool with version information") {
+	if !strings.Contains(output, "Declarative schema migration for Postgres") {
 		t.Errorf("expected output to contain description, got: %s", output)
 	}
 }
 
 func TestRootCommandHasSubcommands(t *testing.T) {
 	commands := RootCmd.Commands()
 
-	expectedCommands := []string{"version", "dump", "plan"}
+	expectedCommands := []string{"dump", "plan", "apply"}
 	commandNames := make([]string, len(commands))
 	for i, cmd := range commands {
 		commandNames[i] = cmd.Name()