use beserial::{Deserialize, Serialize};
use nimiq_messages::*;

const VERSION_MESSAGE: &str = "42042042000000010ee4e19ae300000001040000000400000167aaa7c40d02a84eaf654fe5f3b0bb45d0dd9a70c78fc24d134f5e302aa8270ea107752a6b860053e4c4966637a7de44500e8df82d7b541f578ab25a9e147fed9066361081826337f5511fa27762ecd0e328488e48bcbc4c6e2ded7b552039832768e4f137d809096c6f63616c686f737420fb264aaf8a4f9828a76c550635da078eb466306a189fcc03710bee9f649c869d12c6efcae1d34d135ff562bd75a62ffbcaab81f578ad23da8a02ccf59c7f8b6baa97fabe9dbd9db0acb5e1539bf3155ca1c9565f3363c5c8f1e1cc5b99ba3902c921636f72652d6a732f312e342e3120286e6f64656a733b204c696e75782078363429";
const INV_MESSAGE: &str = "42042042010000007b268c0610000300000002324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf00000002b8b37c1d034e371c7a3b834f9476a746eb62259ff9558ab715b4bff79ebf58e100000001f823f66ba1026e7f711ea5aa4719837bb378fc615b50516b8dabdaff78e8168e";
const GET_DATA_MESSAGE: &str = "42042042020000007b990afcc2000300000002324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf00000002b8b37c1d034e371c7a3b834f9476a746eb62259ff9558ab715b4bff79ebf58e100000001f823f66ba1026e7f711ea5aa4719837bb378fc615b50516b8dabdaff78e8168e";
const BLOCK_MESSAGE: &str = "4204204206000000ba2002774c0001000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000009d5b7130fdd19427406f1d477788ec1f866c650b2eff550afb3505b1435681bbbfacd81c643767f3bf1bf642c97b4efd33daeff2b0d341c613344ae706eedb381f010000000000010000000000018d5800011be440919634a6fe3ba5f8a7181fe4bb8212c13c0000000000";
const HEADER_MESSAGE: &str = "42042042070000009fc8a9c2ed0001000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000009d5b7130fdd19427406f1d477788ec1f866c650b2eff550afb3505b1435681bbbfacd81c643767f3bf1bf642c97b4efd33daeff2b0d341c613344ae706eedb381f010000000000010000000000018d58";
const TX_MESSAGE: &str = "42042042080000009803ad3340008f30a4d938d4130d1a3396dede1505c72b7f75ac9f9b80d1ad7e368b39f3b10591e9240f415223982edc345532630710e94a7f5200000000000022b8000000000000002a0000000004e5564300c360ac729086e2cc806e828a84877f1eb8e5d974d873e065224901555fb8821590a33bacc61e39701cf9b46bd25bf5f0595bbe24655141438e7a100b00";
const GET_BLOCKS_MESSAGE: &str = "420420420500000012f392d9600000000402";
const MEMPOOL_MESSAGE: &str = "42042042090000000d994373bd";
const REJECT_MESSAGE: &str = "420420420a000000194422360c004104746573740003616263";
const SUBSCRIBE_MESSAGE: &str = "420420420b000000601dc2dcab02000491e9240f415223982edc345532630710e94a7f5287298cc2f31fba73181ea2a9e6ef10dce21ed95e47ea70cf08872bdb4afad3432b01d963ac7d165f5d1c3122ada85138a67dfc15267cbeb21dd36041";
const ADDR_MESSAGE: &str = "4204204214000000f5650e831a00020000000000000000000000000000080808088f30a4d938d4130d1a3396dede1505c72b7f75ac9f9b80d1ad7e368b39f3b10500e5564300c360ac729086e2cc806e828a84877f1eb8e5d974d873e065224901555fb8821590a33bacc61e39701cf9b46bd25bf5f0595bbe24655141438e7a100b0000000000000000000000000000080808088f30a4d938d4130d1a3396dede1505c72b7f75ac9f9b80d1ad7e368b39f3b10500e5564300c360ac729086e2cc806e828a84877f1eb8e5d974d873e065224901555fb8821590a33bacc61e39701cf9b46bd25bf5f0595bbe24655141438e7a100b";
const GET_ADDR_MESSAGE: &str = "420420421500000014c09a093a02000000040008";
const PING_MESSAGE: &str = "420420421600000011fde10bd200000002";
const PONG_MESSAGE: &str = "4204204217000000112077d25700000002";
const SIGNAL_MESSAGE: &str = "420420421e000002ca8034eb3a6a3641f9e7fbf89ad46a3e5556d64050b795353533f3d27b671850d2292ddf637733552d030002357b2274797065223a226f66666572222c22736470223a22763d305c725c6e6f3d6d6f7a696c6c612e2e2e544849535f49535f534450415254412d36352e302036343333303332313237353230393538363335203020494e2049503420302e302e302e305c725c6e733d2d5c725c6e743d3020305c725c6e613d73656e64726563765c725c6e613d66696e6765727072696e743a7368612d3235362034453a36323a36463a37353a43443a38463a44313a37423a43383a41323a41363a39383a41453a32463a36303a42433a41463a41313a36313a31383a33373a30373a37333a41463a37303a41463a43363a37333a36383a44323a41393a31305c725c6e613d67726f75703a42554e444c4520305c725c6e613d6963652d6f7074696f6e733a747269636b6c655c725c6e613d6d7369642d73656d616e7469633a574d53202a5c725c6e6d3d6170706c69636174696f6e2039205544502f44544c532f53435450207765627274632d646174616368616e6e656c5c725c6e633d494e2049503420302e302e302e305c725c6e613d73656e64726563765c725c6e613d6963652d7077643a65366461626166316138373236616137643134386164656461326264656361655c725c6e613d6963652d75667261673a37613761303661345c725c6e613d6d69643a305c725c6e613d73657475703a616374706173735c725c6e613d736374702d706f72743a353030305c725c6e613d6d61782d6d6573736167652d73697a653a313037333734313832335c725c6e227dcb6e76297dce489dd3118c37e8523b77482eec35ba544b0f0fdb2a76902636946b6dbbca605acf3de470c88cc3e1ae70772222205d7a9731e5f529057bdb3ac9e462b4fb9a22da499653128c67d95dad232a0ff9982e4eb6f9ea63ade5261f0e";
const GET_CHAIN_PROOF_MESSAGE: &str = "42042042280000000d0cc9e55d";
const CHAIN_PROOF_MESSAGE: &str = "4204204229000000a54744732f00010001000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000009d5b7130fdd19427406f1d477788ec1f866c650b2eff550afb3505b1435681bbbfacd81c643767f3bf1bf642c97b4efd33daeff2b0d341c613344ae706eedb381f010000000000010000000000018d5800000000";
const GET_ACCOUNTS_PROOF_MESSAGE: &str = "420420422a000000570169b3b4324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf000291e9240f415223982edc345532630710e94a7f5287298cc2f31fba73181ea2a9e6ef10dce21ed95e";
const ACCOUNTS_PROOF_MESSAGE_WPROOF: &str = "420420422b00000269694947d7324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf010007ff2830303131313131313131313131313131313131313131313131313131313131313131313131313131000000000000000019ff283030323030303030303030303030303030303030303030303030303030303030303030303030303000000000000000052aff283030323232323232323232323232323232323232323232323232323232323232323232323232323200000000000000005d00033030320225303030303030303030303030303030303030303030303030303030303030303030303030304f4c455662cdbcb30bcc2719dc7184cc30aa01ece81e5a8070f0a2e2f37f5f6b2532323232323232323232323232323232323232323232323232323232323232323232323232d2d93f95e11fc480f2cddb6b3a38e457e24bce6df5c5f470b2501acd5d61e12bff283030333333333333333333333333333333333333333333333333333333333333333333333333333300000000000000000100023030032631313131313131313131313131313131313131313131313131313131313131313131313131314a222c2f95a2f733d51ddea9ea7f6f0bc8f66f7ff4a3c7cc26ae99c4dcc5e28a0132b4e74f0d8054d535e1cb108fab3772a4e19729777b51d4640665a2da6ec7588e263333333333333333333333333333333333333333333333333333333333333333333333333333516c248e6df72f9ffdcc6be8bda172077a8f2de2956bf54d21cffdb77d4c8b02000001023030de17952823fba92b774b125dd808c270dccf1fe3b9c80b1050516ece11c787bb";
const ACCOUNTS_PROOF_MESSAGE_WOPROOF: &str = "420420422b0000002e3fb4ef97324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf00";
const GET_ACCOUNTS_TREE_CHUNK_MESSAGE: &str = "420420422c00000030078e5b06324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf026161";
const ACCOUNTS_TREE_CHUNK_MESSAGE_WCHUNK: &str = "420420422d000003040452ff83324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf010003ff2830303131313131313131313131313131313131313131313131313131313131313131313131313131000000000000000019ff2830303333333333333333333333333333333333333333333333333333333333333333333333333333000000000000000001ff283030323030303030303030303030303030303030303030303030303030303030303030303030303000000000000000052a0007ff2830303131313131313131313131313131313131313131313131313131313131313131313131313131000000000000000019ff283030323030303030303030303030303030303030303030303030303030303030303030303030303000000000000000052aff283030323232323232323232323232323232323232323232323232323232323232323232323232323200000000000000005d00033030320225303030303030303030303030303030303030303030303030303030303030303030303030304f4c455662cdbcb30bcc2719dc7184cc30aa01ece81e5a8070f0a2e2f37f5f6b2532323232323232323232323232323232323232323232323232323232323232323232323232d2d93f95e11fc480f2cddb6b3a38e457e24bce6df5c5f470b2501acd5d61e12bff283030333333333333333333333333333333333333333333333333333333333333333333333333333300000000000000000100023030032631313131313131313131313131313131313131313131313131313131313131313131313131314a222c2f95a2f733d51ddea9ea7f6f0bc8f66f7ff4a3c7cc26ae99c4dcc5e28a0132b4e74f0d8054d535e1cb108fab3772a4e19729777b51d4640665a2da6ec7588e263333333333333333333333333333333333333333333333333333333333333333333333333333516c248e6df72f9ffdcc6be8bda172077a8f2de2956bf54d21cffdb77d4c8b02000001023030de17952823fba92b774b125dd808c270dccf1fe3b9c80b1050516ece11c787bb";
const ACCOUNTS_TREE_CHUNK_MESSAGE_WOCHUNK: &str = "420420422d0000002e4fc3f971324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf00";
const GET_TRANSACTIONS_PROOF_MESSAGE: &str = "420420422f000000571ee934b2324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cf000291e9240f415223982edc345532630710e94a7f5287298cc2f31fba73181ea2a9e6ef10dce21ed95e";
const TRANSACTIONS_PROOF_MESSAGE: &str = "420420423000000182f5f51fe9f9347cc2f81185fd812553d4c4291b8238c26662290c6455ed5f7c30d11fc2c20100010036532fa4cfc4fd660cd9a7a47a167f879f70fc035faced450d09dabc9d47883f7bc82251b2cf9e0da7051f019bbc776da86e89c000000000000077ec000000000000008a0006b4362a5f5abecd5f1dec75d61c81f37605e85f57b993a54bcc21d973a3dd7e75531e4c0ae91b89528f16c1d141ee3b82dd079a795f135e34d01f63be6a32ff7b246506000d40222a020006591c95f4c6ccb08db78234d76aaacdc681afb5f145fbe7c37f0adcc353e42b01181e78ee20c210f6109549e32abc8855102c6d68f087c9a4a132575fbbf8b30fc7e36e4e6c9d36c1b0420090370ae55ce5c781c7029c202f22d8c6afe0125ea14994103db5ca1ceed184b6a4ace228a6f5a727338b043eef2026eff9bc9178494606874a59eafd9cc5384bbbd084f0dcce1d59be6371d4708ac5727d218c409b0f494558ffdddd6004502f0bc7c3555a76e953426a2fa2488b9367f0d49a6e71";
const GET_TRANSACTION_RECEIPTS_MESSAGE: &str = "42042042310000002525dc96f691e9240f415223982edc345532630710e94a7f5200000000";
const GET_BLOCK_PROOF_MESSAGE: &str = "42042042330000004d3f619793324dcf027dd4a30a932c441f365a25e86b173defa4b8e58948253471b81b72cfb8b37c1d034e371c7a3b834f9476a746eb62259ff9558ab715b4bff79ebf58e1";


