[master 8/19] Update luks format tests to test for size information.

[dashea]

From: mulhern

Related: #56

Signed-off-by: mulhern 
---
 tests/formats_test/luks_test.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/tests/formats_test/luks_test.py b/tests/formats_test/luks_test.py
index c4389f0..76bb862 100755
--- a/tests/formats_test/luks_test.py
+++ b/tests/formats_test/luks_test.py
@@ -1,4 +1,5 @@
 import blivet.formats.luks as luks
+from blivet.size import Size

from tests import loopbackedtestcase

@@ -27,3 +28,36 @@ def testSimple(self):
         # test that the device can be closed again
         self.assertIsNone(self.fmt.teardown())
         self.assertFalse(self.fmt.status)
+
+    def testSize(self):
+        """ Test that sizes are calculated correctly. """
+        device = self.loopDevices[0]
+
+        # create the device
+        self.fmt.device = device
+        self.assertIsNone(self.fmt.create())
+
+        # the size is 0
+        self.assertEqual(self.fmt.size, Size(0))
+        self.assertEqual(self.fmt.currentSize, Size(0))
+        self.assertEqual(self.fmt.targetSize, Size(0))
+
+        # open the luks device
+        self.assertIsNone(self.fmt.setup())
+
+        # size is unchanged
+        self.assertEqual(self.fmt.size, Size(0))
+        self.assertEqual(self.fmt.currentSize, Size(0))
+        self.assertEqual(self.fmt.targetSize, Size(0))
+
+        # update the size info
+        self.fmt.updateSizeInfo()
+
+        # the size is greater than zero and less than the size of the device
+        self.assertLess(self.fmt.size, self.DEFAULT_STORE_SIZE)
+        self.assertGreater(self.fmt.size, Size(0))
+
+        self.assertEqual(self.fmt.currentSize, self.fmt.size)
+        self.assertEqual(self.fmt.targetSize, Size(0))
+
+        self.fmt.teardown()