From 560e9adc6439e41bf680db044f008010a57e64e2 Mon Sep 17 00:00:00 2001 From: Adam Chainz Date: Tue, 29 Nov 2016 11:53:26 +0000 Subject: [PATCH] Tested overriding properties when inheriting from BaseStorage --- tests/wizard/test_basestorage.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/wizard/test_basestorage.py diff --git a/tests/wizard/test_basestorage.py b/tests/wizard/test_basestorage.py new file mode 100644 index 0000000..bbf509b --- /dev/null +++ b/tests/wizard/test_basestorage.py @@ -0,0 +1,20 @@ +from django.test import TestCase + +from formtools.wizard.storage.base import BaseStorage + + +class TestBaseStorage(TestCase): + + def test_subclass_can_override_get_current_step(self): + class MyStorage(BaseStorage): + def _get_current_step(self): + return 'foo' + + self.assertEqual(MyStorage('/').current_step, 'foo') + + def test_subclass_can_override_get_extra_data(self): + class MyStorage(BaseStorage): + def _get_extra_data(self): + return 'foo' + + self.assertEqual(MyStorage('/').extra_data, 'foo')