From 2dd0cb4e6a062caa30cde649aeee0c4072f9eb92 Mon Sep 17 00:00:00 2001
From: Arnaud <arnaud@status.im>
Date: Wed, 29 Jan 2025 17:20:02 +0100
Subject: [PATCH] Add availability enabled parameter

---
 tests/codex/sales/testreservations.nim |  2 +-
 tests/codex/sales/testsales.nim        |  2 +-
 tests/integration/testrestapi.nim      | 13 +++++++++++++
 tests/integration/testsales.nim        | 12 ++++++++++++
 4 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/tests/codex/sales/testreservations.nim b/tests/codex/sales/testreservations.nim
index 39528a91..691a1273 100644
--- a/tests/codex/sales/testreservations.nim
+++ b/tests/codex/sales/testreservations.nim
@@ -364,7 +364,7 @@ asyncchecksuite "Reservations module":
         a: Availability
     ) {.gcsafe, async: (raises: []).} =
       called = true
-    availability.freeSize -= 1.u256
+    availability.freeSize -= 1.uint64
     discard await reservations.update(availability)
 
     check not called
diff --git a/tests/codex/sales/testsales.nim b/tests/codex/sales/testsales.nim
index bb8eb357..3aea608c 100644
--- a/tests/codex/sales/testsales.nim
+++ b/tests/codex/sales/testsales.nim
@@ -369,7 +369,7 @@ asyncchecksuite "Sales":
     ): Future[?!void] {.async.} =
       slotIndex = slot
       let blk = bt.Block.new(@[1.byte]).get
-      await onBatch(blk.repeat(request.ask.slotSize.truncate(int)))
+      await onBatch(blk.repeat(request.ask.slotSize))
 
     let sold = newFuture[void]()
     sales.onSale = proc(request: StorageRequest, slotIndex: uint64) =
diff --git a/tests/integration/testrestapi.nim b/tests/integration/testrestapi.nim
index 366cc3e3..c6b2c494 100644
--- a/tests/integration/testrestapi.nim
+++ b/tests/integration/testrestapi.nim
@@ -42,6 +42,19 @@ twonodessuite "REST API":
       space.quotaUsedBytes == 65592.NBytes
       space.quotaReservedBytes == 12.NBytes
 
+  test "created correctly an availability when not enabled by default", twoNodesConfig:
+    let totalSize = 12.u256
+    let minPricePerBytePerSecond = 1.u256
+    let totalCollateral = totalSize * minPricePerBytePerSecond
+    let availability = client1.postAvailability(
+      totalSize = totalSize,
+      duration = 2.u256,
+      minPricePerBytePerSecond = minPricePerBytePerSecond,
+      totalCollateral = totalCollateral,
+      enabled = false.some,
+    ).get
+    check availability.enabled == false
+
   test "node lists local files", twoNodesConfig:
     let content1 = "some file contents"
     let content2 = "some other contents"
diff --git a/tests/integration/testsales.nim b/tests/integration/testsales.nim
index e409e804..8622996a 100644
--- a/tests/integration/testsales.nim
+++ b/tests/integration/testsales.nim
@@ -135,6 +135,18 @@ multinodesuite "Sales":
     check updatedAvailability.totalSize == 100000
     check updatedAvailability.freeSize == 100000
 
+  test "updating availability - updating enabled", salesConfig:
+    let availability = host.postAvailability(
+      totalSize = 140000.u256,
+      duration = 200.u256,
+      minPricePerBytePerSecond = 3.u256,
+      totalCollateral = 300.u256,
+      enabled = true.some,
+    ).get
+    host.patchAvailability(availability.id, enabled = false.some)
+    let updatedAvailability = (host.getAvailabilities().get).findItem(availability).get
+    check updatedAvailability.enabled == false
+
   test "updating availability - updating totalSize does not allow bellow utilized",
     salesConfig:
     let originalSize = 0xFFFFFF.uint64