res_test.go

package res

import (
	"bytes"
	"testing"
)

func TestUnmarshal(t *testing.T) {
	type args struct {
		bytes []byte
	}
	tests := []struct {
		name     string
		args     args
		wantCode int
		wantErr  bool
	}{
		{"invalid data",
			args{nil},
			0,
			true},
		{"invalid json",
			args{
				[]byte(`{
				"code":200,
				"request_id":"bobbook/2Mch7LMzhj-000001",
				"mess`),
			},
			0,
			true},
		{"ok",
			args{
				[]byte(`{
					"code":200,
					"request_id":"bobbook/2Mch7LMzhj-000001",
					"message":"session created",
					"data":{
						"token":"blah"
					}
				}`),
			},
			200,
			false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var gotToken string
			var gotKV = []KV{{Key: "token", Value: &gotToken}}
			got, err := Unmarshal(bytes.NewReader(tt.args.bytes), gotKV...)
			if (err != nil) != tt.wantErr {
				t.Errorf("Unmarshal() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != nil && tt.wantCode != got.HTTPStatusCode {
				t.Errorf("Unmarshal() code = %v, want %v", got.HTTPStatusCode, tt.wantCode)
			}
			for _, kv := range gotKV {
				if kv.Value == "" {
					t.Error("Unmarshal() kv is empty")
				}
			}
		})
	}
}