package nizk

import (
	"testing"

	. "kesim.org/seal/common"
)

func TestStage2Simple(t *testing.T) {
	id := Curve.RandomScalar()
	b1, _, _ := NewBit(id, false) // This is also the junction
	r1, _ := b1.RevealStage1()
	c1 := b1.StageCommit()

	for _, s := range [][2]bool{
		{false, false},
		{true, false},
		{true, true},
		{false, true},
	} {
		b2, bc2, _ := NewBit(id, s[0])
		b3, bc3, _ := NewBit(id, s[1])

		c2 := b2.StageCommit()
		c3 := b3.StageCommit()

		r2, p2 := b2.RevealStage2(true, b1)
		if !bc2.VerifyStage2(c1, c2, r1, r2, p2) {
			t.Fatalf("failed to verify!\nbc2: %#v\nc1: %#v\nc2: %#v\nr1: %#v\nr2: %#v\np2: %#v\n",
				bc2, c1, c2, r1, r2, p2)
		}

		r3, p3 := b3.RevealStage2(true, b1)
		if !bc3.VerifyStage2(c1, c3, r1, r3, p3) {
			t.Fatalf("faild to verify bc3")
		}
	}

}