static MESSAGES: &'static [&str] = &[
    VERSION_MESSAGE,
    INV_MESSAGE,
    GET_DATA_MESSAGE,
    BLOCK_MESSAGE,
    HEADER_MESSAGE,
    TX_MESSAGE,
    GET_BLOCKS_MESSAGE,
    MEMPOOL_MESSAGE,
    REJECT_MESSAGE,
    ADDR_MESSAGE,
    GET_ADDR_MESSAGE,
    PING_MESSAGE,
    PONG_MESSAGE,
//    SUBSCRIBE_MESSAGE, // FIXME: HashSets don't preserve insertion order and addresses on SubscribeMsg are stored in a HashSet
    GET_CHAIN_PROOF_MESSAGE,
    CHAIN_PROOF_MESSAGE,
    ACCOUNTS_PROOF_MESSAGE_WPROOF,
    ACCOUNTS_PROOF_MESSAGE_WOPROOF,
    GET_ACCOUNTS_TREE_CHUNK_MESSAGE,
    ACCOUNTS_TREE_CHUNK_MESSAGE_WCHUNK,
    ACCOUNTS_TREE_CHUNK_MESSAGE_WOCHUNK,
    GET_TRANSACTIONS_PROOF_MESSAGE,
    TRANSACTIONS_PROOF_MESSAGE,
    GET_TRANSACTION_RECEIPTS_MESSAGE,
    GET_BLOCK_PROOF_MESSAGE,
