Function TestScriptHandler_ExecuteScriptQuery_ExecuteError

controllers/script_test.go:127–151

(t *testing.T)

}

func TestScriptHandler_ExecuteScriptQuery_ExecuteError(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	scripts := mockgen.NewMockScriptRunner(ctrl)
	scripts.EXPECT().GetScript(http.MethodGet, "queries", "bad").Return("/tmp/bad.sql", nil)
	scripts.EXPECT().ParseScript("/tmp/bad.sql", gomock.Any()).Return(`SELECT bad`, nil, nil)

	scanner := mockgen.NewMockScanner(ctrl)
	scanner.EXPECT().Err().Return(errors.New("syntax error"))

	executor := mockgen.NewMockQueryExecutor(ctrl)
	executor.EXPECT().ExecuteScriptsCtx(gomock.Any(), http.MethodGet, `SELECT bad`, gomock.Any()).Return(scanner)

	db := mockgen.NewMockDatabaseRegistry(ctrl)
	db.EXPECT().SetDatabase("prest-test")

	h := NewScriptHandler(Deps{Scripts: scripts, Executor: executor, DB: db, PGDatabase: "prest-test"})
	req := httptest.NewRequest(http.MethodGet, "/queries/bad", nil)
	req = req.WithContext(withTestTimeout(req.Context()))

	_, err := h.ExecuteScriptQuery(req, "queries", "bad")
	require.Error(t, err)
	require.Contains(t, err.Error(), "could not execute sql")
}

func TestExtractHeaders(t *testing.T) {
	req := httptest.NewRequest(http.MethodGet, "/", nil)