/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

package utils

import (
	"encoding/json"
	"reflect"
	"testing"

	"go.mongodb.org/mongo-driver/bson"
)

func TestNewMarshaler(t *testing.T) {
	_, err := NewMarshaler(MsgPack)
	if err != nil {
		t.Error(err)
	}

	_, err = NewMarshaler(JSON)
	if err != nil {
		t.Error(err)
	}

	_, err = NewMarshaler("not_valid")
	errExp := "Unsupported marshaler: not_valid"
	if err.Error() != errExp {
		t.Errorf("Expected %v\n but received %v", errExp, err)
	}
}

func TestJSONMarshaler(t *testing.T) {
	v := "text"
	jsnM := &JSONMarshaler{}
	rcv, err := jsnM.Marshal(v)
	if err != nil {
		t.Error(err)
	}
	var exp string
	json.Unmarshal([]byte(string(rcv)), &exp)
	if exp != v {
		t.Errorf("Expected %v\n but received %v", v, exp)
	}
}

func TestJSONMarshalerError(t *testing.T) {
	v := make(chan string)
	jsnM := &JSONMarshaler{}
	_, err := jsnM.Marshal(v)
	errExp := "json: unsupported type: chan string"
	if err.Error() != errExp {
		t.Errorf("Expected %v\n but received %v", errExp, err)
	}
}

type dummyStruct struct {
	Field string
}

// func TestJSONUnmarshaler(t *testing.T) {
// 	data := []byte(`{"Field": "some_string"}`)
// 	jsnM := &JSONMarshaler{}
// 	dS := dummyStruct{
// 		Field: "some_string",
// 	}

// 	var ndS dummyStruct
// 	err := jsnM.Unmarshal(data, &ndS)
// 	if err != nil {
// 		t.Error(err)
// 	}
// 	if reflect.DeepEqual(dS, ndS) {
// 		t.Errorf("Expected: %s , received: %s", ToJSON(dS), ToJSON(ndS))
// 	}
// }

func TestBSONMarshaler(t *testing.T) {
	v := bson.M{"ts": "test"}
	bsnM := &BSONMarshaler{}
	rcv, err := bsnM.Marshal(v)
	dS := dummyStruct{
		Field: "some_string",
	}
	if err != nil {
		t.Error(err)
	}
	if reflect.DeepEqual(dS, rcv) {
		t.Errorf("Expected: %s , received: %s", ToJSON(dS), ToJSON(rcv))
	}
}

func TestBSONMarshalerError(t *testing.T) {
	v := make(chan string)
	bsnM := &BSONMarshaler{}
	_, err := bsnM.Marshal(v)
	errExp := "no encoder found for chan string"
	if err.Error() != errExp {
		t.Errorf("Expected %v\n but received %v", errExp, err)
	}
}

func TestJSONBufMarshaler(t *testing.T) {
	v := `text`
	jsnM := &JSONBufMarshaler{}
	rcv, err := jsnM.Marshal(v)
	if err != nil {
		t.Error(err)
	}
	var exp string
	json.Unmarshal([]byte(string(rcv)), &exp)
	if exp != v {
		t.Errorf("Expected %v\n but received %v", exp, v)
	}
}

func TestJSONBufUnmarshaler(t *testing.T) {
	v := []byte(`{"Field":"some_string"}`)
	jsnM := &JSONBufMarshaler{}
	s := dummyStruct{
		Field: "some_string",
	}
	var rcv dummyStruct
	err := jsnM.Unmarshal(v, &rcv)
	if err != nil {
		t.Error(err)
	}
	if !reflect.DeepEqual(rcv, s) {
		t.Errorf("Expected %v\n but received %v", s, rcv)
	}
}