native_db 0.8.2

// TODO: refactor and move to query/ folder

use itertools::Itertools;
use native_db::*;
use native_model::{native_model, Model};
use serde::{Deserialize, Serialize};
use shortcut_assert_fs::TmpFs;

#[derive(Serialize, Deserialize, Eq, PartialEq, Debug, Clone)]
#[native_model(id = 1, version = 1)]
#[native_db(
    primary_key(generate_my_primary_key -> u32),
    secondary_key(secondary_key_1 -> String, unique),
    secondary_key(secondary_key_2 -> String, unique)
)]
struct Item {
    id: u32,
    name: String,
}

impl Item {
    pub fn new(id: u32, name: &str) -> Self {
        Self {
            id,
            name: name.to_string(),
        }
    }
    pub fn generate_my_primary_key(&self) -> u32 {
        self.id
    }
    pub fn secondary_key_1(&self) -> String {
        format!("{}", self.id)
    }
    pub fn secondary_key_2(&self) -> String {
        self.name.to_string()
    }
}

#[test]
fn test_iter() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let result: Vec<Item> = r
        .scan()
        .primary()
        .unwrap()
        .all()
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 2);
    assert_eq!(obj2.name, "test2");
}

// Check if the use of BigEndian is correct
#[test]
fn test_iter_many_items_to_be_bytes() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    // Insert 1000 items
    for i in 0..257 {
        rw.insert(Item::new(i, format!("test_{}", i).as_str()))
            .unwrap();
    }
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    {
        let iter: Vec<Item> = r
            .scan()
            .primary()
            .unwrap()
            .all()
            .unwrap()
            .try_collect()
            .unwrap();
        assert_eq!(iter.len(), 257);

        let obj1 = iter.first().unwrap();
        assert_eq!(obj1.id, 0);
        assert_eq!(obj1.name, "test_0");

        let obj2 = iter.get(1).unwrap();
        assert_eq!(obj2.id, 1);
        assert_eq!(obj2.name, "test_1");

        let obj3 = iter.get(256).unwrap();
        assert_eq!(obj3.id, 256);
        assert_eq!(obj3.name, "test_256");
    }
}

#[test]
fn test_double_ended_iter() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let scan = r.scan().primary().unwrap();
    let iter = scan.all().unwrap();
    let result: Vec<Item> = iter.rev().try_collect().unwrap();

    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();

    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 1);
    assert_eq!(obj2.name, "test");
}

#[test]
fn test_iter_range() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.insert(Item::new(3, "test3")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let result: Vec<Item> = r
        .scan()
        .primary()
        .unwrap()
        .range(..2u32)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let result: Vec<Item> = r
        .scan()
        .primary()
        .unwrap()
        .range(2u32..)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 3);
    assert_eq!(obj2.name, "test3");

    let result: Vec<Item> = r
        .scan()
        .primary()
        .unwrap()
        .range(2u32..3u32)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");
}

#[test]
fn test_iter_by_key() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let result: Vec<Item> = r
        .scan()
        .secondary(ItemKey::secondary_key_1)
        .unwrap()
        .all()
        .unwrap()
        .try_collect()
        .unwrap();

    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 2);
    assert_eq!(obj2.name, "test2");
}

#[test]
fn test_double_ended_iter_by_key() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let scan = r.scan().secondary(ItemKey::secondary_key_1).unwrap();
    let iter = scan.all().unwrap();
    let result: Vec<Item> = iter.rev().try_collect().unwrap();

    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 1);
    assert_eq!(obj2.name, "test");
}

#[test]
fn test_double_ended_iter_by_key_range() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.insert(Item::new(3, "test3")).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let scan = r.scan().secondary(ItemKey::secondary_key_1).unwrap();
    let iter = scan.range(.."2").unwrap();
    let result: Vec<Item> = iter.rev().try_collect().unwrap();

    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let scan = r.scan().secondary(ItemKey::secondary_key_1).unwrap();
    let iter = scan.range("2"..).unwrap();
    let result: Vec<Item> = iter.rev().try_collect().unwrap();
    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 3);
    assert_eq!(obj1.name, "test3");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 2);
    assert_eq!(obj2.name, "test2");

    let scan = r.scan().secondary(ItemKey::secondary_key_1).unwrap();
    let iter = scan.range("2".."3").unwrap();
    let result: Vec<Item> = iter.rev().try_collect().unwrap();

    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");
}

