1 files changed, 43 insertions, 0 deletions
diff --git a/charms/trusty/cassandra/tests/base.py b/charms/trusty/cassandra/tests/base.py
new file mode 100755
index 0000000..d308985
--- /dev/null
+++ b/charms/trusty/cassandra/tests/base.py
@@ -0,0 +1,43 @@
+# Copyright 2015 Canonical Ltd.
+#
+# This file is part of the Cassandra Charm for Juju.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranties of
+# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
+# PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import functools
+from itertools import count
+import unittest
+from unittest.mock import patch
+
+from testing.mocks import mock_charmhelpers
+
+patch = functools.partial(patch, autospec=True)  # autospec by default.
+
+
+class TestCaseBase(unittest.TestCase):
+    def setUp(self):
+        super(TestCaseBase, self).setUp()
+
+        mock_charmhelpers(self)
+
+        is_lxc = patch('helpers.is_lxc', return_value=False)
+        is_lxc.start()
+        self.addCleanup(is_lxc.stop)
+
+        emit = patch('helpers.emit')
+        emit.start()
+        self.addCleanup(emit.stop)
+
+        time = patch('time.time', side_effect=count(1))
+        time.start()
+        self.addCleanup(time.stop)