use crate::*;
use hex_literal::hex;
use k256::SecretKey;
use std::str::FromStr;

// Pulled from: https://github.com/satoshilabs/slips/blob/master/slip-0010.md

#[test]
fn example() {
    // seed, chain, chain code, private, public
    let tests = [(
        &hex!("000102030405060708090a0b0c0d0e0f")[..],
        DerivationPath::from_str("m").unwrap(),
        hex!("873dff81c02f525623fd1fe5167eac3a55a049de3d314bb42ee227ffed37d508"),
        hex!("e8f32e723decf4051aefac8e2c93c9c5b214313817cdb01a1494b917c8436b35"),
        hex!("0339a36013301597daef41fbe593a02cc513d0b55527ec2df1050e2e8ff49c85c2"),
    ), (
        &hex!("000102030405060708090a0b0c0d0e0f")[..],
        DerivationPath::from_str("m/0'").unwrap(),
        hex!("47fdacbd0f1097043b78c63c20c34ef4ed9a111d980047ad16282c7ae6236141"),
        hex!("edb2e14f9ee77d26dd93b4ecede8d16ed408ce149b6cd80b0715a2d911a0afea"),
        hex!("035a784662a4a20a65bf6aab9ae98a6c068a81c52e4b032c0fb5400c706cfccc56"),
    ), (
        &hex!("000102030405060708090a0b0c0d0e0f")[..],
        DerivationPath::from_str("m/0'/1").unwrap(),
        hex!("2a7857631386ba23dacac34180dd1983734e444fdbf774041578e9b6adb37c19"),
        hex!("3c6cb8d0f6a264c91ea8b5030fadaa8e538b020f0a387421a12de9319dc93368"),
        hex!("03501e454bf00751f24b1b489aa925215d66af2234e3891c3b21a52bedb3cd711c"),
    ), (
        &hex!("000102030405060708090a0b0c0d0e0f")[..],
        DerivationPath::from_str("m/0'/1/2'").unwrap(),
        hex!("04466b9cc8e161e966409ca52986c584f07e9dc81f735db683c3ff6ec7b1503f"),
        hex!("cbce0d719ecf7431d88e6a89fa1483e02e35092af60c042b1df2ff59fa424dca"),
        hex!("0357bfe1e341d01c69fe5654309956cbea516822fba8a601743a012a7896ee8dc2"),
    ), (
        &hex!("000102030405060708090a0b0c0d0e0f")[..],
        DerivationPath::from_str("m/0'/1/2'/2").unwrap(),
        hex!("cfb71883f01676f587d023cc53a35bc7f88f724b1f8c2892ac1275ac822a3edd"),
        hex!("0f479245fb19a38a1954c5c7c0ebab2f9bdfd96a17563ef28a6a4b1a2a764ef4"),
        hex!("02e8445082a72f29b75ca48748a914df60622a609cacfce8ed0e35804560741d29"),
    ), (
        &hex!("fffcf9f6f3f0edeae7e4e1dedbd8d5d2cfccc9c6c3c0bdbab7b4b1aeaba8a5a29f9c999693908d8a8784817e7b7875726f6c696663605d5a5754514e4b484542")[..],
        DerivationPath::from_str("m").unwrap(),
        hex!("60499f801b896d83179a4374aeb7822aaeaceaa0db1f85ee3e904c4defbd9689"),
        hex!("4b03d6fc340455b363f51020ad3ecca4f0850280cf436c70c727923f6db46c3e"),
        hex!("03cbcaa9c98c877a26977d00825c956a238e8dddfbd322cce4f74b0b5bd6ace4a7"),
    ), (
        &hex!("fffcf9f6f3f0edeae7e4e1dedbd8d5d2cfccc9c6c3c0bdbab7b4b1aeaba8a5a29f9c999693908d8a8784817e7b7875726f6c696663605d5a5754514e4b484542")[..],
        DerivationPath::from_str("m/0").unwrap(),
        hex!("f0909affaa7ee7abe5dd4e100598d4dc53cd709d5a5c2cac40e7412f232f7c9c"),
        hex!("abe74a98f6c7eabee0428f53798f0ab8aa1bd37873999041703c742f15ac7e1e"),
        hex!("02fc9e5af0ac8d9b3cecfe2a888e2117ba3d089d8585886c9c826b6b22a98d12ea"),
    ), (
        &hex!("fffcf9f6f3f0edeae7e4e1dedbd8d5d2cfccc9c6c3c0bdbab7b4b1aeaba8a5a29f9c999693908d8a8784817e7b7875726f6c696663605d5a5754514e4b484542")[..],
        DerivationPath::from_str("m/0/2147483647'/1/2147483646'/2").unwrap(),
        hex!("9452b549be8cea3ecb7a84bec10dcfd94afe4d129ebfd3b3cb58eedf394ed271"),
        hex!("bb7d39bdb83ecf58f2fd82b6d918341cbef428661ef01ab97c28a4842125ac23"),
        hex!("024d902e1a2fc7a8755ab5b694c575fce742c48d9ff192e63df5193e4c7afe1f9c"),
    )];
    for (seed, chain, chain_code, private_key, public_key) in tests {
        let xkey = ExtendedPrivateKey::<SecretKey>::new(seed).unwrap();
        let derived_key = xkey.derive_path(&chain).unwrap();
        assert_eq!(derived_key.chain_code, chain_code);
        assert_eq!(derived_key.private_key.to_bytes().as_slice(), private_key);
        assert_eq!(derived_key.public_key().to_bytes(), public_key);
    }
}