mirror of
https://github.com/discourse/discourse.git
synced 2024-12-16 01:53:45 +08:00
103 lines
3.7 KiB
Ruby
103 lines
3.7 KiB
Ruby
require "rails_helper"
|
|
|
|
describe ExportCsvController do
|
|
let(:export_filename) { "user-archive-codinghorror-150115-234817-999.csv.gz" }
|
|
|
|
context "while not logged in" do
|
|
describe ".download" do
|
|
it "returns 404 when the unauthorized user tries to export csv file" do
|
|
get :show, id: export_filename
|
|
expect(response.status).to eq(404)
|
|
end
|
|
end
|
|
end
|
|
|
|
context "while logged in as normal user" do
|
|
before { @user = log_in(:user) }
|
|
|
|
describe ".export_entity" do
|
|
it "enqueues export job" do
|
|
Jobs.expects(:enqueue).with(:export_csv_file, has_entries(entity: "user_archive", user_id: @user.id))
|
|
xhr :post, :export_entity, entity: "user_archive"
|
|
expect(response).to be_success
|
|
end
|
|
|
|
it "should not enqueue export job if rate limit is reached" do
|
|
Jobs::ExportCsvFile.any_instance.expects(:execute).never
|
|
UserExport.create(file_name: "user-archive-codinghorror-150116-003249", user_id: @user.id)
|
|
xhr :post, :export_entity, entity: "user_archive"
|
|
expect(response).not_to be_success
|
|
end
|
|
|
|
it "returns 404 when normal user tries to export admin entity" do
|
|
xhr :post, :export_entity, entity: "staff_action"
|
|
expect(response).not_to be_success
|
|
end
|
|
end
|
|
|
|
describe ".download" do
|
|
it "uses send_file to transmit the export file" do
|
|
file = UserExport.create(file_name: "user-archive-codinghorror-150116-003249", user_id: @user.id)
|
|
file_name = "user-archive-codinghorror-150116-003249-#{file.id}.csv.gz"
|
|
controller.stubs(:render)
|
|
export = UserExport.new()
|
|
UserExport.expects(:get_download_path).with(file_name).returns(export)
|
|
subject.expects(:send_file).with(export)
|
|
get :show, id: file_name
|
|
expect(response).to be_success
|
|
end
|
|
|
|
it "returns 404 when the user tries to export another user's csv file" do
|
|
get :show, id: export_filename
|
|
expect(response).to be_not_found
|
|
end
|
|
|
|
it "returns 404 when the export file does not exist" do
|
|
UserExport.expects(:get_download_path).returns(nil)
|
|
get :show, id: export_filename
|
|
expect(response).to be_not_found
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
context "while logged in as an admin" do
|
|
before { @admin = log_in(:admin) }
|
|
|
|
describe ".export_entity" do
|
|
it "enqueues export job" do
|
|
Jobs.expects(:enqueue).with(:export_csv_file, has_entries(entity: "staff_action", user_id: @admin.id))
|
|
xhr :post, :export_entity, entity: "staff_action"
|
|
expect(response).to be_success
|
|
end
|
|
|
|
it "should not rate limit export for staff" do
|
|
Jobs.expects(:enqueue).with(:export_csv_file, has_entries(entity: "staff_action", user_id: @admin.id))
|
|
UserExport.create(file_name: "screened-email-150116-010145", user_id: @admin.id)
|
|
xhr :post, :export_entity, entity: "staff_action"
|
|
expect(response).to be_success
|
|
end
|
|
end
|
|
|
|
describe ".download" do
|
|
it "uses send_file to transmit the export file" do
|
|
file = UserExport.create(file_name: "screened-email-150116-010145", user_id: @admin.id)
|
|
file_name = "screened-email-150116-010145-#{file.id}.csv.gz"
|
|
controller.stubs(:render)
|
|
export = UserExport.new()
|
|
UserExport.expects(:get_download_path).with(file_name).returns(export)
|
|
subject.expects(:send_file).with(export)
|
|
get :show, id: file_name
|
|
expect(response).to be_success
|
|
end
|
|
|
|
it "returns 404 when the export file does not exist" do
|
|
UserExport.expects(:get_download_path).returns(nil)
|
|
get :show, id: export_filename
|
|
expect(response).to be_not_found
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|