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{0b11010011, 0b10111010, 0b01111011, 0b10001110, 0b01001110, 0b10111011, 0b11001011}

	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)
	}
}