def test_subscription_update(): """Make sure that the new status is written into the DB """ request = _get_fake_request('acct:john@doe', 'smith', True, True) print "request", request.POST with testConfig() as config: configure(config) with patch('h.auth.local.views.Subscriptions') as mock_subs: mock_subs.get_by_id = MagicMock() mock_subs.get_by_id.return_value = Mock(active=True) profile = ProfileController(request) profile.db = Mock() profile.db.add = MagicMock(name='add') profile.edit_profile() assert profile.db.add.called
def test_user_disabled(): """Check if the disabled user flag is set """ request = _get_fake_request("john", "doe") with testConfig() as config: configure(config) with patch("horus.models.UserMixin") as mock_user: with patch("horus.lib.FlashMessage") as mock_flash: with patch("h.auth.local.schemas.EditProfileSchema") as mock_schema: mock_schema.validator = MagicMock(name="validator") mock_user.get_user = MagicMock(side_effect=_good_password) profile = ProfileController(request) profile.User = mock_user profile.db = FakeDB() profile.db.add = MagicMock(name="add") profile.disable_user() assert profile.db.add.called