Function TestLocalKeyStore

crypto/kms/localkeystore_test.go:30–77

(t *testing.T)

)

func TestLocalKeyStore(t *testing.T) {
	Convey("set and get key", t, func() {
		initLocalKeyStore()
		So(localKey, ShouldNotBeNil)
		gotPrivate, err := GetLocalPrivateKey()
		So(gotPrivate, ShouldBeNil)
		So(err, ShouldEqual, ErrNilField)
		gotPublic, err := GetLocalPublicKey()
		So(gotPublic, ShouldBeNil)
		So(err, ShouldEqual, ErrNilField)

		privKey1, pubKey1, _ := asymmetric.GenSecp256k1KeyPair()
		privKey2, pubKey2, _ := asymmetric.GenSecp256k1KeyPair()
		SetLocalKeyPair(privKey1, pubKey1)
		SetLocalKeyPair(privKey2, pubKey2) // no effect
		gotPrivate, err = GetLocalPrivateKey()
		So(err, ShouldBeNil)
		gotPublic, err = GetLocalPublicKey()
		So(err, ShouldBeNil)
		So(bytes.Compare(gotPrivate.Serialize(), privKey1.Serialize()), ShouldBeZeroValue)
		So(gotPublic.IsEqual(pubKey1), ShouldBeTrue)
		So(gotPrivate.PubKey().IsEqual(pubKey1), ShouldBeTrue)
		ResetLocalKeyStore()
	})
	Convey("set and get key", t, func() {
		initLocalKeyStore()
		So(localKey, ShouldNotBeNil)
		gotID, err := GetLocalNodeIDBytes()
		So(gotID, ShouldBeNil)
		So(err, ShouldEqual, ErrNilField)
		gotNonce, err := GetLocalNonce()
		So(gotNonce, ShouldBeNil)
		So(err, ShouldEqual, ErrNilField)
		SetLocalNodeIDNonce([]byte("aaa"), nil)
		SetLocalNodeIDNonce([]byte("aaa"), &mine.Uint256{A: 1, B: 1, C: 1, D: 1})
		gotID, err = GetLocalNodeIDBytes()
		So(bytes.Compare(gotID, []byte("aaa")), ShouldBeZeroValue)
		So(err, ShouldBeNil)
		gotNonce, err = GetLocalNonce()
		So(*gotNonce == mine.Uint256{A: 1, B: 1, C: 1, D: 1}, ShouldBeTrue)
		So(err, ShouldBeNil)
		rawNodeID := &proto.RawNodeID{}
		nodeIDBefore := rawNodeID.ToNodeID()
		SetLocalNodeIDNonce(nodeIDBefore.ToRawNodeID().CloneBytes(), nil)
		nodeID, err := GetLocalNodeID()
		So(nodeID, ShouldResemble, nodeIDBefore)
	})
}