💾 Archived View for source.community › ckaznocha › gemini › blob › main › geminitest › recorder.go captured on 2024-02-05 at 09:58:06. Gemini links have been rewritten to link to archived content

-=-=-=-=-=-=-

                                                         .
,-. ,-. . . ,-. ,-. ,-.    ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | |   |   |-'    |   | | | | | | | | | | | | | |  | |
`-' `-' `-^ '   `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
                                                             /|
                                                            `-'

ckaznocha / gemini

git clone https://source.community/ckaznocha/gemini.git

Branches

Log

Tree

/geminitest/recorder.go (main)

↑ /geminitest

blob

View raw contents of /geminitest/recorder.go (main)

────╮
   1│ package geminitest
   2│ 
   3│ import (
   4│ 	"bytes"
   5│ 	"context"
   6│ 	"io"
   7│ 
   8│ 	"source.community/ckaznocha/gemini"
   9│ )
  10│ 
  11│ // ResponseRecorder may be used in test to capture the output of a Handler.
  12│ type ResponseRecorder struct {
  13│ 	Body *bytes.Buffer
  14│ 	Meta string
  15│ 	Code gemini.StatusCode
  16│ }
  17│ 
  18│ var _ gemini.ResponseWriter = (*ResponseRecorder)(nil)
  19│ 
  20│ // NewResponseRecorder returns a response recorder ready to use.
  21│ func NewResponseRecorder() *ResponseRecorder {
  22│ 	return &ResponseRecorder{
  23│ 		Body: bytes.NewBuffer(nil),
  24│ 	}
  25│ }
  26│ 
  27│ // Failure implements the Failure method of the ResponseWriter interface.
  28│ func (r *ResponseRecorder) Failure(ctx context.Context, code gemini.StatusCode, msg string) {
  29│ 	category := code.ToCategory()
  30│ 	if category < gemini.StatusCategoryTemporaryFailure || category > gemini.StatusCategoryPermanentFailure {
  31│ 		code = gemini.StatusPermanentFailure
  32│ 	}
  33│ 
  34│ 	r.Code = code
  35│ 	r.Meta = msg
  36│ }
  37│ 
  38│ // Input implements the Input method of the ResponseWriter interface.
  39│ func (r *ResponseRecorder) Input(ctx context.Context, prompt string, isSensitive bool) {
  40│ 	code := gemini.StatusInput
  41│ 	if isSensitive {
  42│ 		code = gemini.StatusSensitiveInput
  43│ 	}
  44│ 
  45│ 	r.Code = code
  46│ 	r.Meta = prompt
  47│ }
  48│ 
  49│ // Redirect implements the Redirect method of the ResponseWriter interface.
  50│ func (r *ResponseRecorder) Redirect(ctx context.Context, redirectURL string, isPermanant bool) {
  51│ 	code := gemini.StatusTemporaryRedirect
  52│ 	if isPermanant {
  53│ 		code = gemini.StatusPermanentRedirect
  54│ 	}
  55│ 
  56│ 	r.Code = code
  57│ 	r.Meta = redirectURL
  58│ }
  59│ 
  60│ // Success implements the Success method of the ResponseWriter interface.
  61│ func (r *ResponseRecorder) Success(ctx context.Context, mimeType string) io.Writer {
  62│ 	r.Code = gemini.StatusSuccess
  63│ 	r.Meta = mimeType
  64│ 
  65│ 	return r.Body
  66│ }
────╯

· · ·

info@source.community