mirror of
https://github.com/gortc/stun.git
synced 2024-05-03 14:44:55 +08:00
37 lines
628 B
Go
37 lines
628 B
Go
package stun
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
type errorReader struct{}
|
|
|
|
func (errorReader) Read([]byte) (int, error) {
|
|
return 0, errors.New("failed to read")
|
|
}
|
|
|
|
func TestReadFullHelper(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r == nil {
|
|
t.Error("should panic")
|
|
}
|
|
}()
|
|
readFullOrPanic(errorReader{}, make([]byte, 1))
|
|
}
|
|
|
|
type errorWriter struct{}
|
|
|
|
func (errorWriter) Write([]byte) (int, error) {
|
|
return 0, errors.New("failed to write")
|
|
}
|
|
|
|
func TestWriteHelper(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r == nil {
|
|
t.Error("should panic")
|
|
}
|
|
}()
|
|
writeOrPanic(errorWriter{}, make([]byte, 1))
|
|
}
|