#[derive(Serialize, Deserialize, Eq, PartialEq, Debug, Clone)]
#[native_model(id = 2, version = 1)]
#[native_db(primary_key(generate_my_primary_key -> String))]
struct ItemFlag {
    name: String,
}

impl ItemFlag {
    pub fn new(name: &str) -> Self {
        Self {
            name: name.to_string(),
        }
    }

    pub fn generate_my_primary_key(&self) -> String {
        self.name.to_string()
    }
}

#[test]
fn test_start_with_scenario() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<ItemFlag>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    // Red flag
    rw.insert(ItemFlag::new("red:1")).unwrap();
    rw.insert(ItemFlag::new("red:2")).unwrap();
    rw.insert(ItemFlag::new("red:3")).unwrap();
    // Blue flag
    rw.insert(ItemFlag::new("blue:1")).unwrap();
    rw.insert(ItemFlag::new("blue:2")).unwrap();
    rw.insert(ItemFlag::new("blue:3")).unwrap();
    // Green flag
    rw.insert(ItemFlag::new("green:1")).unwrap();
    rw.insert(ItemFlag::new("green:2")).unwrap();
    rw.insert(ItemFlag::new("green:3")).unwrap();
    rw.commit().unwrap();

    for p in ["red:", "blue:", "green:"] {
        let r = db.r_transaction().unwrap();

        let result: Vec<ItemFlag> = r
            .scan()
            .primary()
            .unwrap()
            .start_with(p.to_string().as_str())
            .unwrap()
            .try_collect()
            .unwrap();
        assert_eq!(result.len(), 3);

        let obj1 = result.first().unwrap();
        assert_eq!(obj1.name, format!("{}1", p));

        let obj2 = result.get(1).unwrap();
        assert_eq!(obj2.name, format!("{}2", p));

        let obj3 = result.get(2).unwrap();
        assert_eq!(obj3.name, format!("{}3", p));
    }
}

#[derive(Serialize, Deserialize, Eq, PartialEq, Debug, Clone)]
#[native_model(id = 3, version = 1)]
#[native_db(primary_key(generate_my_primary_key -> String), secondary_key(flag -> String, unique))]
struct ItemIdFlag {
    id: String,
    flag: String,
}

impl ItemIdFlag {
    pub fn new(id: &str, name: &str) -> Self {
        Self {
            id: id.to_string(),
            flag: name.to_string(),
        }
    }

    pub fn generate_my_primary_key(&self) -> String {
        self.id.clone()
    }
    pub fn flag(&self) -> String {
        format!("{}:{}", self.flag, self.id)
    }
}

#[test]
fn test_start_with_by_key_scenario_write_txn() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<ItemIdFlag>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();

    // Red flag
    rw.insert(ItemIdFlag::new("1", "red")).unwrap();
    rw.insert(ItemIdFlag::new("2", "red")).unwrap();
    rw.insert(ItemIdFlag::new("3", "red")).unwrap();
    // Blue flag
    rw.insert(ItemIdFlag::new("4", "blue")).unwrap();
    rw.insert(ItemIdFlag::new("5", "blue")).unwrap();
    rw.insert(ItemIdFlag::new("6", "blue")).unwrap();
    // Green flag
    rw.insert(ItemIdFlag::new("7", "green")).unwrap();
    rw.insert(ItemIdFlag::new("8", "green")).unwrap();
    rw.insert(ItemIdFlag::new("9", "green")).unwrap();

    rw.commit().unwrap();

    for p in ["red:", "blue:", "green:"] {
        let rw = db.rw_transaction().unwrap();

        let result: Vec<ItemIdFlag> = rw
            .scan()
            .secondary(ItemIdFlagKey::flag)
            .unwrap()
            .start_with(p.to_string().as_str())
            .unwrap()
            .try_collect()
            .unwrap();
        assert_eq!(result.len(), 3);

        let obj1 = result.first().unwrap();
        assert_eq!(format!("{}:", obj1.flag), format!("{}", p));

        let obj2 = result.get(1).unwrap();
        assert_eq!(format!("{}:", obj2.flag), format!("{}", p));

        let obj3 = result.get(2).unwrap();
        assert_eq!(format!("{}:", obj3.flag), format!("{}", p));
    }
}

