Function TestDropProcedure

testing/go/drop_procedure_test.go:23–242

(t *testing.T)

)

func TestDropProcedure(t *testing.T) {
	RunScripts(t, []ScriptTest{
		{
			Name: "Procedure does not exist",
			Assertions: []ScriptTestAssertion{
				{
					Query:       "DROP PROCEDURE doesnotexist;",
					ExpectedErr: "does not exist",
				},
				{
					Query:    "DROP PROCEDURE IF EXISTS doesnotexist;",
					Expected: []sql.Row{},
				},
			},
		},
		{
			Name: "Basic cases",
			SetUpScript: []string{
				`CREATE PROCEDURE proc1() AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
				`CREATE PROCEDURE proc2(input INT) AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
			},
			Assertions: []ScriptTestAssertion{
				{
					Query:    "CALL proc1();",
					Expected: []sql.Row{},
				},
				{
					Query:    "CALL proc2(99);",
					Expected: []sql.Row{},
				},
				{
					Query:    "DROP PROCEDURE proc1;",
					Expected: []sql.Row{},
				},
				{
					Query:    "DROP PROCEDURE proc2(INT);",
					Expected: []sql.Row{},
				},
				{
					Query:       "CALL proc1();",
					ExpectedErr: "does not exist",
				},
				{
					Query:       "CALL proc2(99);",
					ExpectedErr: "does not exist",
				},
			},
		},
		{
			Name: "Optional type information",
			SetUpScript: []string{
				`CREATE PROCEDURE proc1() AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
				`CREATE PROCEDURE proc2() AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
				`CREATE PROCEDURE proc3(input INT) AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
				`CREATE PROCEDURE proc4(input INT) AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
				`CREATE PROCEDURE proc5(input INT, foo TEXT) AS $$ BEGIN NULL; END; $$ LANGUAGE plpgsql;`,
			},
			Assertions: []ScriptTestAssertion{