From 9789b5403fc91049e7224bf527a3c282216068fa Mon Sep 17 00:00:00 2001 From: Maxim Lebedev Date: Wed, 13 Dec 2017 15:06:44 +0500 Subject: [PATCH] :wheelchair: Added ErrInvalidDataType value for errors checking --- content.go | 6 +++++- invalid_test.go | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/content.go b/content.go index 31b1142..2585cd7 100644 --- a/content.go +++ b/content.go @@ -31,6 +31,10 @@ type ( } ) +// ErrInvalidDataType is returned when ContentFormat function are passed a data argument of invalid +// type. +var ErrInvalidDataType = errors.New("invalid data type") + // ContentFormat transforms data to a DOM-based format to represent the content of the page. func ContentFormat(data interface{}) ([]Node, error) { var doc html.Node @@ -54,7 +58,7 @@ func ContentFormat(data interface{}) ([]Node, error) { } doc = *dom default: - return nil, errors.New("invalid data type, use []byte or string") + return nil, ErrInvalidDataType } var content []Node diff --git a/invalid_test.go b/invalid_test.go index 07db307..25b4f07 100644 --- a/invalid_test.go +++ b/invalid_test.go @@ -11,7 +11,7 @@ const ( var invalidAccount = &Account{} func TestInvalidContentFormat(t *testing.T) { - if _, err := ContentFormat(invalidContent); err == nil { + if _, err := ContentFormat(invalidContent); err != ErrInvalidDataType { t.Error() } }