Radish alpha
h
heartwood
Radicle Heartwood Protocol & Stack
Radicle
Git (anonymous pull)
Log in to clone via SSH
Commits ›
REVIEW
✗ CI failure Fintan Halpenny committed 1 month ago
commit b170395241969980e36854d4022524c5fe07b81a
parent 273c3438b21d52c862ed0fd204801f9bf0f3fed6
1 failed (1 total) View logs
1 file changed +218 -0
modified crates/radicle/src/node/db.rs
@@ -38,6 +38,7 @@ const MIGRATIONS: &[&str] = &[
    include_str!("db/migrations/5.sql"),

    include_str!("db/migrations/6.sql"),

    include_str!("db/migrations/7.sql"),
+ 
    include_str!("db/migrations/8.sql"),

];



#[derive(Error, Debug)]
@@ -165,6 +166,34 @@ impl Database {
        transaction(&self.db, bump)

    }
+ 
    #[cfg(test)]
+ 
    fn memory_up_to_migration(n: usize) -> Result<Self, Error> {
+ 
        if n == 0 {
+ 
            panic!("Migration number 'n' must be larger than 0");
+ 
        } else if n > MIGRATIONS.len() {
+ 
            panic!(
+ 
                "Migration number {n} exceeds the number of migrations {}",
+ 
                MIGRATIONS.len()
+ 
            );
+ 
        }
+ 
        let db = sql::Connection::open_thread_safe(":memory:")?;
+ 
        db.execute(Self::PRAGMA)?;
+ 
        {
+ 
            let mut version = version(&db)?;
+ 
            for (i, migration) in MIGRATIONS.iter().enumerate().take(n) {
+ 
                if i >= version {
+ 
                    transaction(&db, |db| {
+ 
                        db.execute(migration)?;
+ 
                        version = bump(db)?;
+
+ 
                        Ok::<_, Error>(())
+ 
                    })?;
+ 
                }
+ 
            }
+ 
        }
+ 
        Ok(Self { db: Arc::new(db) })
+ 
    }
+ 


    fn configure(self, config: config::Config) -> Result<Self, Error> {

        self.journal_mode(config.sqlite.pragma.journal_mode)?

            .synchronous(config.sqlite.pragma.synchronous)
@@ -228,4 +257,193 @@ mod test {
        assert_eq!(v, n + 2);

        assert_eq!(db.version().unwrap(), n + 2);

    }
+
+ 
    mod migration_8 {
+ 
        use super::*;
+
+ 
        const NODE1: &str = "node1";
+ 
        const NODE2: &str = "node2";
+
+ 
        fn db_before_migration() -> Database {
+ 
            let db = Database::memory_up_to_migration(7).unwrap();
+ 
            db.execute(
+ 
                "INSERT INTO nodes (id, features, alias, timestamp)
+ 
             VALUES ('node1', 0, 'alias', 0)",
+ 
            )
+ 
            .unwrap();
+ 
            db.execute(
+ 
                "INSERT INTO nodes (id, features, alias, timestamp)
+ 
         VALUES ('node2', 0, 'alias2', 0)",
+ 
            )
+ 
            .unwrap();
+ 
            db
+ 
        }
+
+ 
        fn run_migration(db: &Database) {
+ 
            db.execute(MIGRATIONS[7]).unwrap();
+ 
        }
+
+ 
        fn address_count(db: &Database, node: &str, address_type: &str, value: &str) -> i64 {
+ 
            db.prepare(format!(
+ 
                "SELECT COUNT(*) FROM addresses
+ 
             WHERE node = '{node}' AND type = '{address_type}' AND value = '{value}'"
+ 
            ))
+ 
            .unwrap()
+ 
            .into_iter()
+ 
            .next()
+ 
            .unwrap()
+ 
            .unwrap()
+ 
            .read::<i64, _>(0)
+ 
        }
+
+ 
        fn insert_address(db: &Database, node: &str, address_type: &str, value: &str) {
+ 
            db.execute(format!(
+ 
                "INSERT INTO addresses (node, type, value, source, timestamp)
+ 
             VALUES ('{node}', '{address_type}', '{value}', 'peer', 0)"
+ 
            ))
+ 
            .unwrap();
+ 
        }
+
+ 
        #[test]
+ 
        fn ipv6_formatted_dns_address_is_retyped_to_ipv6() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "[::1]:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(address_count(&db, NODE1, "ipv6", "[::1]:8776"), 1);
+ 
            assert_eq!(address_count(&db, NODE1, "dns", "[::1]:8776"), 0);
+ 
        }
+
+ 
        #[test]
+ 
        fn ipv6_formatted_dns_address_is_deleted_when_correct_ipv6_row_already_exists() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "ipv6", "[2001:db8::1]:8776");
