From 1d6a9c0950eb8b6cb24bf1c8bedf38873ddd3151 Mon Sep 17 00:00:00 2001 From: Henrik Hautakoski Date: Fri, 12 Aug 2022 17:56:59 +0200 Subject: [PATCH] Adding src/api/test.go --- src/api/test.go | 46 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 src/api/test.go diff --git a/src/api/test.go b/src/api/test.go new file mode 100644 index 0000000..102dc10 --- /dev/null +++ b/src/api/test.go @@ -0,0 +1,46 @@ + +package api + +import ( + "strings" + "github.com/eosswedenorg-go/haproxy/agentcheck" +) + +type TestApi struct { + response agentcheck.Response +} + +func parseResponse(resp string) (agentcheck.Response, error) { + + parts := strings.SplitN(resp, "#", 2) + + // Status with message + if len(parts) > 1 { + rtype := agentcheck.StatusMessageResponseType(parts[0]) + return agentcheck.NewStatusMessageResponse(rtype, parts[1]), nil + } + + // Only status. + rtype := agentcheck.StatusResponseType(parts[0]) + return agentcheck.NewStatusResponse(rtype), nil +} + +func NewTestApi(response string) TestApi { + + resp, _ := parseResponse(response) + + return TestApi{ + response: resp, + } +} + +func (t TestApi) LogInfo() LogParams { + return LogParams{ + "type", "TestApi", + "response", t.response, + } +} + +func (t TestApi) Call() (agentcheck.Response, string) { + return t.response, "" +}