User MarshalJSON and UnmarshalJSON directly (#116)

audience_test.go

@@ -1,15 +1,14 @@
 package jwt
 
 import (
-	"encoding/json"
 	"testing"
 )
 
 func TestAudienceMarshal(t *testing.T) {
 	f := func(got Audience, want string) {
 		t.Helper()
 
-		raw, err := json.Marshal(got)
+		raw, err := got.MarshalJSON()
 		if err != nil {
 			t.Errorf("want no err, got: %#v", err)
 		}
@@ -30,7 +29,7 @@ func TestAudienceUnmarshal(t *testing.T) {
 		t.Helper()
 
 		var a Audience
-		err := json.Unmarshal([]byte(got), &a)
+		err := a.UnmarshalJSON([]byte(got))
 		if err != nil {
 			t.Errorf("want no err, got: %#v", err)
 		}
@@ -56,7 +55,7 @@ func TestAudienceUnmarshalMalformed(t *testing.T) {
 		t.Helper()
 
 		var a Audience
-		err := json.Unmarshal([]byte(got), &a)
+		err := a.UnmarshalJSON([]byte(got))
 		if err == nil {
 			t.Error("want err")
 		}

build.go

@@ -127,7 +127,7 @@ func encodeHeader(header Header) []byte {
 		// another algorithm? encode below
 	}
 	// returned err is always nil, see *Header.MarshalJSON
-	buf, _ := json.Marshal(header)
+	buf, _ := header.MarshalJSON()
 
 	encoded := make([]byte, b64EncodedLen(len(buf)))
 	b64Encode(encoded, buf)

jwt_test.go

@@ -2,7 +2,6 @@ package jwt
 
 import (
 	"encoding/base64"
-	"encoding/json"
 	"strings"
 	"testing"
 )
@@ -39,7 +38,7 @@ func TestMarshalHeader(t *testing.T) {
 	f := func(h *Header, want string) {
 		t.Helper()
 
-		raw, err := json.Marshal(h)
+		raw, err := h.MarshalJSON()
 		if err != nil {
 			t.Errorf("unexpected error: %v", err)
 		}

numeric_date.go

@@ -24,6 +24,9 @@ func NewNumericDate(t time.Time) *NumericDate {
 
 // MarshalJSON implements the json.Marshaler interface.
 func (t *NumericDate) MarshalJSON() ([]byte, error) {
+	if t == nil || t.IsZero() {
+		return []byte("null"), nil
+	}
 	return []byte(strconv.FormatInt(t.Unix(), 10)), nil
 }
 

numeric_date_test.go

@@ -1,7 +1,6 @@
 package jwt
 
 import (
-	"encoding/json"
 	"strconv"
 	"testing"
 	"time"
@@ -11,7 +10,7 @@ func TestNumericDateMarshal(t *testing.T) {
 	f := func(got *NumericDate, want string) {
 		t.Helper()
 
-		raw, err := json.Marshal(got)
+		raw, err := got.MarshalJSON()
 		if err != nil {
 			t.Errorf("want no err, got: %#v", err)
 		}
@@ -33,7 +32,7 @@ func TestNumericDateUnmarshal(t *testing.T) {
 		t.Helper()
 
 		var got NumericDate
-		err := json.Unmarshal([]byte(s), &got)
+		err := got.UnmarshalJSON([]byte(s))
 		if err != nil {
 			t.Errorf("want no err, got: %#v", err)
 		}
@@ -52,7 +51,7 @@ func TestNumericDateUnmarshalMalformed(t *testing.T) {
 		t.Helper()
 
 		var nd NumericDate
-		err := json.Unmarshal([]byte(got), &nd)
+		err := nd.UnmarshalJSON([]byte(got))
 		if err == nil {
 			t.Error("want err")
 		}