package main

import (
	"reflect"
	"testing"
)

func Test_delimiterFramer_Frame(t *testing.T) {
	type fields struct {
		stxCh    []byte
		endCh    []byte
		maxBytes int
	}
	type args struct {
		bs []byte
	}
	tests := []struct {
		name   string
		fields fields
		args   args
		want   FramerResult
	}{
		// TODO: Add test cases.
		{
			name: "no stx, 1 etx",
			fields: fields{
				stxCh: []byte{},
				endCh: []byte("\n"),
			},
			args: args{
				bs: []byte("12345678\n"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("12345678\n")},
				remains: []byte{},
			},
		},
		{
			name: "1 stx, 1 etx, all non-packet byte",
			fields: fields{
				stxCh: []byte("0"),
				endCh: []byte("\n"),
			},
			args: args{
				bs: []byte("12345678\n"),
			},
			want: &frameResult{
				frames:  [][]byte{},
				remains: []byte{},
			},
		},
		{
			name: "1 stx, 1 etx",
			fields: fields{
				stxCh: []byte("0"),
				endCh: []byte("\n"),
			},
			args: args{
				bs: []byte("123045678\n"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("045678\n")},
				remains: []byte{},
			},
		},
		{
			name: "no stx, 2 etx",
			fields: fields{
				stxCh: []byte{},
				endCh: []byte("\r\n"),
			},
			args: args{
				bs: []byte("12345678\r\n"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("12345678\r\n")},
				remains: []byte{},
			},
		},
		{
			name: "1 stx, 2 etx",
			fields: fields{
				stxCh: []byte("0"),
				endCh: []byte("\r\n"),
			},
			args: args{
				bs: []byte("012345678\r\n"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("012345678\r\n")},
				remains: []byte{},
			},
		},
		{
			name: "no stx, 2 etx, multiple",
			fields: fields{
				stxCh: []byte{},
				endCh: []byte("\r\n"),
			},
			args: args{
				bs: []byte("12345678\r\n1\r\n2\r\n"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("12345678\r\n"), []byte("1\r\n"), []byte("2\r\n")},
				remains: []byte{},
			},
		},
		{
			name: "no stx, 2 etx, multiple, incomplete",
			fields: fields{
				stxCh: []byte{},
				endCh: []byte("\r\n"),
			},
			args: args{
				bs: []byte("12345678\r\n1\r\n2\r\nxyz"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("12345678\r\n"), []byte("1\r\n"), []byte("2\r\n")},
				remains: []byte("xyz"),
			},
		},
		{
			name: "2 stx, 2 etx, multiple, incomplete",
			fields: fields{
				stxCh: []byte("\x00\x01"),
				endCh: []byte("\r\n"),
			},
			args: args{
				bs: []byte("123\x00\x0145678\r\n1\r\n2\r\nxyz\x00\x01xyz\r\nxyz"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("\x00\x0145678\r\n"), []byte("\x00\x01xyz\r\n")},
				remains: []byte{},
			},
		},
		{
			name: "2 stx, 2 etx, multiple, max bytes, incomplete",
			fields: fields{
				stxCh:    []byte("\x00\x01"),
				endCh:    []byte("\r\n"),
				maxBytes: 5,
			},
			args: args{
				bs: []byte("123\x00\x014567867\r\n1\r\n2\r\nxyz\x00\x01xyz\r\nxyz"),
			},
			want: &frameResult{
				frames:  [][]byte{[]byte("\x00\x0145678"), []byte("\x00\x01xyz\r\n")},
				remains: []byte{},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			f := &delimiterFramer{
				stxCh:    tt.fields.stxCh,
				endCh:    tt.fields.endCh,
				maxBytes: tt.fields.maxBytes,
			}
			if got := f.Frame(tt.args.bs); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("delimiterFramer.Frame() = %v, want %v", got, tt.want)
			}
		})
	}
}