/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package thrift_test

import (
  . "thrift"
  "encoding/base64"
  "fmt"
  "json"
  "math"
  "strconv"
  "testing"
)

func TestWriteJSONProtocolBool(t *testing.T) {
  thetype := "boolean"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range BOOL_VALUES {
    if e := p.WriteBool(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s != fmt.Sprint(value) {
      t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
    }
    v := false
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolBool(t *testing.T) {
  thetype := "boolean"
  for _, value := range BOOL_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    if value {
      trans.Write(JSON_TRUE)
    } else {
      trans.Write(JSON_FALSE)
    }
    trans.Flush()
    s := trans.String()
    v, e := p.ReadBool()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolByte(t *testing.T) {
  thetype := "byte"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range BYTE_VALUES {
    if e := p.WriteByte(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s != fmt.Sprint(value) {
      t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
    }
    v := byte(0)
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolByte(t *testing.T) {
  thetype := "byte"
  for _, value := range BYTE_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    trans.WriteString(strconv.Itoa(int(value)))
    trans.Flush()
    s := trans.String()
    v, e := p.ReadByte()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolI16(t *testing.T) {
  thetype := "int16"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range INT16_VALUES {
    if e := p.WriteI16(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s != fmt.Sprint(value) {
      t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
    }
    v := int16(0)
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolI16(t *testing.T) {
  thetype := "int16"
  for _, value := range INT16_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    trans.WriteString(strconv.Itoa(int(value)))
    trans.Flush()
    s := trans.String()
    v, e := p.ReadI16()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolI32(t *testing.T) {
  thetype := "int32"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range INT32_VALUES {
    if e := p.WriteI32(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s != fmt.Sprint(value) {
      t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
    }
    v := int32(0)
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolI32(t *testing.T) {
  thetype := "int32"
  for _, value := range INT32_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    trans.WriteString(strconv.Itoa(int(value)))
    trans.Flush()
    s := trans.String()
    v, e := p.ReadI32()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolI64(t *testing.T) {
  thetype := "int64"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range INT64_VALUES {
    if e := p.WriteI64(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s != fmt.Sprint(value) {
      t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
    }
    v := int64(0)
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolI64(t *testing.T) {
  thetype := "int64"
  for _, value := range INT64_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    trans.WriteString(strconv.Itoa64(value))
    trans.Flush()
    s := trans.String()
    v, e := p.ReadI64()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolDouble(t *testing.T) {
  thetype := "double"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range DOUBLE_VALUES {
    if e := p.WriteDouble(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if math.IsInf(value, 1) {
      if s != JsonQuote(JSON_INFINITY) {
        t.Fatalf("Bad value for %s %v, wrote: %v, expected: %v", thetype, value, s, JsonQuote(JSON_INFINITY))
      }
    } else if math.IsInf(value, -1) {
      if s != JsonQuote(JSON_NEGATIVE_INFINITY) {
        t.Fatalf("Bad value for %s %v, wrote: %v, expected: %v", thetype, value, s, JsonQuote(JSON_NEGATIVE_INFINITY))
      }
    } else if math.IsNaN(value) {
      if s != JsonQuote(JSON_NAN) {
        t.Fatalf("Bad value for %s %v, wrote: %v, expected: %v", thetype, value, s, JsonQuote(JSON_NAN))
      }
    } else {
      if s != fmt.Sprint(value) {
        t.Fatalf("Bad value for %s %v: %s", thetype, value, s)
      }
      v := float64(0)
      if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
        t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
      }
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolDouble(t *testing.T) {
  thetype := "double"
  for _, value := range DOUBLE_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    n := NewNumericFromDouble(value)
    trans.WriteString(n.String())
    trans.Flush()
    s := trans.String()
    v, e := p.ReadDouble()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if math.IsInf(value, 1) {
      if !math.IsInf(v, 1) {
        t.Fatalf("Bad value for %s %v, wrote: %v, received: %v", thetype, value, s, v)
      }
    } else if math.IsInf(value, -1) {
      if !math.IsInf(v, -1) {
        t.Fatalf("Bad value for %s %v, wrote: %v, received: %v", thetype, value, s, v)
      }
    } else if math.IsNaN(value) {
      if !math.IsNaN(v) {
        t.Fatalf("Bad value for %s %v, wrote: %v, received: %v", thetype, value, s, v)
      }
    } else {
      if v != value {
        t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
      }
      if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
        t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
      }
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolString(t *testing.T) {
  thetype := "string"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  for _, value := range STRING_VALUES {
    if e := p.WriteString(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
    if e := p.Flush(); e != nil {
      t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
    }
    s := trans.String()
    if s[0] != '"' || s[len(s)-1] != '"' {
      t.Fatalf("Bad value for %s '%v', wrote '%v', expected: %v", thetype, value, s, fmt.Sprint("\"", value, "\""))
    }
    v := new(string)
    if err := json.Unmarshal([]byte(s), v); err != nil || *v != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, *v)
    }
    trans.Reset()
  }
  trans.Close()
}

func TestReadJSONProtocolString(t *testing.T) {
  thetype := "string"
  for _, value := range STRING_VALUES {
    trans := NewTMemoryBuffer()
    p := NewTJSONProtocol(trans)
    trans.WriteString(JsonQuote(value))
    trans.Flush()
    s := trans.String()
    v, e := p.ReadString()
    if e != nil {
      t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
    }
    if v != value {
      t.Fatalf("Bad value for %s value %v, wrote: %v, received: %v", thetype, value, s, v)
    }
    v1 := new(string)
    if err := json.Unmarshal([]byte(s), v1); err != nil || *v1 != value {
      t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, *v1)
    }
    trans.Reset()
    trans.Close()
  }
}

func TestWriteJSONProtocolBinary(t *testing.T) {
  thetype := "binary"
  value := protocol_bdata
  b64value := make([]byte, base64.StdEncoding.EncodedLen(len(protocol_bdata)))
  base64.StdEncoding.Encode(b64value, value)
  b64String := string(b64value)
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  if e := p.WriteBinary(value); e != nil {
    t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
  }
  if e := p.Flush(); e != nil {
    t.Fatalf("Unable to write %s value %v due to error flushing: %s", thetype, value, e.String())
  }
  s := trans.String()
  expectedString := fmt.Sprint("\"", b64String, "\"")
  if s != expectedString {
    t.Fatalf("Bad value for %s %v\n  wrote:  \"%v\"\nexpected: \"%v\"", thetype, value, s, expectedString)
  }
  v1, err := p.ReadBinary()
  if err != nil {
    t.Fatalf("Unable to read binary: %s", err.String())
  }
  if len(v1) != len(value) {
    t.Fatalf("Invalid value for binary\nexpected: \"%v\"\n   read: \"%v\"", value, v1)
  }
  for k, v := range value {
    if v1[k] != v {
      t.Fatalf("Invalid value for binary at %v\nexpected: \"%v\"\n   read: \"%v\"", k, v, v1[k])
    }
  }
  trans.Close()
}

func TestReadJSONProtocolBinary(t *testing.T) {
  thetype := "binary"
  value := protocol_bdata
  b64value := make([]byte, base64.StdEncoding.EncodedLen(len(protocol_bdata)))
  base64.StdEncoding.Encode(b64value, value)
  b64String := string(b64value)
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  trans.WriteString(JsonQuote(b64String))
  trans.Flush()
  s := trans.String()
  v, e := p.ReadBinary()
  if e != nil {
    t.Fatalf("Unable to read %s value %v due to error: %s", thetype, value, e.String())
  }
  if len(v) != len(value) {
    t.Fatalf("Bad value for %s value length %v, wrote: %v, received length: %v", thetype, len(value), s, len(v))
  }
  for i := 0; i < len(v); i++ {
    if v[i] != value[i] {
      t.Fatalf("Bad value for %s at index %d value %v, wrote: %v, received: %v", thetype, i, value[i], s, v[i])
    }
  }
  v1 := new(string)
  if err := json.Unmarshal([]byte(s), v1); err != nil || *v1 != b64String {
    t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, *v1)
  }
  trans.Reset()
  trans.Close()
}

func TestWriteJSONProtocolList(t *testing.T) {
  thetype := "list"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  p.WriteListBegin(TType(DOUBLE), len(DOUBLE_VALUES))
  for _, value := range DOUBLE_VALUES {
    if e := p.WriteDouble(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
  }
  p.WriteListEnd()
  if e := p.Flush(); e != nil {
    t.Fatalf("Unable to write %s due to error flushing: %s", thetype, e.String())
  }
  str := trans.String()
  str1 := new([]interface{})
  err := json.Unmarshal([]byte(str), str1)
  if err != nil {
    t.Fatalf("Unable to decode %s, wrote: %s", thetype, str)
  }
  l := *str1
  if len(l) < 2 {
    t.Fatalf("List must be at least of length two to include metadata")
  }
  if int(l[0].(float64)) != DOUBLE {
    t.Fatal("Invalid type for list, expected: ", DOUBLE, ", but was: ", l[0])
  }
  if int(l[1].(float64)) != len(DOUBLE_VALUES) {
    t.Fatal("Invalid length for list, expected: ", len(DOUBLE_VALUES), ", but was: ", l[1])
  }
  for k, value := range DOUBLE_VALUES {
    s := l[k+2]
    if math.IsInf(value, 1) {
      if s.(string) != JSON_INFINITY {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_INFINITY), str)
      }
    } else if math.IsInf(value, 0) {
      if s.(string) != JSON_NEGATIVE_INFINITY {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_NEGATIVE_INFINITY), str)
      }
    } else if math.IsNaN(value) {
      if s.(string) != JSON_NAN {
        t.Fatalf("Bad value for %s at index %v  %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_NAN), str)
      }
    } else {
      if s.(float64) != value {
        t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s'", thetype, value, s)
      }
    }
    trans.Reset()
  }
  trans.Close()
}

func TestWriteJSONProtocolSet(t *testing.T) {
  thetype := "set"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  p.WriteSetBegin(TType(DOUBLE), len(DOUBLE_VALUES))
  for _, value := range DOUBLE_VALUES {
    if e := p.WriteDouble(value); e != nil {
      t.Fatalf("Unable to write %s value %v due to error: %s", thetype, value, e.String())
    }
  }
  p.WriteSetEnd()
  if e := p.Flush(); e != nil {
    t.Fatalf("Unable to write %s due to error flushing: %s", thetype, e.String())
  }
  str := trans.String()
  str1 := new([]interface{})
  err := json.Unmarshal([]byte(str), str1)
  if err != nil {
    t.Fatalf("Unable to decode %s, wrote: %s", thetype, str)
  }
  l := *str1
  if len(l) < 2 {
    t.Fatalf("Set must be at least of length two to include metadata")
  }
  if int(l[0].(float64)) != DOUBLE {
    t.Fatal("Invalid type for set, expected: ", DOUBLE, ", but was: ", l[0])
  }
  if int(l[1].(float64)) != len(DOUBLE_VALUES) {
    t.Fatal("Invalid length for set, expected: ", len(DOUBLE_VALUES), ", but was: ", l[1])
  }
  for k, value := range DOUBLE_VALUES {
    s := l[k+2]
    if math.IsInf(value, 1) {
      if s.(string) != JSON_INFINITY {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_INFINITY), str)
      }
    } else if math.IsInf(value, 0) {
      if s.(string) != JSON_NEGATIVE_INFINITY {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_NEGATIVE_INFINITY), str)
      }
    } else if math.IsNaN(value) {
      if s.(string) != JSON_NAN {
        t.Fatalf("Bad value for %s at index %v  %v, wrote: %q, expected: %q, originally wrote: %q", thetype, k, value, s, JsonQuote(JSON_NAN), str)
      }
    } else {
      if s.(float64) != value {
        t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s'", thetype, value, s)
      }
    }
    trans.Reset()
  }
  trans.Close()
}

func TestWriteJSONProtocolMap(t *testing.T) {
  thetype := "map"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  p.WriteMapBegin(TType(I32), TType(DOUBLE), len(DOUBLE_VALUES))
  for k, value := range DOUBLE_VALUES {
    if e := p.WriteI32(int32(k)); e != nil {
      t.Fatalf("Unable to write %s key int32 value %v due to error: %s", thetype, k, e.String())
    }
    if e := p.WriteDouble(value); e != nil {
      t.Fatalf("Unable to write %s value float64 value %v due to error: %s", thetype, value, e.String())
    }
  }
  p.WriteMapEnd()
  if e := p.Flush(); e != nil {
    t.Fatalf("Unable to write %s due to error flushing: %s", thetype, e.String())
  }
  str := trans.String()
  if str[0] != '[' || str[len(str)-1] != ']' {
    t.Fatalf("Bad value for %s, wrote: %q, in go: %q", thetype, str, DOUBLE_VALUES)
  }
  expectedKeyType, expectedValueType, expectedSize, err := p.ReadMapBegin()
  if err != nil {
    t.Fatalf("Error while reading map begin: %s", err.String())
  }
  if expectedKeyType != I32 {
    t.Fatal("Expected map key type ", I32, ", but was ", expectedKeyType)
  }
  if expectedValueType != DOUBLE {
    t.Fatal("Expected map value type ", DOUBLE, ", but was ", expectedValueType)
  }
  if expectedSize != len(DOUBLE_VALUES) {
    t.Fatal("Expected map size of ", len(DOUBLE_VALUES), ", but was ", expectedSize)
  }
  for k, value := range DOUBLE_VALUES {
    ik, err := p.ReadI32()
    if err != nil {
      t.Fatalf("Bad key for %s index %v, wrote: %v, expected: %v, error: %s", thetype, k, ik, string(k), err.String())
    }
    if int(ik) != k {
      t.Fatalf("Bad key for %s index %v, wrote: %v, expected: %v", thetype, k, ik, k)
    }
    dv, err := p.ReadDouble()
    if err != nil {
      t.Fatalf("Bad value for %s index %v, wrote: %v, expected: %v, error: %s", thetype, k, dv, value, err.String())
    }
    s := strconv.Ftoa64(dv, 'g', 10)
    if math.IsInf(value, 1) {
      if !math.IsInf(dv, 1) {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %v, expected: %v", thetype, k, value, s, JsonQuote(JSON_INFINITY))
      }
    } else if math.IsInf(value, 0) {
      if !math.IsInf(dv, 0) {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %v, expected: %v", thetype, k, value, s, JsonQuote(JSON_NEGATIVE_INFINITY))
      }
    } else if math.IsNaN(value) {
      if !math.IsNaN(dv) {
        t.Fatalf("Bad value for %s at index %v  %v, wrote: %v, expected: %v", thetype, k, value, s, JsonQuote(JSON_NAN))
      }
    } else {
      expected := strconv.Ftoa64(value, 'g', 10)
      if s != expected {
        t.Fatalf("Bad value for %s at index %v %v, wrote: %v, expected %v", thetype, k, value, s, expected)
      }
      v := float64(0)
      if err := json.Unmarshal([]byte(s), &v); err != nil || v != value {
        t.Fatalf("Bad json-decoded value for %s %v, wrote: '%s', expected: '%v'", thetype, value, s, v)
      }
    }
    trans.Reset()
  }
  trans.Close()
}


func TestReadWriteJSONStruct(t *testing.T) {
  thetype := "struct"
  trans := NewTMemoryBuffer()
  p := NewTJSONProtocol(trans)
  orig := NewWork()
  orig.Num1 = 25
  orig.Num2 = 102
  orig.Op = ADD
  orig.Comment = "Add: 25 + 102"
  if e := orig.Write(p); e != nil {
    t.Fatalf("Unable to write %s value %#v due to error: %s", thetype, orig, e.String())
  }
  p.Flush()
  t.Log("Memory buffer contents: ", trans.String())
  expectedString := "{\"1\":{\"i32\":25},\"2\":{\"i32\":102},\"3\":{\"i32\":1},\"4\":{\"str\":\"Add: 25 + 102\"}}"
  if expectedString != trans.String() {
    t.Fatalf("Expected JSON Struct with value %#v but have %#v", expectedString, trans.String())
  }
  read := NewWork()
  e := read.Read(p)
  t.Logf("Read %s value: %#v", thetype, read)
  if e != nil {
    t.Fatalf("Unable to read %s due to error: %s", thetype, e.String())
  }
  if !orig.Equals(read) {
    t.Fatalf("Original Write != Read: %#v != %#v ", orig, read)
  }
}

func TestReadWriteJSONProtocol(t *testing.T) {
  ReadWriteProtocolTest(t, NewTJSONProtocolFactory())
}