Add read/write duration tests

msgp/read.go

@@ -36,6 +36,7 @@ const (
 	IntType
 	UintType
 	NilType
+	DurationType
 	ExtensionType
 
 	// pseudo-types provided
@@ -1308,6 +1309,10 @@ func (m *Reader) ReadIntf() (i interface{}, err error) {
 		i, err = m.ReadTime()
 		return
 
+	case DurationType:
+		i, err = m.ReadDuration()
+		return
+
 	case ExtensionType:
 		var t int8
 		t, err = m.peekExtensionType()

msgp/read_test.go

@@ -30,6 +30,7 @@ func TestReadIntf(t *testing.T) {
 		int64(-40),
 		uint64(9082981),
 		time.Now(),
+		48*time.Hour + 3*time.Minute + 2*time.Second + 3*time.Nanosecond,
 		"hello!",
 		[]byte("hello!"),
 		map[string]interface{}{
@@ -66,6 +67,12 @@ func TestReadIntf(t *testing.T) {
 			if !tm.Equal(v.(time.Time)) {
 				t.Errorf("%v != %v", ts, v)
 			}
+		} else if intd, ok := ts.(time.Duration); ok {
+			/* for time.Duration, cast before comparing */
+			outtd := time.Duration(v.(int64))
+			if intd != outtd {
+				t.Errorf("%v in; %v out", intd, outtd)
+			}
 		} else if !reflect.DeepEqual(v, ts) {
 			t.Errorf("%v in; %v out", ts, v)
 		}

msgp/write.go

@@ -687,6 +687,8 @@ func (mw *Writer) WriteIntf(v interface{}) error {
 		return mw.WriteMapStrIntf(v)
 	case time.Time:
 		return mw.WriteTime(v)
+	case time.Duration:
+		return mw.WriteDuration(v)
 	}
 
 	val := reflect.ValueOf(v)

msgp/write_test.go

@@ -214,6 +214,39 @@ func TestWriteFloat64(t *testing.T) {
 	}
 }
 
+func TestReadWriterDuration(t *testing.T) {
+	var buf bytes.Buffer
+	wr := NewWriter(&buf)
+
+	for i := 0; i < 10000; i++ {
+		buf.Reset()
+		dur := time.Duration(rand.Int63())
+		err := wr.WriteDuration(dur)
+		if err != nil {
+			t.Errorf("Error with %v: %s", dur, err)
+		}
+		err = wr.Flush()
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		bts := buf.Bytes()
+
+		if bts[0] != mint64 {
+			t.Errorf("Leading byte was %x and not %x", bts[0], mint64)
+		}
+
+		wr := NewReader(&buf)
+		d, err := wr.ReadDuration()
+		if err != nil {
+			t.Errorf("Error reading duration: %v", err)
+		}
+		if d != dur {
+			t.Errorf("Got duration %v, want %v", d, dur)
+		}
+	}
+}
+
 func BenchmarkWriteFloat64(b *testing.B) {
 	f := rand.Float64()
 	wr := NewWriter(Nowhere)