package qr

import (
	"image/color"
	"testing"
)

func Test_NewQRCode(t *testing.T) {
	bc := newBarcode(2)
	if bc == nil {
		t.Fail()
	}
	if bc.data.Len() != 4 {
		t.Fail()
	}
	if bc.dimension != 2 {
		t.Fail()
	}
}

func Test_QRBasics(t *testing.T) {
	qr := newBarcode(10)
	if qr.ColorModel() != color.Gray16Model {
		t.Fail()
	}
	code, _ := Encode("test", L, Unicode)
	if code.Content() != "test" {
		t.Fail()
	}
	if code.Metadata().Dimensions != 2 {
		t.Fail()
	}
	bounds := code.Bounds()
	if bounds.Min.X != 0 || bounds.Min.Y != 0 || bounds.Max.X != 21 || bounds.Max.Y != 21 {
		t.Fail()
	}
	if code.At(0, 0) != color.Black || code.At(0, 7) != color.White {
		t.Fail()
	}
	qr = code.(*qrcode)
	if !qr.Get(0, 0) || qr.Get(0, 7) {
		t.Fail()
	}
	sum := qr.calcPenaltyRule1() + qr.calcPenaltyRule2() + qr.calcPenaltyRule3() + qr.calcPenaltyRule4()
	if qr.calcPenalty() != sum {
		t.Fail()
	}
}

func Test_Penalty1(t *testing.T) {
	qr := newBarcode(7)
	if qr.calcPenaltyRule1() != 70 {
		t.Fail()
	}
	qr.Set(0, 0, true)
	if qr.calcPenaltyRule1() != 68 {
		t.Fail()
	}
	qr.Set(0, 6, true)
	if qr.calcPenaltyRule1() != 66 {
		t.Fail()
	}
}

func Test_Penalty2(t *testing.T) {
	qr := newBarcode(3)
	if qr.calcPenaltyRule2() != 12 {
		t.Fail()
	}
	qr.Set(0, 0, true)
	qr.Set(1, 1, true)
	qr.Set(2, 0, true)
	if qr.calcPenaltyRule2() != 0 {
		t.Fail()
	}
	qr.Set(1, 1, false)
	if qr.calcPenaltyRule2() != 6 {
		t.Fail()
	}
}

func Test_Penalty3(t *testing.T) {
	runTest := func(content string, result uint) {
		code, _ := Encode(content, L, AlphaNumeric)
		qr := code.(*qrcode)
		if qr.calcPenaltyRule3() != result {
			t.Errorf("Failed Penalty Rule 3 for content \"%s\" got %d but expected %d", content, qr.calcPenaltyRule3(), result)
		}
	}
	runTest("A", 80)
	runTest("FOO", 40)
	runTest("0815", 0)
}

func Test_Penalty4(t *testing.T) {
	qr := newBarcode(3)
	if qr.calcPenaltyRule4() != 100 {
		t.Fail()
	}
	qr.Set(0, 0, true)
	if qr.calcPenaltyRule4() != 70 {
		t.Fail()
	}
	qr.Set(0, 1, true)
	if qr.calcPenaltyRule4() != 50 {
		t.Fail()
	}
	qr.Set(0, 2, true)
	if qr.calcPenaltyRule4() != 30 {
		t.Fail()
	}
	qr.Set(1, 0, true)
	if qr.calcPenaltyRule4() != 10 {
		t.Fail()
	}
	qr.Set(1, 1, true)
	if qr.calcPenaltyRule4() != 10 {
		t.Fail()
	}
	qr = newBarcode(2)
	qr.Set(0, 0, true)
	qr.Set(1, 0, true)
	if qr.calcPenaltyRule4() != 0 {
		t.Fail()
	}
}