package plugctl

import (
	"bytes"
	"encoding/base64"
	"io"
	"testing"
)

func TestDecrypt(t *testing.T) {
	// on message from hs100.sh
	in64 := "AAAAKtDygfiL/5r31e+UtsWg1Iv5nPCR6LfEsNGlwOLYo4HyhueT9tTu36Lfog=="
	in, err := base64.StdEncoding.DecodeString(in64)
	if err != nil {
		t.Fatalf("base64.DecodeString(%q) error: %v", in64, err)
	}
	in = in[4:]
	want := []byte(`{"system":{"set_relay_state":{"state":1}}}`)
	dec := newDecrypter(bytes.NewReader(in))
	got, err := io.ReadAll(dec)
	if err != nil {
		t.Fatalf("io.ReadAll(decryptReader) error: %v", err)
	}
	if !bytes.Equal(got, want) {
		t.Errorf("decrypt(%02x) = %q, want %q", in, got, want)
	}
}

func TestEncrypt(t *testing.T) {
	// on message from hs100.sh
	want64 := "AAAAKtDygfiL/5r31e+UtsWg1Iv5nPCR6LfEsNGlwOLYo4HyhueT9tTu36Lfog=="
	want, err := base64.StdEncoding.DecodeString(want64)
	if err != nil {
		t.Fatalf("base64.DecodeString(%q) error: %v", want64, err)
	}
	want = want[4:]
	in := []byte(`{"system":{"set_relay_state":{"state":1}}}`)
	var buf bytes.Buffer
	newEncrypter(&buf).Write(in)

	if got := buf.Bytes(); !bytes.Equal(got, want) {
		t.Errorf("encrypt(%q) = %02x, want %02x", in, got, want)
	}
}

// func TestEncryptDecrypt(t *testing.T) {
// 	want := make([]byte, 64)
// 	if _, err := rand.Read(want); err != nil {
// 		t.Fatalf("rand.Read(want) error: %v", err)
// 	}
// 	if got := decrypt(encrypt(want)); !bytes.Equal(got, want) {
// 		t.Errorf("decrypt(encrypt(%02x)) = %02x, want %02x", want, got, want)
// 	}
// }