def setUp(self):
self.user_storage = MemoryUserStorage()
def test_specify_storage(self):
rollout = Rollout(feature_storage=MemoryFeatureStorage(),
user_storage=MemoryUserStorage())
rollout.add_feature(Feature('feature_for_all', groups=['ALL']))
user = self._add_user()
self.assertTrue(self.rollout.can(user, 'feature_for_all') is True)
