From d9f1bced2261646e6118f368619ef41cf7bbca30 Mon Sep 17 00:00:00 2001
From: 3hhh <tripleh@hackingthe.net>
Date: Sat, 11 Jul 2020 12:53:30 +0200
Subject: [PATCH] tests/callback: add rudimentary tests for the callback driver

---
 qubes/tests/__init__.py         |   1 +
 qubes/tests/storage_callback.py | 109 ++++++++++++++++++++++++++++++++
 setup.py                        |   1 +
 3 files changed, 111 insertions(+)
 create mode 100644 qubes/tests/storage_callback.py

diff --git a/qubes/tests/__init__.py b/qubes/tests/__init__.py
index 07e78ef1..9b8eafeb 100644
--- a/qubes/tests/__init__.py
+++ b/qubes/tests/__init__.py
@@ -1406,6 +1406,7 @@ def load_tests(loader, tests, pattern):  # pylint: disable=unused-argument
             'qubes.tests.storage_file',
             'qubes.tests.storage_reflink',
             'qubes.tests.storage_lvm',
+            'qubes.tests.storage_callback',
             'qubes.tests.storage_kernels',
             'qubes.tests.ext',
             'qubes.tests.vm.qubesvm',
diff --git a/qubes/tests/storage_callback.py b/qubes/tests/storage_callback.py
new file mode 100644
index 00000000..caf7669a
--- /dev/null
+++ b/qubes/tests/storage_callback.py
@@ -0,0 +1,109 @@
+#
+# The Qubes OS Project, http://www.qubes-os.org
+#
+# Copyright (C) 2020 David Hobach <david@hobach.de>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, see <https://www.gnu.org/licenses/>.
+#
+''' Tests for the callback storage driver.
+
+    They are mostly identical to the lvm storage driver tests.
+'''
+# pylint: disable=line-too-long
+
+import os
+import json
+import subprocess
+import qubes.tests
+import qubes.tests.storage
+import qubes.tests.storage_lvm
+from qubes.tests.storage_lvm import skipUnlessLvmPoolExists
+import qubes.storage.callback
+
+POOL_CLASS = qubes.storage.callback.CallbackPool
+VOLUME_CLASS = qubes.storage.callback.CallbackVolume
+POOL_CONF = {'name': 'test-callback',
+             'driver': 'callback',
+             'conf_id': 'utest-callback'}
+
+CB_CONF = '/etc/qubes_callback.json'
+
+CB_DATA = {'utest-callback': {
+                'bdriver': 'lvm_thin',
+                'bdriver_args': {
+                     'volume_group': qubes.tests.storage_lvm.DEFAULT_LVM_POOL.split('/')[0],
+                     'thin_pool':    qubes.tests.storage_lvm.DEFAULT_LVM_POOL.split('/')[1]
+                },
+                'description': 'For unit testing of the callback pool driver.'
+              }
+          }
+
+class CallbackBase:
+    ''' Mixin base class for callback tests. Has no base class. '''
+    bak_pool_class = None
+    bak_volume_class = None
+    bak_pool_conf = None
+
+    @classmethod
+    def setUpClass(cls):
+        CallbackBase.bak_pool_class = qubes.tests.storage_lvm.POOL_CLASS
+        CallbackBase.bak_volume_class = qubes.tests.storage_lvm.VOLUME_CLASS
+        CallbackBase.bak_pool_conf = qubes.tests.storage_lvm.POOL_CONF
+        qubes.tests.storage_lvm.POOL_CLASS = POOL_CLASS
+        qubes.tests.storage_lvm.VOLUME_CLASS = VOLUME_CLASS
+        qubes.tests.storage_lvm.POOL_CONF = POOL_CONF
+
+        assert not(os.path.exists(CB_CONF)), '%s must NOT exist. Please delete it, if you do not need it.' % CB_CONF
+
+        sudo = [] if os.getuid() == 0 else ['sudo']
+        subprocess.run(sudo + ['install', '-m', '666', '/dev/null', CB_CONF], check=True)
+
+        with open(CB_CONF, 'w') as outfile:
+            json.dump(CB_DATA, outfile)
+        super().setUpClass()
+
+    @classmethod
+    def tearDownClass(cls):
+        super().tearDownClass()
+        if CallbackBase.bak_pool_class:
+            qubes.tests.storage_lvm.POOL_CLASS = CallbackBase.bak_pool_class
+        if CallbackBase.bak_volume_class:
+            qubes.tests.storage_lvm.VOLUME_CLASS = CallbackBase.bak_volume_class
+        if CallbackBase.bak_pool_conf:
+            qubes.tests.storage_lvm.POOL_CONF = CallbackBase.bak_pool_conf
+
+        sudo = [] if os.getuid() == 0 else ['sudo']
+        subprocess.run(sudo + ['rm', '-f', CB_CONF], check=True)
+
+    def setUp(self):
+        super().setUp()
+        #tests from other pools will assume that they're fully initialized after calling __init__()
+        self.loop.run_until_complete(self.pool._assert_initialized())
+
+    def test_000_000_callback_test_init(self):
+        ''' Check whether the test init did work. '''
+        self.assertIsInstance(self.pool, qubes.storage.callback.CallbackPool)
+        self.assertTrue(os.path.isfile(CB_CONF))
+
+@skipUnlessLvmPoolExists
+class TC_00_CallbackPool(CallbackBase, qubes.tests.storage_lvm.TC_00_ThinPool):
+    pass
+
+@skipUnlessLvmPoolExists
+class TC_01_CallbackPool(CallbackBase, qubes.tests.storage_lvm.TC_01_ThinPool):
+    pass
+
+@skipUnlessLvmPoolExists
+class TC_02_cb_StorageHelpers(CallbackBase, qubes.tests.storage_lvm.TC_02_StorageHelpers):
+    pass
diff --git a/setup.py b/setup.py
index ea5beefa..e63e4114 100644
--- a/setup.py
+++ b/setup.py
@@ -88,5 +88,6 @@ if __name__ == '__main__':
                 'file-reflink = qubes.storage.reflink:ReflinkPool',
                 'linux-kernel = qubes.storage.kernels:LinuxKernel',
                 'lvm_thin = qubes.storage.lvm:ThinPool',
+                'callback = qubes.storage.callback:CallbackPool',
             ],
         })