From eb572055f86c0e93f12bab58435885d67265c2a0 Mon Sep 17 00:00:00 2001
From: Kevin Greenan <kmg@box.com>
Date: Mon, 9 Mar 2015 22:25:22 -0700
Subject: [PATCH] Example test case thst tests plumbing liberasurecode errors
 up into pyeclib

---
 test/test_pyeclib_api.py | 17 +++++++++++++++++
 test/test_pyeclib_c.py   |  1 -
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/test/test_pyeclib_api.py b/test/test_pyeclib_api.py
index 1131423..969fee1 100644
--- a/test/test_pyeclib_api.py
+++ b/test/test_pyeclib_api.py
@@ -30,6 +30,7 @@ import unittest
 from pyeclib.ec_iface import ECDriver, VALID_EC_TYPES, ECDriverError, \
     PyECLib_EC_Types
 from test_pyeclib_c import _available_backends
+import pyeclib_c
 
 if sys.version < '3':
     def b2i(b):
@@ -195,6 +196,14 @@ class TestPyECLibDriver(unittest.TestCase):
 #                pyeclib_driver.verify_stripe_metadata(fragment_metadata_list) == -1)
 #
 
+    def get_available_backend(self, k, m, ec_type, chksum_type = "inline_crc32"):
+      if ec_type[:11] == "flat_xor_hd":
+        return ECDriver(k=k, m=m, ec_type="flat_xor_hd", chksum_type=chksum_type)
+      elif ec_type in self.available_backends:
+        return ECDriver(k=k, m=m, ec_type=ec_type, chksum_type=chksum_type)
+      else:
+        return None
+
     def test_get_metadata_formatted(self):
         pyeclib_driver = ECDriver(k=12, m=2, ec_type="jerasure_rs_vand", chksum_type="inline_crc32")
         
@@ -486,6 +495,14 @@ class TestPyECLibDriver(unittest.TestCase):
                         reconstructed_fragments[0] == orig_fragments[
                             idxs_to_remove[0]])
 
+                    #
+                    # Test reconstructor with insufficient fragments
+                    #
+                    try:
+                      pyeclib_driver.reconstruct([fragments[0]], [])
+                    except pyeclib_c.error as e:
+                      self.assertTrue(e.message.find("Insufficient number of fragments") > -1) 
+
                     #
                     # Test decode with integrity checks
                     #
diff --git a/test/test_pyeclib_c.py b/test/test_pyeclib_c.py
index ddd7e27..c7ebb83 100644
--- a/test/test_pyeclib_c.py
+++ b/test/test_pyeclib_c.py
@@ -385,7 +385,6 @@ class TestPyECLib(unittest.TestCase):
                 print(("Unexpected required fragments list "
                        "(exp != req): %s != %s" %
                        (expected_fragments, required_fragments)))
-                #sys.exit(2)
 
         return success