Tested overriding properties when inheriting from BaseStorage

This commit is contained in:
Adam Chainz
2016-11-29 11:53:26 +00:00
committed by Adam Chainz
parent bb97a8abd3
commit 560e9adc64

View File

@@ -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')