1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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')
|
