package encoding

import (
	"bytes"
	"testing"
)

// region encode
func TestGSM7EncodeSimpleASCIIString(t *testing.T) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Sunshine"

	expected := []byte{0b11010011, 0b10111010, 0b01111011, 0b10001110, 0b01001110, 0b10111011, 0b11001011}
	err := coder.Encode(input, &buf)

	if err != nil {
		t.Errorf("Expected no error, but got %v", err)
	}

	if !bytes.Equal(buf.Bytes(), expected) {
		t.Errorf("Expected '%v', but got '%v'", expected, buf.Bytes())
	}
}

func TestGSM7EncodeComplexASCIIString(t *testing.T) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Golden rays play, Chasing night away."

	expected := []byte{0b11000111, 0b00110111, 0b10011011, 0b01011100, 0b01110110, 0b10000011, 0b11100100, 0b11100001, 0b11111100, 0b00011100, 0b00000100, 0b01100111, 0b10000111, 0b11110011, 0b00101100, 0b11010000, 0b00010000, 0b00011101, 0b10011110, 0b10100111, 0b11011101, 0b01100111, 0b10010000, 0b00111011, 0b01111101, 0b01000110, 0b11010011, 0b01000001, 0b11100001, 0b01111011, 0b00111000, 0b11101111, 0b00000010}
	err := coder.Encode(input, &buf)

	if err != nil {
		t.Errorf("Expected no error, but got %v", err)
	}

	if !bytes.Equal(buf.Bytes(), expected) {
		t.Errorf("Expected '%v', but got '%v'", expected, buf.Bytes())
	}
}

func TestGSM7EncodeComplex8nASCIIString(t *testing.T) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Ducks are fucking great, they quacks, O quackers, what the fuck."

	expected := []byte{0b11000100, 0b11111010, 0b01111000, 0b00111101, 0b00000111, 0b10000101, 0b11100101, 0b01100101, 0b10010000, 0b10111001, 0b00111110, 0b01011110, 0b10100111, 0b11011101, 0b01100111, 0b11010000, 0b01011001, 0b01011110, 0b00001110, 0b11010011, 0b01011001, 0b00100000, 0b00111010, 0b10111010, 0b10011100, 0b00000111, 0b11000101, 0b11101011, 0b11100001, 0b11110001, 0b01111010, 0b11001110, 0b00000010, 0b00111101, 0b01000001, 0b11110001, 0b01111010, 0b01111000, 0b10111100, 0b00101110, 0b11001011, 0b11100111, 0b00101100, 0b11010000, 0b00011101, 0b00011101, 0b10100110, 0b10000011, 0b11101000, 0b11101000, 0b00110010, 0b11001000, 0b01011100, 0b00011111, 0b10101111, 0b01011101}
	err := coder.Encode(input, &buf)

	if err != nil {
		t.Errorf("Expected no error, but got %v", err)
	}

	if !bytes.Equal(buf.Bytes(), expected) {
		t.Errorf("Expected '%v', but got '%v'", expected, buf.Bytes())
	}
}

func TestGSM7EncodeEmptyString(t *testing.T) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := ""

	expected := []byte{}
	err := coder.Encode(input, &buf)

	if err != nil {
		t.Errorf("Expected no error, but got %v", err)
	}

	if !bytes.Equal(buf.Bytes(), expected) {
		t.Errorf("Expected '%v', but got '%v'", expected, buf.Bytes())
	}
}

// region decode
func TestGSM7DecodeSimpleASCIIString(t *testing.T) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := []byte{0b11010011, 0b10111010, 0b01111011, 0b10001110, 0b01001110, 0b10111011, 0b11001011}

	expected := "Sunshine"
	buf.Write(input)
	actual, err := coder.Decode(&buf)

	if err != nil {
		t.Errorf("Expected no error, but got %v", err)
	}

	if actual != expected {
		t.Errorf("Expected '%v', but got '%v'", expected, actual)
	}
}

// func TestGSM7DecodeComplexASCIIString(t *testing.T) {
// 	coder := &GSM7Coder{}
// 	var buf bytes.Buffer
// 	input := []byte{0b11000111, 0b00110111, 0b10011011, 0b01011100, 0b01110110, 0b10000011, 0b11100100, 0b11100001, 0b11111100, 0b00011100, 0b00000100, 0b01100111, 0b10000111, 0b11110011, 0b00101100, 0b11010000, 0b00010000, 0b00011101, 0b10011110, 0b10100111, 0b11011101, 0b01100111, 0b10010000, 0b00111011, 0b01111101, 0b01000110, 0b11010011, 0b01000001, 0b11100001, 0b01111011, 0b00111000, 0b11101111, 0b00000010}

// 	expected := "Golden rays play, Chasing night away."
// 	buf.Write(input)
// 	actual, err := coder.Decode(&buf)

// 	if err != nil {
// 		t.Errorf("Expected no error, but got %v", err)
// 	}

// 	if actual != expected {
// 		t.Errorf("Expected '%v', but got '%v'", expected, actual)
// 	}
// }

