package nizk

import (
	"testing"

	. "kesim.org/seal/common"
)

func TestStage2Simple(t *testing.T) {
	id1 := Curve.RandomScalar()
	id2 := Curve.RandomScalar()
	b1 := NewBit(id1, true)
	b2 := NewBit(id2, true)

	s1, c1 := b1.Stage()
	r1, _, _ := s1.RevealStage1()

	bc2, _ := b2.Commit()
	s2, c2 := b2.Stage()
	r2, p2, e := s2.RevealStage2(true, s1)
	if e != nil {
		t.Fatalf("e: %v", e)
	}
	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)
	}
}