rustfs/crates/ecstore/tests/ecstore_contract_compat_test.rs

// Copyright 2024 RustFS Team
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

mod storage_api;

use rustfs_common::heal_channel::HealOpts;
use rustfs_filemeta::FileInfo;
use rustfs_lock::NamespaceLockWrapper;
use rustfs_madmin::heal_commands::HealResultItem;
use storage_api::contract_compat::{
    CompletePart, DeletedObject, DiskStore, ECStore, Error, GetObjectReader, HTTPRangeSpec, ListMultipartsInfo, ListPartsInfo,
    MultipartInfo, MultipartUploadResult, ObjectInfo, ObjectOptions, ObjectToDelete, PartInfo, PutObjReader, SetDisks,
    StorageAdminApi, StorageBucketOperations, StorageHealOperations, StorageListObjectVersionsInfo, StorageListObjectsV2Info,
    StorageListOperations, StorageMultipartOperations, StorageNamespaceLocking, StorageObjectIO, StorageObjectInfoOrErr,
    StorageObjectOperations, StorageWalkOptions,
};
use tokio_util::sync::CancellationToken;

type ListObjectsV2Info = StorageListObjectsV2Info<ObjectInfo>;
type ListObjectVersionsInfo = StorageListObjectVersionsInfo<ObjectInfo>;
type ObjectInfoOrErr = StorageObjectInfoOrErr<ObjectInfo, Error>;
type WalkOptions = StorageWalkOptions<fn(&FileInfo) -> bool>;

fn storage_admin_api_type_name<T>() -> &'static str
where
    T: StorageAdminApi<
            BackendInfo = rustfs_madmin::BackendInfo,
            StorageInfo = rustfs_madmin::StorageInfo,
            Disk = DiskStore,
            Error = Error,
        >,
{
    std::any::type_name::<T>()
}

fn storage_namespace_locking_type_name<T>() -> &'static str
where
    T: StorageNamespaceLocking<Error = Error, NamespaceLock = NamespaceLockWrapper>,
{
    std::any::type_name::<T>()
}

fn storage_object_io_type_name<T>() -> &'static str
where
    T: StorageObjectIO<
            Error = Error,
            RangeSpec = HTTPRangeSpec,
            HeaderMap = http::HeaderMap,
            ObjectOptions = ObjectOptions,
            ObjectInfo = ObjectInfo,
            GetObjectReader = GetObjectReader,
            PutObjectReader = PutObjReader,
        >,
{
    std::any::type_name::<T>()
}

fn storage_bucket_operations_type_name<T>() -> &'static str
where
    T: StorageBucketOperations<Error = Error>,
{
    std::any::type_name::<T>()
}

fn storage_object_operations_type_name<T>() -> &'static str
where
    T: StorageObjectOperations<
            Error = Error,
            ObjectInfo = ObjectInfo,
            ObjectOptions = ObjectOptions,
            FileInfo = FileInfo,
            ObjectToDelete = ObjectToDelete,
            DeletedObject = DeletedObject,
        >,
{
    std::any::type_name::<T>()
}

fn storage_list_operations_type_name<T>() -> &'static str
where
    T: StorageListOperations<
            Error = Error,
            ListObjectsV2Info = ListObjectsV2Info,
            ListObjectVersionsInfo = ListObjectVersionsInfo,
            ObjectInfoOrErr = ObjectInfoOrErr,
            WalkOptions = WalkOptions,
            WalkCancellation = CancellationToken,
            WalkResultSender = tokio::sync::mpsc::Sender<ObjectInfoOrErr>,
        >,
{
    std::any::type_name::<T>()
}

fn storage_multipart_operations_type_name<T>() -> &'static str
where
    T: StorageMultipartOperations<
            Error = Error,
            ObjectInfo = ObjectInfo,
            ObjectOptions = ObjectOptions,
            PutObjectReader = PutObjReader,
            CompletePart = CompletePart,
            ListMultipartsInfo = ListMultipartsInfo,
            MultipartUploadResult = MultipartUploadResult,
            PartInfo = PartInfo,
            MultipartInfo = MultipartInfo,
            ListPartsInfo = ListPartsInfo,
        >,
{
    std::any::type_name::<T>()
}

fn storage_heal_operations_type_name<T>() -> &'static str
where
    T: StorageHealOperations<Error = Error, HealResultItem = HealResultItem, HealOptions = HealOpts>,
{
    std::any::type_name::<T>()
}

#[test]
fn ecstore_implements_storage_admin_api_contract() {
    assert!(storage_admin_api_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_namespace_locking_contract() {
    assert!(storage_namespace_locking_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_object_io_contract() {
    assert!(storage_object_io_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_bucket_operations_contract() {
    assert!(storage_bucket_operations_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_object_operations_contract() {
    assert!(storage_object_operations_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_list_operations_contract() {
    assert!(storage_list_operations_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_pools_expose_storage_list_operations_contract() {
    fn assert_contract(store: &ECStore) {
        let future = store.pools[0]
            .clone()
            .list_objects_v2("bucket", "", None, None, 1, false, None, false);
        drop(future);
    }

    let _ = assert_contract;
}

#[test]
fn ecstore_implements_storage_multipart_operations_contract() {
    assert!(storage_multipart_operations_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn ecstore_implements_storage_heal_operations_contract() {
    assert!(storage_heal_operations_type_name::<ECStore>().ends_with("::ECStore"));
}

#[test]
fn set_disks_implements_storage_namespace_locking_contract() {
    assert!(storage_namespace_locking_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_object_io_contract() {
    assert!(storage_object_io_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_bucket_operations_contract() {
    assert!(storage_bucket_operations_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_object_operations_contract() {
    assert!(storage_object_operations_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_list_operations_contract() {
    assert!(storage_list_operations_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_multipart_operations_contract() {
    assert!(storage_multipart_operations_type_name::<SetDisks>().ends_with("::SetDisks"));
}

#[test]
fn set_disks_implements_storage_heal_operations_contract() {
    assert!(storage_heal_operations_type_name::<SetDisks>().ends_with("::SetDisks"));
}