diff --git a/internal/handlers/base.go b/internal/handlers/base.go index babd9fb..aa83e0b 100644 --- a/internal/handlers/base.go +++ b/internal/handlers/base.go @@ -87,7 +87,7 @@ func(f *BaseSessionHandler) Process(rqs RequestSession) (RequestSession, error) func(f *BaseSessionHandler) Output(rqs RequestSession) (RequestSession, error) { var err error - _, err = rqs.Engine.WriteResult(rqs.Ctx, rqs.Writer) + _, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer) return rqs, err } diff --git a/internal/http/at_session_handler.go b/internal/http/at_session_handler.go index 53c4ba2..25da954 100644 --- a/internal/http/at_session_handler.go +++ b/internal/http/at_session_handler.go @@ -87,6 +87,6 @@ func (ash *ATSessionHandler) Output(rqs handlers.RequestSession) (handlers.Reque return rqs, err } - _, err = rqs.Engine.WriteResult(rqs.Ctx, rqs.Writer) + _, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer) return rqs, err } \ No newline at end of file diff --git a/internal/http/http_test.go b/internal/http/http_test.go index 48d04ca..8f6f312 100644 --- a/internal/http/http_test.go +++ b/internal/http/http_test.go @@ -69,7 +69,7 @@ func TestATSessionHandler_ServeHTTP(t *testing.T) { mh.GetRequestParserFunc = func() handlers.RequestParser { return mrp } mh.OutputFunc = func(rs handlers.RequestSession) (handlers.RequestSession, error) { return rs, nil } mh.ResetFunc = func(rs handlers.RequestSession) (handlers.RequestSession, error) { return rs, nil } - me.WriteResultFunc = func(context.Context, io.Writer) (int, error) { return 0, nil } + me.FlushFunc = func(context.Context, io.Writer) (int, error) { return 0, nil } }, formData: url.Values{ "phoneNumber": []string{"+1234567890"}, @@ -178,7 +178,7 @@ func TestATSessionHandler_Output(t *testing.T) { input: handlers.RequestSession{ Continue: true, Engine: &httpmocks.MockEngine{ - WriteResultFunc: func(context.Context, io.Writer) (int, error) { + FlushFunc: func(context.Context, io.Writer) (int, error) { return 0, nil }, }, @@ -192,7 +192,7 @@ func TestATSessionHandler_Output(t *testing.T) { input: handlers.RequestSession{ Continue: false, Engine: &httpmocks.MockEngine{ - WriteResultFunc: func(context.Context, io.Writer) (int, error) { + FlushFunc: func(context.Context, io.Writer) (int, error) { return 0, nil }, }, @@ -202,11 +202,11 @@ func TestATSessionHandler_Output(t *testing.T) { expectedError: false, }, { - name: "WriteResult error", + name: "Flush error", input: handlers.RequestSession{ Continue: true, Engine: &httpmocks.MockEngine{ - WriteResultFunc: func(context.Context, io.Writer) (int, error) { + FlushFunc: func(context.Context, io.Writer) (int, error) { return 0, errors.New("write error") }, }, diff --git a/internal/mocks/httpmocks/enginemock.go b/internal/mocks/httpmocks/enginemock.go index d5c6b20..12d07f4 100644 --- a/internal/mocks/httpmocks/enginemock.go +++ b/internal/mocks/httpmocks/enginemock.go @@ -7,10 +7,10 @@ import ( // MockEngine implements the engine.Engine interface for testing type MockEngine struct { - InitFunc func(context.Context) (bool, error) - ExecFunc func(context.Context, []byte) (bool, error) - WriteResultFunc func(context.Context, io.Writer) (int, error) - FinishFunc func() error + InitFunc func(context.Context) (bool, error) + ExecFunc func(context.Context, []byte) (bool, error) + FlushFunc func(context.Context, io.Writer) (int, error) + FinishFunc func() error } func (m *MockEngine) Init(ctx context.Context) (bool, error) { @@ -21,8 +21,8 @@ func (m *MockEngine) Exec(ctx context.Context, input []byte) (bool, error) { return m.ExecFunc(ctx, input) } -func (m *MockEngine) WriteResult(ctx context.Context, w io.Writer) (int, error) { - return m.WriteResultFunc(ctx, w) +func (m *MockEngine) Flush(ctx context.Context, w io.Writer) (int, error) { + return m.FlushFunc(ctx, w) } func (m *MockEngine) Finish() error {