#[test]
fn test_start_with_by_key_scenario_readonly_txn() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<ItemIdFlag>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    // Red flag
    rw.insert(ItemIdFlag::new("1", "red")).unwrap();
    rw.insert(ItemIdFlag::new("2", "red")).unwrap();
    rw.insert(ItemIdFlag::new("3", "red")).unwrap();
    // Blue flag
    rw.insert(ItemIdFlag::new("4", "blue")).unwrap();
    rw.insert(ItemIdFlag::new("5", "blue")).unwrap();
    rw.insert(ItemIdFlag::new("6", "blue")).unwrap();
    // Green flag
    rw.insert(ItemIdFlag::new("7", "green")).unwrap();
    rw.insert(ItemIdFlag::new("8", "green")).unwrap();
    rw.insert(ItemIdFlag::new("9", "green")).unwrap();
    rw.commit().unwrap();

    for p in ["red:", "blue:", "green:"] {
        let r = db.r_transaction().unwrap();
        let result: Vec<ItemIdFlag> = r
            .scan()
            .secondary(ItemIdFlagKey::flag)
            .unwrap()
            .start_with(p.to_string().as_str())
            .unwrap()
            .try_collect()
            .unwrap();
        assert_eq!(result.len(), 3);

        let obj1 = result.first().unwrap();
        assert_eq!(format!("{}:", obj1.flag), format!("{}", p));

        let obj2 = result.get(1).unwrap();
        assert_eq!(format!("{}:", obj2.flag), format!("{}", p));

        let obj3 = result.get(2).unwrap();
        assert_eq!(format!("{}:", obj3.flag), format!("{}", p));
    }
}

#[test]
fn test_txn_write_iter() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.commit().unwrap();

    let rw = db.rw_transaction().unwrap();
    let result: Vec<Item> = rw
        .scan()
        .primary()
        .unwrap()
        .all()
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 2);
    assert_eq!(obj2.name, "test2");
}

#[test]
fn test_txn_write_iter_range() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<Item>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(Item::new(1, "test")).unwrap();
    rw.insert(Item::new(2, "test2")).unwrap();
    rw.insert(Item::new(3, "test3")).unwrap();
    rw.commit().unwrap();

    let rw = db.rw_transaction().unwrap();
    let result: Vec<Item> = rw
        .scan()
        .primary()
        .unwrap()
        .range(..2u32)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 1);
    assert_eq!(obj1.name, "test");

    let result: Vec<Item> = rw
        .scan()
        .primary()
        .unwrap()
        .range(2u32..)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 2);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");

    let obj2 = result.get(1).unwrap();
    assert_eq!(obj2.id, 3);
    assert_eq!(obj2.name, "test3");

    let result: Vec<Item> = rw
        .scan()
        .primary()
        .unwrap()
        .range(2u32..3u32)
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 1);

    let obj1 = result.first().unwrap();
    assert_eq!(obj1.id, 2);
    assert_eq!(obj1.name, "test2");
}

#[test]
fn test_txn_write_start_with_scenario() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<ItemFlag>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    // Red flag
    rw.insert(ItemFlag::new("red:1")).unwrap();
    rw.insert(ItemFlag::new("red:2")).unwrap();
    rw.insert(ItemFlag::new("red:3")).unwrap();
    // Blue flag
    rw.insert(ItemFlag::new("blue:1")).unwrap();
    rw.insert(ItemFlag::new("blue:2")).unwrap();
    rw.insert(ItemFlag::new("blue:3")).unwrap();
    // Green flag
    rw.insert(ItemFlag::new("green:1")).unwrap();
    rw.insert(ItemFlag::new("green:2")).unwrap();
    rw.insert(ItemFlag::new("green:3")).unwrap();
    rw.commit().unwrap();

    for p in ["red:", "blue:", "green:"] {
        let rw = db.rw_transaction().unwrap();

        let result: Vec<ItemFlag> = rw
            .scan()
            .primary()
            .unwrap()
            .start_with(p.to_string().as_str())
            .unwrap()
            .try_collect()
            .unwrap();
        assert_eq!(result.len(), 3);

        let obj1 = result.first().unwrap();
        assert_eq!(obj1.name, format!("{}1", p));

        let obj2 = result.get(1).unwrap();
        assert_eq!(obj2.name, format!("{}2", p));

        let obj3 = result.get(2).unwrap();
        assert_eq!(obj3.name, format!("{}3", p));
    }
}

#[derive(Serialize, Deserialize, Eq, PartialEq, Clone, Debug)]
#[native_model(id = 4, version = 1)]
#[native_db]
struct ItemScanRange {
    #[primary_key]
    id: u32,

    #[secondary_key]
    nr: u32,

    #[secondary_key(unique)]
    unique_nr: u32,
}

