raw_message_test.go

package traversal_test

import (
	"encoding/json"
	"fmt"
	"testing"

	tr "github.com/dougfort/traversal"
	"github.com/pkg/errors"
)

func TestGetStringFromRawMessage(t *testing.T) {
	var err error
	const testString = "test string"
	testJSON := fmt.Sprintf("\"%s\"", testString)

	var rawMessage json.RawMessage
	err = rawMessage.UnmarshalJSON([]byte(testJSON))
	if err != nil {
		t.Fatalf("rawMessage.UnmarshalJSON() failed: %s", err)
	}

	result, err := tr.GetStringFromRawMessage(rawMessage)
	if err != nil {
		t.Fatalf("GetStringFromRawMessage(%s) failed: %s", rawMessage, err)
	}
	if result != testString {
		t.Fatalf("expected '%s', got '%s", testString, result)
	}
}

func TestGetBoolFromRawMessage(t *testing.T) {
	var err error
	testJSON := "true"

	var rawMessage json.RawMessage
	err = rawMessage.UnmarshalJSON([]byte(testJSON))
	if err != nil {
		t.Fatalf("rawMessage.UnmarshalJSON() failed: %s", err)
	}

	result, err := tr.GetBoolFromRawMessage(rawMessage)
	if err != nil {
		t.Fatalf("GetBoolFromRawMessage(%s) failed: %s", rawMessage, err)
	}
	if !result {
		t.Fatalf("expected 'true', got %t", result)
	}
}

func TestGetInt32FromRawMessage(t *testing.T) {
	var err error
	var testNumber int32 = 1423
	testJSON := fmt.Sprintf("%d", testNumber)

	var rawMessage json.RawMessage
	err = rawMessage.UnmarshalJSON([]byte(testJSON))
	if err != nil {
		t.Fatalf("rawMessage.UnmarshalJSON() failed: %s", err)
	}

	result, err := tr.GetInt32FromRawMessage(rawMessage)
	if err != nil {
		t.Fatalf("GetInt32FromRawMessage(%s) failed: %s", rawMessage, err)
	}
	if result != testNumber {
		t.Fatalf("expected %d, got %d", testNumber, result)
	}
}

func TestGetSliceFromRawMessage(t *testing.T) {
	var err error
	data := `[
		{"a": "a1"},
		{"b": "b1"}
	]`

	var rawMessage json.RawMessage
	err = rawMessage.UnmarshalJSON([]byte(data))
	if err != nil {
		t.Fatalf("rawMessage.UnmarshalJSON() failed: %s", err)
	}

	result, err := tr.GetSliceFromRawMessage(rawMessage)
	if err != nil {
		t.Fatalf("GetSliceFromRawMessage(%s) failed: %s", rawMessage, err)
	}
	if len(result) != 2 {
		t.Fatalf("expected %d, got %d", 2, len(result))
	}

	result0, err := result[0].MarshalJSON()
	if err != nil {
		t.Fatalf("MarshalJSON failed: %s", err)
	}
	if string(result0) != "{\"a\": \"a1\"}" {
		t.Fatalf("result0 = %s", result0)
	}

	result1, err := result[1].MarshalJSON()
	if err != nil {
		t.Fatalf("MarshalJSON failed: %s", err)
	}
	if string(result1) != "{\"b\": \"b1\"}" {
		t.Fatalf("result1 = %s", result1)
	}
}

func TestGetMsgFromRawMessage(t *testing.T) {
	var err error
	const testString = "test string"
	testJSON := fmt.Sprintf("\"%s\"", testString)

	var rawMessage json.RawMessage
	err = rawMessage.UnmarshalJSON([]byte(testJSON))
	if err != nil {
		t.Fatalf("rawMessage.UnmarshalJSON() failed: %s", err)
	}

	result, err := tr.GetMsgFromRawMessage(rawMessage)
	if err != nil {
		t.Fatalf("GetStringFromRawMessage(%s) failed: %s", rawMessage, err)
	}

	var res string
	err = json.Unmarshal(result, &res)
	if err != nil {
		t.Fatal(errors.Wrap(err, "failed to unmarshal msg from raw message"))
	}
	if res != testString {
		t.Fatalf("expected '%s', got '%s", testString, result)
	}
}