package runtime_test

import (
	"bytes"
	"testing"

	"github.com/grpc-ecosystem/grpc-gateway/runtime"
	"google.golang.org/genproto/googleapis/api/httpbody"
)

func TestHTTPBodyContentType(t *testing.T) {
	m := runtime.HTTPBodyMarshaler{
		&runtime.JSONPb{
			OrigName: true,
		},
	}
	expected := "CustomContentType"
	message := &httpbody.HttpBody{
		ContentType: expected,
	}
	res := m.ContentType()
	if res != "application/json" {
		t.Errorf("content type not equal (%q, %q)", res, expected)
	}
	res = m.ContentTypeFromMessage(message)
	if res != expected {
		t.Errorf("content type not equal (%q, %q)", res, expected)
	}
}

func TestHTTPBodyMarshal(t *testing.T) {
	m := runtime.HTTPBodyMarshaler{
		&runtime.JSONPb{
			OrigName: true,
		},
	}
	expected := []byte("Some test")
	message := &httpbody.HttpBody{
		Data: expected,
	}
	res, err := m.Marshal(message)
	if err != nil {
		t.Errorf("m.Marshal(%#v) failed with %v; want success", message, err)
	}
	if !bytes.Equal(res, expected) {
		t.Errorf("Marshalled data not equal (%q, %q)", res, expected)

	}
}