use rpki::storage::{RocksStore, VerifiedKey};

#[test]
fn storage_iter_all_lists_raw_and_verified_entries() {
    let temp = tempfile::tempdir().expect("tempdir");
    let store = RocksStore::open(temp.path()).expect("open rocksdb");

    store
        .put_raw("rsync://example.test/repo/a.cer", b"a")
        .expect("put_raw a");
    store
        .put_raw("rsync://example.test/repo/b.roa", b"b")
        .expect("put_raw b");

    let key = VerifiedKey::from_manifest_rsync_uri("rsync://example.test/repo/m.mft");
    store.put_verified(&key, b"x").expect("put_verified");

    let raw_keys = store
        .raw_iter_all()
        .expect("raw_iter_all")
        .map(|(k, _v)| String::from_utf8(k.to_vec()).expect("utf8 key"))
        .collect::<Vec<_>>();
    assert_eq!(raw_keys.len(), 2);
    assert!(raw_keys.contains(&"rsync://example.test/repo/a.cer".to_string()));
    assert!(raw_keys.contains(&"rsync://example.test/repo/b.roa".to_string()));

    let verified_keys = store
        .verified_iter_all()
        .expect("verified_iter_all")
        .map(|(k, _v)| String::from_utf8(k.to_vec()).expect("utf8 key"))
        .collect::<Vec<_>>();
    assert_eq!(verified_keys, vec![key.as_str().to_string()]);
}