discourse/spec/controllers/email_controller_spec.rb

require 'rails_helper'

describe EmailController do

  context '.preferences_redirect' do

    it 'requires you to be logged in' do
      expect { get :preferences_redirect, format: :json }.to raise_error(Discourse::NotLoggedIn)
    end

    context 'when logged in' do
      let!(:user) { log_in }

      it 'redirects to your user preferences' do
        get :preferences_redirect, format: :json
        expect(response).to redirect_to("/u/#{user.username}/preferences")
      end
    end

  end

  context '.perform unsubscribe' do
    it 'raises not found on invalid key' do
      post :perform_unsubscribe, params: { key: "123" }, format: :json
      expect(response.status).to eq(404)
    end

    it 'can fully unsubscribe' do
      user = Fabricate(:user)
      key = UnsubscribeKey.create_key_for(user, "all")

      user.user_option.update_columns(email_always: true,
                                      email_digests: true,
                                      email_direct: true,
                                      email_private_messages: true)

      post :perform_unsubscribe,
        params: { key: key, unsubscribe_all: "1" },
        format: :json

      expect(response.status).to eq(302)

      user.user_option.reload

      expect(user.user_option.email_always).to eq(false)
      expect(user.user_option.email_digests).to eq(false)
      expect(user.user_option.email_direct).to eq(false)
      expect(user.user_option.email_private_messages).to eq(false)

    end

    it 'can disable mailing list' do
      user = Fabricate(:user)
      key = UnsubscribeKey.create_key_for(user, "all")

      user.user_option.update_columns(mailing_list_mode: true)

      post :perform_unsubscribe,
        params: { key: key, disable_mailing_list: "1" },
        format: :json

      expect(response.status).to eq(302)

      user.user_option.reload

      expect(user.user_option.mailing_list_mode).to eq(false)
    end

    it 'can disable digest' do
      user = Fabricate(:user)
      key = UnsubscribeKey.create_key_for(user, "all")

      user.user_option.update_columns(email_digests: true)

      post :perform_unsubscribe,
        params: { key: key, disable_digest_emails: "1" },
        format: :json

      expect(response.status).to eq(302)

      user.user_option.reload

      expect(user.user_option.email_digests).to eq(false)
    end

    it 'can unwatch topic' do
      p = Fabricate(:post)
      key = UnsubscribeKey.create_key_for(p.user, p)

      TopicUser.change(p.user_id, p.topic_id, notification_level: TopicUser.notification_levels[:watching])

      post :perform_unsubscribe,
        params: { key: key, unwatch_topic: "1" },
        format: :json

      expect(response.status).to eq(302)

      expect(TopicUser.get(p.topic, p.user).notification_level).to eq(TopicUser.notification_levels[:tracking])
    end

    it 'can mute topic' do
      p = Fabricate(:post)
      key = UnsubscribeKey.create_key_for(p.user, p)

      TopicUser.change(p.user_id, p.topic_id, notification_level: TopicUser.notification_levels[:watching])

      post :perform_unsubscribe,
        params: { key: key, mute_topic: "1" },
        format: :json

      expect(response.status).to eq(302)

      expect(TopicUser.get(p.topic, p.user).notification_level).to eq(TopicUser.notification_levels[:muted])
    end

    it 'can unwatch category' do
      p = Fabricate(:post)
      key = UnsubscribeKey.create_key_for(p.user, p)

      cu = CategoryUser.create!(user_id: p.user.id,
                                category_id: p.topic.category_id,
                                notification_level: CategoryUser.notification_levels[:watching])

      post :perform_unsubscribe,
        params: { key: key, unwatch_category: "1" },
        format: :json

      expect(response.status).to eq(302)

      expect(CategoryUser.find_by(id: cu.id)).to eq(nil)
    end

    it 'can unwatch first post from category' do
      p = Fabricate(:post)
      key = UnsubscribeKey.create_key_for(p.user, p)

      cu = CategoryUser.create!(user_id: p.user.id,
                                category_id: p.topic.category_id,
                                notification_level: CategoryUser.notification_levels[:watching_first_post])

      post :perform_unsubscribe,
        params: { key: key, unwatch_category: "1" },
        format: :json

      expect(response.status).to eq(302)

      expect(CategoryUser.find_by(id: cu.id)).to eq(nil)
    end
  end

  context '.unsubscribe' do

    render_views

    it 'displays logo ut button if wrong user logged in' do
      log_in_user Fabricate(:admin)
      user = Fabricate(:user)
      key = UnsubscribeKey.create_key_for(user, "digest")

      get :unsubscribe, params: { key: key }

      expect(response.body).to include(I18n.t("unsubscribe.log_out"))
      expect(response.body).to include(I18n.t("unsubscribe.different_user_description"))
    end

    it 'displays not found if key is not found' do
      get :unsubscribe, params: { key: SecureRandom.hex }
      expect(response.body).to include(CGI.escapeHTML(I18n.t("unsubscribe.not_found_description")))
    end

    it 'correctly handles mailing list mode' do

      user = Fabricate(:user)
      key = UnsubscribeKey.create_key_for(user, "digest")

      user.user_option.update_columns(mailing_list_mode: true)

      get :unsubscribe, params: { key: key }
      expect(response.body).to include(I18n.t("unsubscribe.mailing_list_mode"))

      SiteSetting.disable_mailing_list_mode = true

      get :unsubscribe, params: { key: key }
      expect(response.body).not_to include(I18n.t("unsubscribe.mailing_list_mode"))

      user.user_option.update_columns(mailing_list_mode: false)
      SiteSetting.disable_mailing_list_mode = false

      get :unsubscribe, params: { key: key }
      expect(response.body).not_to include(I18n.t("unsubscribe.mailing_list_mode"))

    end

    it 'correctly handles digest unsubscribe' do

      user = Fabricate(:user)
      user.user_option.update_columns(email_digests: false)
      key = UnsubscribeKey.create_key_for(user, "digest")

      # because we are type digest we will always show digest and it will be selected
      get :unsubscribe, params: { key: key }
      expect(response.body).to include(I18n.t("unsubscribe.disable_digest_emails"))

      source = Nokogiri::HTML::fragment(response.body)
      expect(source.css("#disable_digest_emails")[0]["checked"]).to eq("checked")

      SiteSetting.disable_digest_emails = true

      get :unsubscribe, params: { key: key }
      expect(response.body).not_to include(I18n.t("unsubscribe.disable_digest_emails"))

      SiteSetting.disable_digest_emails = false
      key = UnsubscribeKey.create_key_for(user, "not_digest")

      get :unsubscribe, params: { key: key }
      expect(response.body).to include(I18n.t("unsubscribe.disable_digest_emails"))
    end

    it 'correctly handles watched categories' do
      post = Fabricate(:post)
      user = post.user
      cu = CategoryUser.create!(user_id: user.id,
                                category_id: post.topic.category_id,
                                notification_level: CategoryUser.notification_levels[:watching])

      key = UnsubscribeKey.create_key_for(user, post)
      get :unsubscribe, params: { key: key }
      expect(response.body).to include("unwatch_category")

      cu.destroy!

      get :unsubscribe, params: { key: key }
      expect(response.body).not_to include("unwatch_category")

    end

    it 'correctly handles watched first post categories' do
      post = Fabricate(:post)
      user = post.user
      cu = CategoryUser.create!(user_id: user.id,
                                category_id: post.topic.category_id,
                                notification_level: CategoryUser.notification_levels[:watching_first_post])

      key = UnsubscribeKey.create_key_for(user, post)
      get :unsubscribe, params: { key: key }
      expect(response.body).to include("unwatch_category")

      cu.destroy!

      get :unsubscribe, params: { key: key }
      expect(response.body).not_to include("unwatch_category")

    end
  end

end