commit 304ee27e889d0b16c66a9b5d2faed379f8cad627
author Tao Bao <tbao@google.com> Mon Dec 19 11:01:38 2016 -0800
committer Tao Bao <tbao@google.com> Mon Dec 19 11:17:12 2016 -0800
tree ed119be7c7f7831784ebfd139e898a9ac0c96fb6
parent cbdc0bd096fdde4f92f1ec5bf65d6fc07e910a31

blockimgdiff.py: Add a test case for ReviseStashSize().

Bug: 33687949
Test: python -m unittest test_blockimgdiff
Change-Id: I31e42648d2c0e60b941908c098ce5ccc967861ca

tools/releasetools/blockimgdiff.py

diff --git a/tools/releasetools/blockimgdiff.py b/tools/releasetools/blockimgdiff.py
index 87cc72b..ded34b9 100644
--- a/tools/releasetools/blockimgdiff.py
+++ b/tools/releasetools/blockimgdiff.py
@@ -723,6 +723,7 @@
     num_of_bytes = new_blocks * self.tgt.blocksize
     print("  Total %d blocks (%d bytes) are packed as new blocks due to "
           "insufficient cache size." % (new_blocks, num_of_bytes))
+    return new_blocks
 
   def ComputePatches(self, prefix):
     print("Reticulating splines...")