use crate::chain::Id as ChainId;
use crate::Error;
use std::convert::TryFrom;
use tendermint_proto::privval::PubKeyRequest as RawPubKeyRequest;
use tendermint_proto::Protobuf;
#[derive(Clone, PartialEq, Debug)]
pub struct PubKeyRequest {
pub chain_id: ChainId,
}
impl Protobuf<RawPubKeyRequest> for PubKeyRequest {}
impl TryFrom<RawPubKeyRequest> for PubKeyRequest {
type Error = Error;
fn try_from(value: RawPubKeyRequest) -> Result<Self, Self::Error> {
Ok(PubKeyRequest {
chain_id: ChainId::try_from(value.chain_id)?,
})
}
}
impl From<PubKeyRequest> for RawPubKeyRequest {
fn from(value: PubKeyRequest) -> Self {
RawPubKeyRequest {
chain_id: value.chain_id.as_str().to_string(),
}
}
}
#[cfg(test)]
mod tests {
use super::PubKeyRequest;
use crate::chain::Id as ChainId;
use std::str::FromStr;
use tendermint_proto::Protobuf;
#[test]
fn test_empty_pubkey_msg() {
let want: Vec<u8> = vec![10, 1, 65];
let msg = PubKeyRequest {
chain_id: ChainId::from_str("A").unwrap(),
};
let mut got = vec![];
let _have = msg.encode(&mut got);
assert_eq!(got, want);
match PubKeyRequest::decode(want.as_ref()) {
Ok(have) => assert_eq!(have, msg),
Err(err) => panic!("{}", err.to_string()),
}
}
}