Function TestScriptHandler_Execute_DefaultDatabase

controllers/script_test.go:45–73

(t *testing.T)

}

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

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

	scanner := mockgen.NewMockScanner(ctrl)
	scanner.EXPECT().Err().Return(nil)
	scanner.EXPECT().Bytes().Return([]byte(`[]`))

	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")
	db.EXPECT().GetDatabase().Return("prest-test").AnyTimes()

	h := NewScriptHandler(Deps{Scripts: scripts, Executor: executor, DB: db, PGDatabase: "prest-test"})
	req := httptest.NewRequest(http.MethodGet, "/queries/ping", nil)
	req = mux.SetURLVars(req, map[string]string{"queriesLocation": "queries", "script": "ping"})
	req = req.WithContext(withTestTimeout(req.Context()))
	rec := httptest.NewRecorder()

	h.Execute(rec, req)

	require.Equal(t, http.StatusOK, rec.Code)
}

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