Function TestScriptHandler_Execute_Success

controllers/script_test.go:15–43

(t *testing.T)

)

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

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

	scanner := mockgen.NewMockScanner(ctrl)
	scanner.EXPECT().Err().Return(nil)
	scanner.EXPECT().Bytes().Return([]byte(`[{"n":1}]`))

	executor := mockgen.NewMockQueryExecutor(ctrl)
	executor.EXPECT().ExecuteScriptsCtx(gomock.Any(), http.MethodGet, `SELECT 1`, 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/list", nil)
	req = mux.SetURLVars(req, map[string]string{"queriesLocation": "queries", "script": "list", "database": "prest-test"})
	req = req.WithContext(withTestTimeout(req.Context()))
	rec := httptest.NewRecorder()

	h.Execute(rec, req)

	require.Equal(t, http.StatusOK, rec.Code)
	require.Contains(t, rec.Body.String(), `"n":1`)
}

func TestScriptHandler_Execute_DefaultDatabase(t *testing.T) {
	ctrl := gomock.NewController(t)