You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.1 KiB
88 lines
2.1 KiB
package encoding |
|
|
|
import ( |
|
"encoding/xml" |
|
"fmt" |
|
"runtime/debug" |
|
"testing" |
|
) |
|
|
|
type codec struct{} |
|
|
|
func (c codec) Marshal(v interface{}) ([]byte, error) { |
|
panic("implement me") |
|
} |
|
|
|
func (c codec) Unmarshal(data []byte, v interface{}) error { |
|
panic("implement me") |
|
} |
|
|
|
func (c codec) Name() string { |
|
return "" |
|
} |
|
|
|
// codec2 is a Codec implementation with xml. |
|
type codec2 struct{} |
|
|
|
func (codec2) Marshal(v interface{}) ([]byte, error) { |
|
return xml.Marshal(v) |
|
} |
|
|
|
func (codec2) Unmarshal(data []byte, v interface{}) error { |
|
return xml.Unmarshal(data, v) |
|
} |
|
|
|
func (codec2) Name() string { |
|
return "xml" |
|
} |
|
|
|
func TestRegisterCodec(t *testing.T) { |
|
f := func() { RegisterCodec(nil) } |
|
funcDidPanic, panicValue, _ := didPanic(f) |
|
if !funcDidPanic { |
|
t.Fatalf(fmt.Sprintf("func should panic\n\tPanic value:\t%#v", panicValue)) |
|
} |
|
if panicValue != "cannot register a nil Codec" { |
|
t.Fatalf("panic error got %s want cannot register a nil Codec", panicValue) |
|
} |
|
f = func() { |
|
RegisterCodec(codec{}) |
|
} |
|
funcDidPanic, panicValue, _ = didPanic(f) |
|
if !funcDidPanic { |
|
t.Fatalf(fmt.Sprintf("func should panic\n\tPanic value:\t%#v", panicValue)) |
|
} |
|
if panicValue != "cannot register Codec with empty string result for Name()" { |
|
t.Fatalf("panic error got %s want cannot register Codec with empty string result for Name()", panicValue) |
|
} |
|
codec := codec2{} |
|
RegisterCodec(codec) |
|
got := GetCodec("xml") |
|
if got != codec { |
|
t.Fatalf("RegisterCodec(%v) want %v got %v", codec, codec, got) |
|
} |
|
} |
|
|
|
// PanicTestFunc defines a func that should be passed to the assert.Panics and assert.NotPanics |
|
// methods, and represents a simple func that takes no arguments, and returns nothing. |
|
type PanicTestFunc func() |
|
|
|
// didPanic returns true if the function passed to it panics. Otherwise, it returns false. |
|
func didPanic(f PanicTestFunc) (bool, interface{}, string) { |
|
didPanic := false |
|
var message interface{} |
|
var stack string |
|
func() { |
|
defer func() { |
|
if message = recover(); message != nil { |
|
didPanic = true |
|
stack = string(debug.Stack()) |
|
} |
|
}() |
|
|
|
// call the target function |
|
f() |
|
}() |
|
|
|
return didPanic, message, stack |
|
}
|
|
|