storage_benchmarks/
testing.rs

1// Copyright 2022 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use crate::filesystem::MountedFilesystemInstance;
6use crate::{CacheClearableFilesystem, Filesystem};
7use async_trait::async_trait;
8use futures::lock::Mutex;
9use std::path::{Path, PathBuf};
10use std::sync::Arc;
11use tempfile::TempDir;
12
13/// Filesystem implementation that records `clear_cache` calls for validating warm and cold
14/// benchmarks.
15#[derive(Clone)]
16pub struct TestFilesystem {
17    benchmark_dir: PathBuf,
18    inner: Arc<Mutex<TestFilesystemInner>>,
19}
20
21struct TestFilesystemInner {
22    fs: Option<Box<MountedFilesystemInstance>>,
23    clear_cache_count: u64,
24}
25
26impl TestFilesystem {
27    pub fn new() -> Self {
28        let benchmark_dir = TempDir::new_in("/tmp").unwrap().into_path();
29        Self {
30            inner: Arc::new(Mutex::new(TestFilesystemInner {
31                fs: Some(Box::new(MountedFilesystemInstance::new(&benchmark_dir))),
32                clear_cache_count: 0,
33            })),
34            benchmark_dir,
35        }
36    }
37
38    pub async fn clear_cache_count(&self) -> u64 {
39        self.inner.lock().await.clear_cache_count
40    }
41}
42
43#[async_trait]
44impl Filesystem for TestFilesystem {
45    async fn shutdown(self) {
46        let mut inner = self.inner.lock().await;
47        inner.fs.take().unwrap().shutdown().await;
48    }
49
50    fn benchmark_dir(&self) -> &Path {
51        &self.benchmark_dir
52    }
53}
54
55#[async_trait]
56impl CacheClearableFilesystem for TestFilesystem {
57    async fn clear_cache(&mut self) {
58        let mut inner = self.inner.lock().await;
59        inner.clear_cache_count += 1;
60    }
61}
62
63impl Drop for TestFilesystem {
64    fn drop(&mut self) {
65        assert!(
66            !self.benchmark_dir.try_exists().unwrap(),
67            "The benchmark directory still exists. Shutdown may not have been called."
68        );
69    }
70}
storage_benchmarks/testing.rs

storage_benchmarks/
testing.rs