// func TestGSM7DecodeComplex8nASCIIString(t *testing.T) {
// 	coder := &GSM7Coder{}
// 	var buf bytes.Buffer
// 	input := []byte{0b11000100, 0b11111010, 0b01111000, 0b00111101, 0b00000111, 0b10000101, 0b11100101, 0b01100101, 0b10010000, 0b10111001, 0b00111110, 0b01011110, 0b10100111, 0b11011101, 0b01100111, 0b11010000, 0b01011001, 0b01011110, 0b00001110, 0b11010011, 0b01011001, 0b00100000, 0b00111010, 0b10111010, 0b10011100, 0b00000111, 0b11000101, 0b11101011, 0b11100001, 0b11110001, 0b01111010, 0b11001110, 0b00000010, 0b00111101, 0b01000001, 0b11110001, 0b01111010, 0b01111000, 0b10111100, 0b00101110, 0b11001011, 0b11100111, 0b00101100, 0b11010000, 0b00011101, 0b00011101, 0b10100110, 0b10000011, 0b11101000, 0b11101000, 0b00110010, 0b11001000, 0b01011100, 0b00011111, 0b10101111, 0b01011101}

// 	expected := "Ducks are fucking great, they quacks, O quackers, what the fuck."
// 	buf.Write(input)
// 	actual, err := coder.Decode(&buf)

// 	if err != nil {
// 		t.Errorf("Expected no error, but got %v", err)
// 	}

// 	if actual != expected {
// 		t.Errorf("Expected '%v', but got '%v'", expected, actual)
// 	}
// }

// func TestGSM7DecodeEmptyString(t *testing.T) {
// 	coder := &GSM7Coder{}
// 	buf := bytes.NewBuffer([]byte{})

// 	expected := ""
// 	actual, err := coder.Decode(buf)

// 	if err != nil {
// 		t.Errorf("Expected no error, but got %v", err)
// 	}

// 	if actual != expected {
// 		t.Errorf("Expected '%v', but got '%v'", expected, actual)
// 	}
// }

// region insertat
func TestInsertAtBeginning(t *testing.T) {
    data := []byte{2, 3, 4, 0}
    InsertAt(&data, 0, 1)
    expected := []byte{1, 2, 3, 4}
    if !bytes.Equal(data, expected) {
        t.Errorf("expected %v, got %v", expected, data)
    }
}

func TestInsertInMiddle(t *testing.T) {
	data := []byte{2, 3, 4, 0}
	InsertAt(&data, 1, 5)
	expected := []byte{2, 5, 3, 4}
	if !bytes.Equal(data, expected) {
		t.Errorf("expected %v, got %v", expected, data)
	}
}

func TestInsertAtEnd(t *testing.T) {
	data := []byte{1, 2, 3, 0}
	InsertAt(&data, 3, 4)
	expected := []byte{1, 2, 3, 4}
	if !bytes.Equal(data, expected) {
		t.Errorf("expected %v, got %v", expected, data)
	}
}

func TestIndexOutOfBounds(t *testing.T) {
    data := []byte{2, 3, 4}
    defer func() {
        if r := recover(); r == nil {
            t.Errorf("The code did not panic")
        }
    }()
    InsertAt(&data, 4, 5)
}

func TestNegativeIndex(t *testing.T) {
	data := []byte{2, 3, 4}
	defer func() {
		if r := recover(); r == nil {
			t.Errorf("The code did not panic")
		}
	}()
	InsertAt(&data, -1, 1)
}

func TestMaintainsOrderAfterInsertion(t *testing.T) {
    data := []byte{2, 3, 4, 0}
    InsertAt(&data, 1, 1)
    expected := []byte{2, 1, 3, 4}
    if !bytes.Equal(data, expected) {
        t.Errorf("expected %v, got %v", expected, data)
    }
}

func TestMaintainsLength(t *testing.T) {
	data := []byte{2, 3, 4, 0}
	InsertAt(&data, 1, 1)
	if len(data) != 4 {
		t.Errorf("expected length 4, got %d", len(data))
	}
}

func TestDeletesLastValue(t *testing.T) {
	data := []byte{2, 3, 4, 5}
	InsertAt(&data, 0, 1)
	expected := []byte{1, 2, 3, 4}
	if !bytes.Equal(data, expected) {
		t.Errorf("expected %v, got %v", expected, data)
	}
}

// benchmark
func BenchmarkGSM7EncodeSimpleASCIIString(b *testing.B) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Sunshine"
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		coder.Encode(input, &buf)
	}
}

func BenchmarkGSM7EncodeComplexASCIIString(b *testing.B) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Golden rays play, Chasing night away."
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		coder.Encode(input, &buf)
	}
}

func BenchmarkGSM7EncodeComplex8nASCIIString(b *testing.B) {
	coder := &GSM7Coder{}
	var buf bytes.Buffer
	input := "Ducks are fucking great, they quacks, O quackers, what the fuck."
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		coder.Encode(input, &buf)
	}
}