new test cases added

alienrobotninja · Jan 4, 2024 · 666ad86 · 666ad86
1 parent e1a1e2a
commit 666ad86
Show file tree

Hide file tree

Showing 2 changed files with 79 additions and 4 deletions.
diff --git a/src/bee_py/utils/type.py b/src/bee_py/utils/type.py
@@ -492,7 +492,7 @@ def assert_transaction_hash(transaction_hash: Any, name: str = "TransactionHash"
         raise TypeError(msg)
 
     if not is_prefixed_hex_string(transaction_hash):
-        raise TypeError()
+        raise TypeError(f"Invalid transaction hash. Expected hex string got: {transaction_hash}")
 
     # Hash is 64 long + '0x' prefix = 66
     if len(transaction_hash) != PUBKEY_HEX_LENGTH:

diff --git a/tests/unit/test_bee_debug.py b/tests/unit/test_bee_debug.py
@@ -70,7 +70,6 @@
 transaction_options_assertions: list[tuple] = [
     (1, TypeError),
     (True, TypeError),
-    ([], TypeError),
     ("string", TypeError),
     ({"gasPrice": "plur"}, TypeError),
     ({"gasPrice": True}, TypeError),
@@ -84,6 +83,17 @@
     ({"gasLimit": -1}, TypeError),
 ]
 
+batch_id_assertion_data: list[tuple] = [
+    (1, TypeError),
+    (True, TypeError),
+    ({}, BeeError),
+    (None, BeeError),
+    ([], BeeError),
+    ("", BeeError),
+    ("ZZZfb5a872396d9693e5c9f9d7233cfa93f395c093371017ff44aa9ae6564cdd", TypeError),
+    ("0x634fb5a872396d9693e5c9f9d7233cfa93f395c093371017ff44aa9ae6564cdd", TypeError),
+]
+
 
 @pytest.mark.parametrize(
     "url",
@@ -423,7 +433,7 @@ def test_create_postage_batch_transaction_assertions(input_value, expected_error
 def test_no_headers_if_no_create_postage_batch_is_set(requests_mock):
     url = "http://localhost:12345/stamps/10/17"
 
-    requests_mock.post(url, json=BATCH_ID)
+    requests_mock.post(url, json=BATCH_RESPONSE)
 
     bee = BeeDebug(MOCK_SERVER_URL)
 
@@ -433,8 +443,73 @@ def test_no_headers_if_no_create_postage_batch_is_set(requests_mock):
 def test_no_headers_if_create_postage_batch_is_set(requests_mock):
     url = "http://localhost:12345/stamps/10/17"
 
-    requests_mock.post(url, headers={"gas-price": "100000000000"}, json=BATCH_ID)
+    requests_mock.post(url, headers={"gas-price": "100000000000"}, json=BATCH_RESPONSE)
 
     bee = BeeDebug(MOCK_SERVER_URL)
 
     assert bee.create_postage_batch("10", 17, {"waitForUsable": False, "gasPrice": "100"}) == BATCH_ID
+
+
+#! Test from here
+
+
+@pytest.mark.parametrize(
+    "input_value, expected_error_type",
+    [
+        ({"immutable_flag": "asd"}, TypeError),
+        ({"immutable_flag": -1}, TypeError),
+        ({"immutable_flag": "true"}, TypeError),
+    ],
+)
+def test_throw_error_if_wrong_immutable_input(input_value, expected_error_type):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(expected_error_type):
+        bee.create_postage_batch("10", 17, input_value)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", [(-1, TypeError), (15, TypeError)])
+def test_throw_error_if_too_small_depth(input_value, expected_error_type):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(TypeError):
+        bee.create_postage_batch("10", input_value)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", [("-10", TypeError), ("0", TypeError)])
+def test_throw_error_if_too_small_amount(input_value, expected_error_type):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(TypeError):
+        bee.create_postage_batch(input_value, 17)
+
+
+def test_throw_error_if_too_big_depth():
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(TypeError):
+        bee.create_postage_batch("10", 256)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", request_options_assertions)
+def test_get_postage_batch(input_value, expected_error_type, test_batch_id):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(expected_error_type):
+        bee.get_postage_batch(test_batch_id, input_value)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", transaction_options_assertions)
+def test_get_postage_batch_transaction_assertions(input_value, expected_error_type):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(expected_error_type):
+        bee.get_postage_batch(input_value)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", request_options_assertions)
+def test_get_postage_batch_buckets(input_value, expected_error_type, test_batch_id):
+    with pytest.raises(expected_error_type):
+        bee = BeeDebug(MOCK_SERVER_URL, input_value)
+        bee.get_postage_batch_buckets(input_value, test_batch_id)
+
+
+@pytest.mark.parametrize("input_value, expected_error_type", batch_id_assertion_data)
+def test_get_postage_batch_buckets_batch_id_assertion(input_value, expected_error_type, test_batch_id):
+    bee = BeeDebug(MOCK_SERVER_URL)
+    with pytest.raises(expected_error_type):
+        bee.get_postage_batch_buckets(input_value)