+ 
            insert_address(&db, NODE1, "dns", "[2001:db8::1]:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(
+ 
                address_count(&db, NODE1, "ipv6", "[2001:db8::1]:8776"),
+ 
                1,
+ 
                "existing ipv6 row survives"
+ 
            );
+ 
            assert_eq!(
+ 
                address_count(&db, NODE1, "dns", "[2001:db8::1]:8776"),
+ 
                0,
+ 
                "stale dns row is removed"
+ 
            );
+ 
        }
+
+ 
        #[test]
+ 
        fn plain_dns_hostname_without_brackets_is_unaffected() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "example.com:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(address_count(&db, NODE1, "dns", "example.com:8776"), 1);
+ 
        }
+
+ 
        #[test]
+ 
        fn dns_address_with_bracket_not_at_start_is_unaffected() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "foo[::1]:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(address_count(&db, NODE1, "dns", "foo[::1]:8776"), 1);
+ 
        }
+
+ 
        // The `Address` type always contains a port so this case should never
+ 
        // be hit, but recording it here for posterity.
+ 
        #[test]
+ 
        fn dns_address_starting_with_bracket_but_missing_closing_bracket_colon_is_unaffected() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "[::1]");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(address_count(&db, NODE1, "dns", "[::1]"), 1);
+ 
        }
+
+ 
        #[test]
+ 
        fn ipv4_address_is_unaffected() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "ipv4", "192.168.1.1:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(address_count(&db, NODE1, "ipv4", "192.168.1.1:8776"), 1);
+ 
        }
+
+ 
        #[test]
+ 
        fn retype_preserves_address_metadata() {
+ 
            let db = db_before_migration();
+ 
            db.execute(
+ 
        "INSERT INTO addresses (node, type, value, source, timestamp, last_attempt, last_success, banned)
+ 
         VALUES ('node1', 'dns', '[::1]:8776', 'peer', 0, 1000, 2000, 1)"
+ 
    ).unwrap();
+
+ 
            run_migration(&db);
+
+ 
            let row = db
+ 
                .prepare(
+ 
                    "SELECT last_attempt, last_success, banned FROM addresses
+ 
         WHERE node = 'node1' AND type = 'ipv6' AND value = '[::1]:8776'",
+ 
                )
+ 
                .unwrap()
+ 
                .into_iter()
+ 
                .next()
+ 
                .unwrap()
+ 
                .unwrap();
+
+ 
            assert_eq!(row.read::<i64, _>("last_attempt"), 1000);
+ 
            assert_eq!(row.read::<i64, _>("last_success"), 2000);
+ 
            assert_eq!(row.read::<i64, _>("banned"), 1);
+ 
        }
+
+ 
        #[test]
+ 
        fn migration_applies_to_all_nodes() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "[::1]:8776");
+ 
            insert_address(&db, NODE2, "dns", "[::1]:8776");
+
+ 
            run_migration(&db);
+
+ 
            assert_eq!(
+ 
                address_count(&db, NODE1, "ipv6", "[::1]:8776"),
+ 
                1,
+ 
                "node1 address retyped"
+ 
            );
+ 
            assert_eq!(
+ 
                address_count(&db, NODE2, "ipv6", "[::1]:8776"),
+ 
                1,
+ 
                "node1 address retyped"
+ 
            );
+ 
        }
+
+ 
        #[test]
+ 
        fn all_ipv6_formatted_dns_addresses_are_retyped() {
+ 
            let db = db_before_migration();
+ 
            insert_address(&db, NODE1, "dns", "[::1]:8776");
+ 
            insert_address(&db, NODE1, "dns", "[2001:db8::1]:8776");
+ 
            insert_address(&db, NODE1, "dns", "[fe80::1]:8776");
+
+ 
            run_migration(&db);
+
+ 
            for value in ["[::1]:8776", "[2001:db8::1]:8776", "[fe80::1]:8776"] {
+ 
                assert_eq!(
+ 
                    address_count(&db, NODE1, "ipv6", value),
+ 
                    1,
+ 
                    "{value} should be ipv6"
+ 
                );
+ 
                assert_eq!(
+ 
                    address_count(&db, NODE1, "dns", value),
+ 
                    0,
+ 
                    "{value} dns row should be gone"
+ 
                );
+ 
            }
+ 
        }
+ 
    }

}