#[test]
fn test_scan_range() {
    let item_1 = ItemScanRange {
        id: 1,
        nr: 1,
        unique_nr: 1,
    };
    let item_2 = ItemScanRange {
        id: 2,
        nr: 2,
        unique_nr: 2,
    };
    let item_3 = ItemScanRange {
        id: 3,
        nr: 2,
        unique_nr: 3,
    };
    let item_4 = ItemScanRange {
        id: 4,
        nr: 3,
        unique_nr: 4,
    };

    let mut models = Models::new();
    models.define::<ItemScanRange>().unwrap();
    let db = Builder::new().create_in_memory(&models).unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(item_1.clone()).unwrap();
    rw.insert(item_2.clone()).unwrap();
    rw.insert(item_3.clone()).unwrap();
    rw.insert(item_4.clone()).unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let result = r
        .scan()
        .secondary(ItemScanRangeKey::nr)
        .unwrap()
        .range(0u32..10u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![1, 2, 2, 3], "range 0..10 for nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::nr)
        .unwrap()
        .range(2u32..3u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![2, 2], "range 2..3 for nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::unique_nr)
        .unwrap()
        .range(1u32..3u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.unique_nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![1, 2], "range 1..3 for unique_nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::unique_nr)
        .unwrap()
        .range(1u32..=3u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.unique_nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![1, 2, 3], "range 1..=3 for unique_nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::unique_nr)
        .unwrap()
        .range(3u32..=3u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.unique_nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![3], "range 3..=3 for unique_nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::nr)
        .unwrap()
        .range(2u32..=3u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![2, 2, 3], "range 2..=3 for nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::nr)
        .unwrap()
        .range(2u32..=2u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![2, 2], "range 2..=2 for nr");

    let result = r
        .scan()
        .secondary(ItemScanRangeKey::nr)
        .unwrap()
        .range(0u32..=2u32)
        .unwrap()
        .collect::<Result<Vec<ItemScanRange>, _>>()
        .unwrap()
        .iter()
        .map(|x| x.nr)
        .collect::<Vec<_>>();
    assert_eq!(result, vec![1, 2, 2], "range 0..=2 for nr");
}

#[derive(Serialize, Deserialize, Eq, PartialEq, Clone, Debug)]
#[native_model(id = 4, version = 1)]
#[native_db]
struct ItemLowLevelScanRange {
    #[primary_key]
    primary_key: Vec<u8>,

    #[secondary_key]
    secondary_key: Vec<u8>,

    #[secondary_key(unique)]
    secondary_key_unique: Vec<u8>,

    #[secondary_key(optional)]
    secondary_key_optional: Option<Vec<u8>>,

    #[secondary_key(unique, optional)]
    secondary_key_unique_optional: Option<Vec<u8>>,
}

#[test]
fn test_low_level_scan_range() {
    let tf = TmpFs::new().unwrap();

    let mut models = Models::new();
    models.define::<ItemLowLevelScanRange>().unwrap();
    let db = Builder::new()
        .create(&models, tf.path("test").as_std_path())
        .unwrap();

    let rw = db.rw_transaction().unwrap();
    rw.insert(ItemLowLevelScanRange {
        primary_key: vec![255],
        secondary_key: vec![2],
        secondary_key_unique: vec![2],
        secondary_key_optional: None,
        secondary_key_unique_optional: None,
    })
    .unwrap();
    rw.insert(ItemLowLevelScanRange {
        primary_key: vec![2],
        secondary_key: vec![2, 0, 254],
        secondary_key_unique: vec![255],
        secondary_key_optional: None,
        secondary_key_unique_optional: None,
    })
    .unwrap();
    rw.commit().unwrap();

    let r = db.r_transaction().unwrap();
    let result: Vec<ItemLowLevelScanRange> = r
        .scan()
        .secondary(ItemLowLevelScanRangeKey::secondary_key)
        .unwrap()
        .all()
        .unwrap()
        .try_collect()
        .unwrap();
    assert_eq!(result.len(), 2);
    assert_eq!(result[0].secondary_key, vec![2]);
    assert_eq!(result[1].secondary_key, vec![2, 0, 254]);

    // Maybe this idea does not work because it's lexicographical order, and the fact to have th same size of the primary key part
    // change nothing.
    // Use XXhash for the primary_key part https://docs.rs/xxhash-rust/latest/xxhash_rust/xxh64/index.html of the secondary_key.
    // And detect colision, use the return value of https://docs.rs/redb/latest/redb/struct.Table.html#method.insert
    // and re-insert the item if the return value is not null, recompute the primary key hash with a timestamp.
}