Add tests

Author: fafhrd91

sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/_decode.py

@@ -282,7 +282,7 @@ def _decode_array_large(buffer: memoryview) -> Tuple[memoryview, List[Any]]:
 
         if subconstructor == 0:
             composite_type = buffer[9]
-            buffer, descriptor = _DECODE_BY_CONSTRUCTOR[composite_type](buffer[9:])
+            buffer, descriptor = _DECODE_BY_CONSTRUCTOR[composite_type](buffer[10:])
             subconstructor = buffer[0]
             buffer = buffer[1:]
             for i in range(count):
@@ -453,11 +453,11 @@ def decode_empty_frame(header: memoryview) -> Tuple[int, bytes]:
     130: described.DescribedFloat,
     131: described.DescribedInt,
     160: described.DescribedBytes,
-    161: described.DescribedStr,
-    163: described.DescribedStr,
+    161: described.DescribedBytes,
+    163: described.DescribedBytes,
     176: described.DescribedBytes,
-    177: described.DescribedStr,
-    179: described.DescribedStr,
+    177: described.DescribedBytes,
+    179: described.DescribedBytes,
     192: described.DescribedList,
     193: described.DescribedDict,
     208: described.DescribedList,

sdk/eventhub/azure-eventhub/azure/eventhub/_pyamqp/described.py

@@ -16,10 +16,6 @@ class DescribedInt(Described, int):
 
 
 class DescribedFloat(Described, float):
-        pass
-
-
-class DescribedStr(Described, str):
     pass
 
 

sdk/eventhub/azure-eventhub/tests/pyamqp_tests/unittest/test_decode.py

@@ -1,5 +1,5 @@
 import pytest
-from azure.eventhub._pyamqp._decode import _decode_decimal128
+from azure.eventhub._pyamqp._decode import _decode_decimal128, _decode_described, _decode_array_small, _decode_array_large
 from decimal import Decimal
 
 
@@ -18,3 +18,30 @@ def test_decimal_decode(value, expected):
     assert output[1] == expected
 
 
+def test_described():
+    value = b"\x80\0\0\x017\0\0\x07\xd3\xd0\0\0\0\x12\0\0\0\x02\xa1\ntest/topicP\0"
+    buffer, output = _decode_described(memoryview(value))
+    assert output.descriptor == 1335734831059
+    assert output == [b'test/topic', 0]
+
+
+def test_array_of_described():
+    value = b"\0\x03\0\x80\0\0\x017\0\0\x07\xd4\xd0\0\0\0\x0c\0\0\0\x02\xa1\x02n1\xa1\x02v1\0\0\0\x0c\0\0\0\x02\xa1\x02n2\xa1\x02v2\0\0\0\n\0\0\0\x02\xa1\x02n1\xa1\0"
+
+    buffer, output = _decode_array_small(memoryview(value))
+    assert output == [[b'n1', b'v1'], [b'n2', b'v2'], [b'n1', b'']]
+    assert output[0].descriptor == 1335734831060
+    assert output[1].descriptor == 1335734831060
+    assert output[2].descriptor == 1335734831060
+
+
+def test_array_of_described_large():
+    value = b"\0\0\x0e\x0f\0\0\x01\0\0\x80\0\0\x017\0\0\x07\xd4\xd0"
+    for i in range(256):
+        value += b"\0\0\0\n\0\0\0\x02\xa1\x01n\xa1\x01v"
+
+    buffer, output = _decode_array_large(memoryview(value))
+    assert len(output) == 256
+    for i in range(256):
+        assert output[i] == [b'n', b'v']
+        assert output[i].descriptor == 1335734831060