// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use {
    async_trait::async_trait,
    fs_management::{Blobfs, Filesystem},
    storage_stress_test_utils::fvm::FvmInstance,
    stress_test::actor::{Actor, ActorError},
};

/// An actor that kills blobfs and destroys the ramdisk
pub struct InstanceActor {
    pub instance: Option<(Filesystem<Blobfs>, FvmInstance)>,
}

impl InstanceActor {
    pub fn new(fvm: FvmInstance, blobfs: Filesystem<Blobfs>) -> Self {
        Self { instance: Some((blobfs, fvm)) }
    }
}

#[async_trait]
impl Actor for InstanceActor {
    async fn perform(&mut self) -> Result<(), ActorError> {
        if let Some((mut blobfs, _)) = self.instance.take() {
            blobfs.kill().expect("Could not kill blobfs");
        } else {
            panic!("Instance was already killed!")
        }
        Err(ActorError::ResetEnvironment)
    }
}