add QuotaType::kHard

reductstore · Sep 12, 2024 · 3779578 · 3779578
1 parent 6a498ba
commit 3779578
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/src/reduct/bucket.h b/src/reduct/bucket.h
@@ -29,7 +29,7 @@ class IBucket {
  public:
   virtual ~IBucket() = default;
 
-  enum class QuotaType { kNone, kFifo };
+  enum class QuotaType { kNone, kFifo, kHard };
 
   /**
    * Bucket Settings

diff --git a/src/reduct/internal/serialisation.cc b/src/reduct/internal/serialisation.cc
@@ -25,6 +25,9 @@ nlohmann::json BucketSettingToJsonString(const IBucket::Settings& settings) {
       case IBucket::QuotaType::kFifo:
         data["quota_type"] = "FIFO";
         break;
+      case IBucket::QuotaType::kHard:
+        data["quota_type"] = "HARD";
+        break;
     }
   }
 
@@ -51,6 +54,8 @@ Result<IBucket::Settings> ParseBucketSettings(const nlohmann::json& json) {
         settings.quota_type = IBucket::QuotaType::kNone;
       } else if (json["quota_type"] == "FIFO") {
         settings.quota_type = IBucket::QuotaType::kFifo;
+      } else if (json["quota_type"] == "HARD") {
+        settings.quota_type = IBucket::QuotaType::kHard;
       }
     }
 

diff --git a/tests/reduct/bucket_api_test.cc b/tests/reduct/bucket_api_test.cc
@@ -51,6 +51,21 @@ TEST_CASE("reduct::Client should get or create a bucket", "[bucket_api]") {
   REQUIRE(bucket);
 }
 
+TEST_CASE("reduct::Client should create a bucket with different quota type", "[bucket_api][1_12]") {
+  Fixture ctx;
+
+  auto quota_type = GENERATE(IBucket::QuotaType::kNone, IBucket::QuotaType::kFifo, IBucket::QuotaType::kHard);
+  [[maybe_unused]] auto _ = ctx.client->GetOrCreateBucket(kBucketName, {.quota_type = quota_type});
+  auto [bucket, err] = ctx.client->GetOrCreateBucket(kBucketName);
+
+  REQUIRE(err == Error::kOk);
+  REQUIRE(bucket);
+
+  auto [settings, get_error] = bucket->GetSettings();
+  REQUIRE(get_error == Error::kOk);
+  REQUIRE(settings.quota_type == quota_type);
+}
+
 TEST_CASE("reduct::IBucket should have settings", "[bucket_api]") {
   Fixture ctx;
   const IBucket::Settings kSettings{