//    BLOCK_PROOF_MESSAGE,
    SIGNAL_MESSAGE
];

#[test]
fn parse_version_message() {
    let vec = ::hex::decode(VERSION_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Version(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_inv_message() {
    let vec = ::hex::decode(INV_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Inv(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_data_message() {
    let vec = ::hex::decode(GET_DATA_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetData(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_block_message() {
    let vec = ::hex::decode(BLOCK_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Block(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_header_message() {
    let vec = ::hex::decode(HEADER_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Header(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_tx_message() {
    let vec = ::hex::decode(TX_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Tx(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_blocks_message() {
    let vec = ::hex::decode(GET_BLOCKS_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetBlocks(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_mempool_message() {
    let vec = ::hex::decode(MEMPOOL_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Mempool => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_reject_message() {
    let vec = ::hex::decode(REJECT_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Reject(_) => assert!(true), _ => assert!(false) };
}

//#[test]
//fn parse_subscribe_message() {
//    let hex_msg = "420420420b000000382a2e67c102000291e9240f415223982edc345532630710e94a7f5287298cc2f31fba73181ea2a9e6ef10dce21ed95e";
//    let vec = ::hex::decode(hex_msg).unwrap();
//    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
//    match message { Message::Subscribe(_) => assert!(true), _ => assert!(false) };
//}

#[test]
fn parse_addr_message() {
    let vec = ::hex::decode(ADDR_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Addr(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_addr_message() {
    let vec = ::hex::decode(GET_ADDR_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetAddr(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_ping_message() {
    let vec = ::hex::decode(PING_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Ping(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_pong_message() {
    let vec = ::hex::decode(PONG_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Pong(_) => assert!(true), _ => assert!(false) };
}


#[test]
fn parse_subscribe_message() {
    let vec = ::hex::decode(SUBSCRIBE_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Subscribe(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_chain_proof_message() {
    let vec = ::hex::decode(GET_CHAIN_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetChainProof => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_chain_proof_message() {
    let vec = ::hex::decode(CHAIN_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::ChainProof(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_accounts_proof_message() {
    let vec = ::hex::decode(GET_ACCOUNTS_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::GetAccountsProof(get_accounts_proof_message) => {
            assert_eq!(get_accounts_proof_message.addresses.len(), 2);
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_accounts_proof_message_wproof() {
    let vec = ::hex::decode(ACCOUNTS_PROOF_MESSAGE_WPROOF).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::AccountsProof(accounts_proof_message) => {
            assert!(accounts_proof_message.proof.is_some());
            assert!(accounts_proof_message.proof.unwrap().verify());
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_accounts_proof_message_woproof() {
    let vec = ::hex::decode(ACCOUNTS_PROOF_MESSAGE_WOPROOF).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::AccountsProof(accounts_proof_message) => {
            assert!(accounts_proof_message.proof.is_none());
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_accounts_tree_chunk_message_woproof() {
    let vec = ::hex::decode(GET_ACCOUNTS_TREE_CHUNK_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::GetAccountsTreeChunk(get_accounts_tree_chunk_message) => {
            assert!(get_accounts_tree_chunk_message.start_prefix == "aa");
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_accounts_tree_chunk_message_wchunk() {
    let vec = ::hex::decode(ACCOUNTS_TREE_CHUNK_MESSAGE_WCHUNK).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::AccountsTreeChunk(accounts_tree_chunk_message) => {
            assert!(accounts_tree_chunk_message.chunk.is_some());
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_accounts_tree_chunk_message_wochunk() {
    let vec = ::hex::decode(ACCOUNTS_TREE_CHUNK_MESSAGE_WOCHUNK).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::AccountsTreeChunk(accounts_tree_chunk_message) => {
            assert!(accounts_tree_chunk_message.chunk.is_none());
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_get_transactions_proof_message() {
    let vec = ::hex::decode(GET_TRANSACTIONS_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::GetTransactionsProof(get_transactions_proof_message) => {
            assert_eq!(get_transactions_proof_message.addresses.len(), 2);
        },
        _ => assert!(false)
    };
}

#[test]
fn parse_transactions_proof_message() {
    let vec = ::hex::decode(TRANSACTIONS_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message {
        Message::TransactionsProof(_) => assert!(true),
        _ => assert!(false)
    };
}

#[test]
fn parse_get_transaction_receipts_message() {
    let vec = ::hex::decode(GET_TRANSACTION_RECEIPTS_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetTransactionReceipts(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_get_block_proof() {
    let vec = ::hex::decode(GET_BLOCK_PROOF_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::GetBlockProof(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn parse_signal() {
    let vec = ::hex::decode(SIGNAL_MESSAGE).unwrap();
    let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
    match message { Message::Signal(_) => assert!(true), _ => assert!(false) };
}

#[test]
fn reserialize_messages() {
    for message in MESSAGES.iter() {
        let vec = ::hex::decode(message).unwrap();
        let message: Message = Deserialize::deserialize(&mut &vec[..]).unwrap();
        assert!(message.serialize_to_vec() == vec);
    }
}