salsa/tests/gc/derived_tests.rs
Niko Matsakis 1002d7e70a
Merge pull request #138 from matklad/flexible-gc
Make GC API more orthogonal and flexible
2019-01-28 04:51:02 -05:00

154 lines
3.8 KiB
Rust

use crate::db;
use crate::group::*;
use salsa::debug::DebugQueryTable;
use salsa::{Database, SweepStrategy};
#[test]
fn compute_one() {
let mut db = db::DatabaseImpl::default();
// Will compute fibonacci(5)
db.set_use_triangular(5, false);
db.compute(5);
db.salsa_runtime().next_revision();
assert_keys! {
db,
TriangularQuery => (),
FibonacciQuery => (0, 1, 2, 3, 4, 5),
ComputeQuery => (5),
UseTriangularQuery => (5),
MinQuery => (),
MaxQuery => (),
}
// Memoized, but will compute fibonacci(5) again
db.compute(5);
db.sweep_all(SweepStrategy::discard_outdated());
assert_keys! {
db,
TriangularQuery => (),
FibonacciQuery => (5),
ComputeQuery => (5),
UseTriangularQuery => (5),
MinQuery => (),
MaxQuery => (),
}
}
#[test]
fn compute_switch() {
let mut db = db::DatabaseImpl::default();
// Will compute fibonacci(5)
db.set_use_triangular(5, false);
assert_eq!(db.compute(5), 5);
// Change to triangular mode
db.set_use_triangular(5, true);
// Now computes triangular(5)
assert_eq!(db.compute(5), 15);
// We still have entries for Fibonacci, even though they
// are not relevant to the most recent value of `Compute`
assert_keys! {
db,
TriangularQuery => (0, 1, 2, 3, 4, 5),
FibonacciQuery => (0, 1, 2, 3, 4, 5),
ComputeQuery => (5),
UseTriangularQuery => (5),
MinQuery => (),
MaxQuery => (),
}
db.sweep_all(SweepStrategy::discard_outdated());
// Now we just have `Triangular` and not `Fibonacci`
assert_keys! {
db,
TriangularQuery => (0, 1, 2, 3, 4, 5),
FibonacciQuery => (),
ComputeQuery => (5),
UseTriangularQuery => (5),
MinQuery => (),
MaxQuery => (),
}
// Now run `compute` *again* in next revision.
db.salsa_runtime().next_revision();
assert_eq!(db.compute(5), 15);
db.sweep_all(SweepStrategy::discard_outdated());
// We keep triangular, but just the outermost one.
assert_keys! {
db,
TriangularQuery => (5),
FibonacciQuery => (),
ComputeQuery => (5),
UseTriangularQuery => (5),
MinQuery => (),
MaxQuery => (),
}
}
/// Test a query with multiple layers of keys.
#[test]
fn compute_all() {
let mut db = db::DatabaseImpl::default();
for i in 0..6 {
db.set_use_triangular(i, (i % 2) != 0);
}
db.set_min(0);
db.set_max(6);
db.compute_all();
db.salsa_runtime().next_revision();
db.compute_all();
db.sweep_all(SweepStrategy::discard_outdated());
assert_keys! {
db,
TriangularQuery => (1, 3, 5),
FibonacciQuery => (0, 2, 4),
ComputeQuery => (0, 1, 2, 3, 4, 5),
ComputeAllQuery => (()),
UseTriangularQuery => (0, 1, 2, 3, 4, 5),
MinQuery => (()),
MaxQuery => (()),
}
// Reduce the range to exclude index 5.
db.set_max(5);
db.compute_all();
assert_keys! {
db,
TriangularQuery => (1, 3, 5),
FibonacciQuery => (0, 2, 4),
ComputeQuery => (0, 1, 2, 3, 4, 5),
ComputeAllQuery => (()),
UseTriangularQuery => (0, 1, 2, 3, 4, 5),
MinQuery => (()),
MaxQuery => (()),
}
db.sweep_all(SweepStrategy::discard_outdated());
// We no longer used `Compute(5)` and `Triangular(5)`; note that
// `UseTriangularQuery(5)` is not collected, as it is an input.
assert_keys! {
db,
TriangularQuery => (1, 3),
FibonacciQuery => (0, 2, 4),
ComputeQuery => (0, 1, 2, 3, 4),
ComputeAllQuery => (()),
UseTriangularQuery => (0, 1, 2, 3, 4, 5),
MinQuery => (()),
MaxQuery => (()),
}
}