Function load_1_2_data

crates/engine/src/relational_db.rs:3908–3959

Load a v1.2 database, verify that the expected tables are present, and we can add a table.

(use_snapshot: bool)

    /// Load a v1.2 database, verify that the expected tables are present, and we can add a table.
    fn load_1_2_data(use_snapshot: bool) -> ResultTest<()> {
        let data_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("testdata/v1.2/replicas");
        let tempdir = copy_fixture_dir(&data_dir);

        let dir = ReplicaDir::from_path_unchecked(tempdir.path().join("replicas/22000001"));

        let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;
        // Enter the runtime so that `Self::durable_internal` can spawn a `SnapshotWorker`.
        let _rt = rt.enter();

        let db_identity = Identity::from_hex("c2006c1c2ede44b44e9835dfe00e3e1edc3bc67024b504cb6a3b8491db5d98a3")?;
        let owner_identity = Identity::from_hex("c2001c551217385bdeba5f1e1b5b19a28910852a94ce43428508e7a098760cea")?;
        let (db, _durability_handle) = TestDB::open_existing_durable(
            &dir,
            rt.handle().clone(),
            22000001,
            db_identity,
            owner_identity,
            use_snapshot,
        )?;
        let mut tx = db.begin_mut_tx(IsolationLevel::Serializable, Workload::ForTests);
        let schemas = db.get_all_tables_mut(&tx)?;
        let user_table_names: Vec<String> = schemas
            .iter()
            .filter(|s| s.table_id.0 > 4000)
            .sorted_by_key(|s| s.table_id.0)
            .map(|s| s.table_name.clone().to_string())
            .collect();
        let expected_table_names = vec!["message", "user"];
        assert_eq!(user_table_names, expected_table_names);

        let table_id = db.create_table(&mut tx, my_table(AlgebraicType::I32))?;
        db.commit_tx(tx)?;

        let tx = db.begin_mut_tx(IsolationLevel::Serializable, Workload::ForTests);
        let t_id = db.table_id_from_name_mut(&tx, "MyTable")?.unwrap();
        assert_eq!(table_id, t_id);
        assert_eq!("MyTable", db.table_name_from_id_mut(&tx, t_id)?.unwrap());
        db.commit_tx(tx)?;
        let tx = db.begin_tx(Workload::ForTests);
        let user_table_names: Vec<String> = db
            .get_all_tables(&tx)?
            .iter()
            .filter(|s| s.table_id.0 > 4000)
            .sorted_by_key(|s| s.table_id.0)
            .map(|s| s.table_name.clone().to_string())
            .collect();
        let expected_table_names = vec!["message", "user", "MyTable"];
        assert_eq!(user_table_names, expected_table_names);

        Ok(())
    }

    #[test]
    fn load_1_2_quickstart_from_snapshot_test() -> ResultTest<()> {