2019-06-28 05:42:21 +00:00
|
|
|
//////////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
//// Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
|
|
|
//// Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
|
|
|
//// Proprietary and confidential.
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
//
|
|
|
|
|
//#include "storage/IStorage.h"
|
|
|
|
|
//#include "storage/s3/S3ClientWrapper.h"
|
|
|
|
|
//#include <gtest/gtest.h>
|
|
|
|
|
//#include <memory.h>
|
|
|
|
|
//#include <fstream>
|
|
|
|
|
//
|
|
|
|
|
//
|
|
|
|
|
//using namespace zilliz::milvus::engine;
|
|
|
|
|
//
|
|
|
|
|
//TEST(s3_client_wrapper, CLIENT_WRAPPER_TEST) {
|
|
|
|
|
//
|
|
|
|
|
// std::shared_ptr<storage::IStorage> storage_ptr = std::make_shared<storage::S3ClientWrapper>();
|
|
|
|
|
//
|
|
|
|
|
// std::string ip_address = "127.0.0.1";
|
|
|
|
|
// std::string port = "9000";
|
|
|
|
|
// std::string access_key = "AKIAIOSFODNN7EXAMPLE";
|
|
|
|
|
// std::string secret_key = "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY";
|
|
|
|
|
//
|
|
|
|
|
// Status status = storage_ptr->Create(ip_address, port, access_key, secret_key);
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//
|
|
|
|
|
// std::string filename = "/tmp/s3_test_file";
|
|
|
|
|
// std::string bucket_name = "bucktname";
|
|
|
|
|
// std::string object_name = "test_file";
|
|
|
|
|
//
|
|
|
|
|
// status = storage_ptr->CreateBucket(bucket_name);
|
|
|
|
|
// std::cout << status.IsAlreadyExist() << std::endl;
|
|
|
|
|
// if (status.IsAlreadyExist()) {
|
|
|
|
|
// status = storage_ptr->DeleteBucket(bucket_name);
|
|
|
|
|
// status = storage_ptr->CreateBucket(bucket_name);
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//
|
|
|
|
|
// std::ofstream ofile(filename);
|
|
|
|
|
// std::stringstream ss;
|
|
|
|
|
// for (int i = 0; i < 1024; ++i) {
|
|
|
|
|
// ss << i;
|
|
|
|
|
// }
|
|
|
|
|
// ofile << ss.str() << std::endl;
|
|
|
|
|
// ofile.close();
|
|
|
|
|
// status = storage_ptr->UploadFile(bucket_name, object_name, filename);
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//
|
|
|
|
|
// status = storage_ptr->DownloadFile(bucket_name, object_name, filename);
|
|
|
|
|
// std::ifstream infile(filename);
|
|
|
|
|
// std::string in_buffer;
|
|
|
|
|
// infile >> in_buffer;
|
|
|
|
|
// ASSERT_STREQ(in_buffer.c_str(), ss.str().c_str());
|
|
|
|
|
//
|
|
|
|
|
// status = storage_ptr->DeleteFile(bucket_name, object_name);
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//
|
|
|
|
|
// status = storage_ptr->DeleteBucket(bucket_name);
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//
|
|
|
|
|
// status = storage_ptr->Close();
|
|
|
|
|
// ASSERT_TRUE(status.ok());
|
|
|
|
|
//}
|
|
|
|
|
//
|
