add GetData/SetData/IsSame

1 files changed, 32 insertions, 0 deletions

diff --git a/test/bls_if_test.cpp b/test/bls_if_test.cpp
index ed59396..2292bfa 100644
--- a/test/bls_if_test.cpp
+++ b/test/bls_if_test.cpp
@@ -49,6 +49,37 @@ void bls_if_use_stackTest()
 	CYBOZU_TEST_ASSERT(blsSignVerify(&sign, &pub, msg, msgSize));
 }
 
+void bls_ifDataTest()
+{
+	const char *msg = "test test";
+	const size_t msgSize = strlen(msg);
+	const size_t fpSize = blsGetOpUnitSize() * sizeof(uint64_t);
+	blsSecretKey sec1, sec2;
+	blsSecretKeyInit(&sec1);
+	char buf[BLS_MAX_OP_UNIT_SIZE * sizeof(uint64_t) * 2];
+	size_t n;
+	int ret;
+	n = blsSecretKeyGetData(&sec1, buf, sizeof(buf));
+	CYBOZU_TEST_EQUAL(n, fpSize);
+	ret = blsSecretKeySetData(&sec2, buf, n);
+	CYBOZU_TEST_EQUAL(ret, 0);
+	CYBOZU_TEST_ASSERT(blsSecretKeyIsSame(&sec1, &sec2));
+	blsPublicKey pub1, pub2;
+	blsSecretKeyGetPublicKey(&sec1, &pub1);
+	n = blsPublicKeyGetData(&pub1, buf, sizeof(buf));
+	CYBOZU_TEST_EQUAL(n, fpSize * 2);
+	ret = blsPublicKeySetData(&pub2, buf, n);
+	CYBOZU_TEST_EQUAL(ret, 0);
+	CYBOZU_TEST_ASSERT(blsPublicKeyIsSame(&pub1, &pub2));
+	blsSign sign1, sign2;
+	blsSecretKeySign(&sec1, &sign1, msg, msgSize);
+	n = blsSignGetData(&sign1, buf, sizeof(buf));
+	CYBOZU_TEST_EQUAL(n, fpSize);
+	ret = blsSignSetData(&sign2, buf, n);
+	CYBOZU_TEST_EQUAL(ret, 0);
+	CYBOZU_TEST_ASSERT(blsSignIsSame(&sign1, &sign2));
+}
+
 CYBOZU_TEST_AUTO(all)
 {
 	const int tbl[] = {
@@ -63,5 +94,6 @@ CYBOZU_TEST_AUTO(all)
 		blsInit(tbl[i], BLS_MAX_OP_UNIT_SIZE);
 		bls_ifTest();
 		bls_if_use_stackTest();
+		bls_ifDataTest();
 	}
 }