selfcare init

This commit is contained in:
zhangsz
2025-03-03 11:40:37 +08:00
parent 19f09dd7ea
commit aca2bace68
692 changed files with 273972 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
/*
Package oitest provides useful tools to be used for testing implementations of interfaces in Go's standard "io" package.
*/
package oitest

View File

@@ -0,0 +1,12 @@
package oitest
import (
"math/rand"
"time"
)
var (
randomness = rand.New(rand.NewSource( time.Now().UTC().UnixNano() ))
)

View File

@@ -0,0 +1,80 @@
package oitest
import (
"bytes"
"io"
)
// ShortWriter is useful for testing things that makes use of
// or accept an io.Writer.
//
// In particular, it is useful to see if that thing can handle
// an io.Writer that does a "short write".
//
// A "short write" is the case when a call like:
//
// n, err := w.Write(p)
//
// Returns an n < len(p) and err == io.ErrShortWrite.
//
// A thing that can "handle" this situation might try
// writing again, but only what didn't get written.
//
// For a simple example of this:
//
// n, err := w.Write(p)
//
// if io.ErrShortWrite == err {
// n2, err2 := w.Write(p[n:])
// }
//
// Note that the second call to the Write method passed
// 'p[n:]' (instead of just 'p'), to account for 'n' bytes
// already written (with the first call to the Write
// method).
//
// A more "production quality" version of this would likely
// be in a loop, but such that that loop had "guards" against
// looping forever, and also possibly looping for "too long".
type ShortWriter struct {
buffer bytes.Buffer
}
// Write makes it so ShortWriter fits the io.Writer interface.
//
// ShortWriter's version of Write will "short write" if len(p) >= 2,
// else it will
func (w *ShortWriter) Write(p []byte) (int, error) {
if len(p) < 1 {
return 0, nil
}
m := 1
if limit := len(p)-1; limit > 1 {
m += randomness.Intn(len(p)-1)
}
n, err := w.buffer.Write(p[:m])
err = nil
if n < len(p) {
err = io.ErrShortWrite
}
return n, err
}
// Returns what was written to the ShortWriter as a []byte.
func (w ShortWriter) Bytes() []byte {
return w.buffer.Bytes()
}
// Returns what was written to the ShortWriter as a string.
func (w ShortWriter) String() string {
return w.buffer.String()
}

View File

@@ -0,0 +1,96 @@
package oitest
import (
"bytes"
"io"
)
// WritesThenErrorWriter is useful for testing things that makes
// use of or accept an io.Writer.
//
// In particular, it is useful to see if that thing can handle
// an io.Writer that does a "short write".
//
// A "short write" is the case when a call like:
//
// n, err := w.Write(p)
//
// Returns an n < len(p) and err == io.ErrShortWrite.
//
// A thing that can "handle" this situation might try
// writing again, but only what didn't get written.
//
// For a simple example of this:
//
// n, err := w.Write(p)
//
// if io.ErrShortWrite == err {
// n2, err2 := w.Write(p[n:])
// }
//
// Note that the second call to the Write method passed
// 'p[n:]' (instead of just 'p'), to account for 'n' bytes
// already written (with the first call to the Write
// method).
//
// A more "production quality" version of this would likely
// be in a loop, but such that that loop had "guards" against
// looping forever, and also possibly looping for "too long".
type WritesThenErrorWriter struct {
buffer bytes.Buffer
err error
numbersWritten []int
writeNumber int
}
// NewWritesThenErrorWriter returns a new *WritesThenErrorWriter.
func NewWritesThenErrorWriter(err error, numbersWritten ...int) *WritesThenErrorWriter {
slice := make([]int, len(numbersWritten))
copy(slice, numbersWritten)
writer := WritesThenErrorWriter{
err:err,
numbersWritten:slice,
writeNumber:0,
}
return &writer
}
// Write makes it so *WritesThenErrorWriter fits the io.Writer interface.
//
// *WritesThenErrorWriter's version of Write will "short write" all but
// the last call to write, where it will return the specified error (which
// could, of course, be nil, if that was specified).
func (writer *WritesThenErrorWriter) Write(p []byte) (int, error) {
m := writer.numbersWritten[writer.writeNumber]
writer.buffer.Write(p[:m])
writer.writeNumber++
if len(writer.numbersWritten) == writer.writeNumber {
return m, writer.err
}
return m, io.ErrShortWrite
}
// Returns what was written to the ShortWriter as a []byte.
func (w WritesThenErrorWriter) Bytes() []byte {
return w.buffer.Bytes()
}
// Returns what was written to the ShortWriter as a string.
func (w WritesThenErrorWriter) String() string {
return w.buffer.String()
}