552 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			552 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| require 'spec_helper'
 | |
| shared_examples 'content not cached without revalidation' do
 | |
|   it 'ensures content will not be cached without revalidation' do
 | |
|     expect(subject['Cache-Control']).to eq('max-age=0, private, must-revalidate')
 | |
|   end
 | |
| end
 | |
| 
 | |
| describe UploadsController do
 | |
|   let!(:user) { create(:user, avatar: fixture_file_upload(Rails.root + "spec/fixtures/dk.png", "image/png")) }
 | |
| 
 | |
|   describe 'POST create' do
 | |
|     let(:model)   { 'personal_snippet' }
 | |
|     let(:snippet) { create(:personal_snippet, :public) }
 | |
|     let(:jpg)     { fixture_file_upload(Rails.root + 'spec/fixtures/rails_sample.jpg', 'image/jpg') }
 | |
|     let(:txt)     { fixture_file_upload(Rails.root + 'spec/fixtures/doc_sample.txt', 'text/plain') }
 | |
| 
 | |
|     context 'when a user does not have permissions to upload a file' do
 | |
|       it "returns 401 when the user is not logged in" do
 | |
|         post :create, model: model, id: snippet.id, format: :json
 | |
| 
 | |
|         expect(response).to have_gitlab_http_status(401)
 | |
|       end
 | |
| 
 | |
|       it "returns 404 when user can't comment on a snippet" do
 | |
|         private_snippet = create(:personal_snippet, :private)
 | |
| 
 | |
|         sign_in(user)
 | |
|         post :create, model: model, id: private_snippet.id, format: :json
 | |
| 
 | |
|         expect(response).to have_gitlab_http_status(404)
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context 'when a user is logged in' do
 | |
|       before do
 | |
|         sign_in(user)
 | |
|       end
 | |
| 
 | |
|       it "returns an error without file" do
 | |
|         post :create, model: model, id: snippet.id, format: :json
 | |
| 
 | |
|         expect(response).to have_gitlab_http_status(422)
 | |
|       end
 | |
| 
 | |
|       it "returns an error with invalid model" do
 | |
|         expect { post :create, model: 'invalid', id: snippet.id, format: :json }
 | |
|         .to raise_error(ActionController::UrlGenerationError)
 | |
|       end
 | |
| 
 | |
|       it "returns 404 status when object not found" do
 | |
|         post :create, model: model, id: 9999, format: :json
 | |
| 
 | |
|         expect(response).to have_gitlab_http_status(404)
 | |
|       end
 | |
| 
 | |
|       context 'with valid image' do
 | |
|         before do
 | |
|           post :create, model: 'personal_snippet', id: snippet.id, file: jpg, format: :json
 | |
|         end
 | |
| 
 | |
|         it 'returns a content with original filename, new link, and correct type.' do
 | |
|           expect(response.body).to match '\"alt\":\"rails_sample\"'
 | |
|           expect(response.body).to match "\"url\":\"/uploads"
 | |
|         end
 | |
| 
 | |
|         it 'creates a corresponding Upload record' do
 | |
|           upload = Upload.last
 | |
| 
 | |
|           aggregate_failures do
 | |
|             expect(upload).to exist
 | |
|             expect(upload.model).to eq snippet
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context 'with valid non-image file' do
 | |
|         before do
 | |
|           post :create, model: 'personal_snippet', id: snippet.id, file: txt, format: :json
 | |
|         end
 | |
| 
 | |
|         it 'returns a content with original filename, new link, and correct type.' do
 | |
|           expect(response.body).to match '\"alt\":\"doc_sample.txt\"'
 | |
|           expect(response.body).to match "\"url\":\"/uploads"
 | |
|         end
 | |
| 
 | |
|         it 'creates a corresponding Upload record' do
 | |
|           upload = Upload.last
 | |
| 
 | |
|           aggregate_failures do
 | |
|             expect(upload).to exist
 | |
|             expect(upload.model).to eq snippet
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context 'temporal with valid image' do
 | |
|         subject do
 | |
|           post :create, model: 'personal_snippet', file: jpg, format: :json
 | |
|         end
 | |
| 
 | |
|         it 'returns a content with original filename, new link, and correct type.' do
 | |
|           subject
 | |
| 
 | |
|           expect(response.body).to match '\"alt\":\"rails_sample\"'
 | |
|           expect(response.body).to match "\"url\":\"/uploads/-/system/temp"
 | |
|         end
 | |
| 
 | |
|         it 'does not create an Upload record' do
 | |
|           expect { subject }.not_to change { Upload.count }
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context 'temporal with valid non-image file' do
 | |
|         subject do
 | |
|           post :create, model: 'personal_snippet', file: txt, format: :json
 | |
|         end
 | |
| 
 | |
|         it 'returns a content with original filename, new link, and correct type.' do
 | |
|           subject
 | |
| 
 | |
|           expect(response.body).to match '\"alt\":\"doc_sample.txt\"'
 | |
|           expect(response.body).to match "\"url\":\"/uploads/-/system/temp"
 | |
|         end
 | |
| 
 | |
|         it 'does not create an Upload record' do
 | |
|           expect { subject }.not_to change { Upload.count }
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   describe "GET show" do
 | |
|     context 'Content-Disposition security measures' do
 | |
|       let(:project) { create(:project, :public) }
 | |
| 
 | |
|       context 'for PNG files' do
 | |
|         it 'returns Content-Disposition: inline' do
 | |
|           note = create(:note, :with_attachment, project: project)
 | |
|           get :show, model: 'note', mounted_as: 'attachment', id: note.id, filename: 'image.png'
 | |
| 
 | |
|           expect(response['Content-Disposition']).to start_with('inline;')
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context 'for SVG files' do
 | |
|         it 'returns Content-Disposition: attachment' do
 | |
|           note = create(:note, :with_svg_attachment, project: project)
 | |
|           get :show, model: 'note', mounted_as: 'attachment', id: note.id, filename: 'image.svg'
 | |
| 
 | |
|           expect(response['Content-Disposition']).to start_with('attachment;')
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context "when viewing a user avatar" do
 | |
|       context "when signed in" do
 | |
|         before do
 | |
|           sign_in(user)
 | |
|         end
 | |
| 
 | |
|         context "when the user is blocked" do
 | |
|           before do
 | |
|             user.block
 | |
|           end
 | |
| 
 | |
|           it "redirects to the sign in page" do
 | |
|             get :show, model: "user", mounted_as: "avatar", id: user.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to redirect_to(new_user_session_path)
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when the user isn't blocked" do
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "user", mounted_as: "avatar", id: user.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'user', mounted_as: 'avatar', id: user.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context "when not signed in" do
 | |
|         it "responds with status 200" do
 | |
|           get :show, model: "user", mounted_as: "avatar", id: user.id, filename: "image.png"
 | |
| 
 | |
|           expect(response).to have_gitlab_http_status(200)
 | |
|         end
 | |
| 
 | |
|         it_behaves_like 'content not cached without revalidation' do
 | |
|           subject do
 | |
|             get :show, model: 'user', mounted_as: 'avatar', id: user.id, filename: 'image.png'
 | |
|             response
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context "when viewing a project avatar" do
 | |
|       let!(:project) { create(:project, avatar: fixture_file_upload(Rails.root + "spec/fixtures/dk.png", "image/png")) }
 | |
| 
 | |
|       context "when the project is public" do
 | |
|         before do
 | |
|           project.update_attribute(:visibility_level, Project::PUBLIC)
 | |
|         end
 | |
| 
 | |
|         context "when not signed in" do
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'project', mounted_as: 'avatar', id: project.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'project', mounted_as: 'avatar', id: project.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context "when the project is private" do
 | |
|         before do
 | |
|           project.update_attribute(:visibility_level, Project::PRIVATE)
 | |
|         end
 | |
| 
 | |
|         context "when not signed in" do
 | |
|           it "redirects to the sign in page" do
 | |
|             get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to redirect_to(new_user_session_path)
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           context "when the user has access to the project" do
 | |
|             before do
 | |
|               project.add_master(user)
 | |
|             end
 | |
| 
 | |
|             context "when the user is blocked" do
 | |
|               before do
 | |
|                 user.block
 | |
|                 project.add_master(user)
 | |
|               end
 | |
| 
 | |
|               it "redirects to the sign in page" do
 | |
|                 get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to redirect_to(new_user_session_path)
 | |
|               end
 | |
|             end
 | |
| 
 | |
|             context "when the user isn't blocked" do
 | |
|               it "responds with status 200" do
 | |
|                 get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to have_gitlab_http_status(200)
 | |
|               end
 | |
| 
 | |
|               it_behaves_like 'content not cached without revalidation' do
 | |
|                 subject do
 | |
|                   get :show, model: 'project', mounted_as: 'avatar', id: project.id, filename: 'image.png'
 | |
|                   response
 | |
|                 end
 | |
|               end
 | |
|             end
 | |
|           end
 | |
| 
 | |
|           context "when the user doesn't have access to the project" do
 | |
|             it "responds with status 404" do
 | |
|               get :show, model: "project", mounted_as: "avatar", id: project.id, filename: "image.png"
 | |
| 
 | |
|               expect(response).to have_gitlab_http_status(404)
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context "when viewing a group avatar" do
 | |
|       let!(:group)   { create(:group, avatar: fixture_file_upload(Rails.root + "spec/fixtures/dk.png", "image/png")) }
 | |
| 
 | |
|       context "when the group is public" do
 | |
|         context "when not signed in" do
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "group", mounted_as: "avatar", id: group.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'group', mounted_as: 'avatar', id: group.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "group", mounted_as: "avatar", id: group.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'group', mounted_as: 'avatar', id: group.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context "when the group is private" do
 | |
|         before do
 | |
|           group.update_attribute(:visibility_level, Gitlab::VisibilityLevel::PRIVATE)
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           context "when the user has access to the project" do
 | |
|             before do
 | |
|               group.add_developer(user)
 | |
|             end
 | |
| 
 | |
|             context "when the user is blocked" do
 | |
|               before do
 | |
|                 user.block
 | |
|               end
 | |
| 
 | |
|               it "redirects to the sign in page" do
 | |
|                 get :show, model: "group", mounted_as: "avatar", id: group.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to redirect_to(new_user_session_path)
 | |
|               end
 | |
|             end
 | |
| 
 | |
|             context "when the user isn't blocked" do
 | |
|               it "responds with status 200" do
 | |
|                 get :show, model: "group", mounted_as: "avatar", id: group.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to have_gitlab_http_status(200)
 | |
|               end
 | |
| 
 | |
|               it_behaves_like 'content not cached without revalidation' do
 | |
|                 subject do
 | |
|                   get :show, model: 'group', mounted_as: 'avatar', id: group.id, filename: 'image.png'
 | |
|                   response
 | |
|                 end
 | |
|               end
 | |
|             end
 | |
|           end
 | |
| 
 | |
|           context "when the user doesn't have access to the project" do
 | |
|             it "responds with status 404" do
 | |
|               get :show, model: "group", mounted_as: "avatar", id: group.id, filename: "image.png"
 | |
| 
 | |
|               expect(response).to have_gitlab_http_status(404)
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context "when viewing a note attachment" do
 | |
|       let!(:note) { create(:note, :with_attachment) }
 | |
|       let(:project) { note.project }
 | |
| 
 | |
|       context "when the project is public" do
 | |
|         before do
 | |
|           project.update_attribute(:visibility_level, Project::PUBLIC)
 | |
|         end
 | |
| 
 | |
|         context "when not signed in" do
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'note', mounted_as: 'attachment', id: note.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           it "responds with status 200" do
 | |
|             get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'note', mounted_as: 'attachment', id: note.id, filename: 'image.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context "when the project is private" do
 | |
|         before do
 | |
|           project.update_attribute(:visibility_level, Project::PRIVATE)
 | |
|         end
 | |
| 
 | |
|         context "when not signed in" do
 | |
|           it "redirects to the sign in page" do
 | |
|             get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|             expect(response).to redirect_to(new_user_session_path)
 | |
|           end
 | |
|         end
 | |
| 
 | |
|         context "when signed in" do
 | |
|           before do
 | |
|             sign_in(user)
 | |
|           end
 | |
| 
 | |
|           context "when the user has access to the project" do
 | |
|             before do
 | |
|               project.add_master(user)
 | |
|             end
 | |
| 
 | |
|             context "when the user is blocked" do
 | |
|               before do
 | |
|                 user.block
 | |
|                 project.add_master(user)
 | |
|               end
 | |
| 
 | |
|               it "redirects to the sign in page" do
 | |
|                 get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to redirect_to(new_user_session_path)
 | |
|               end
 | |
|             end
 | |
| 
 | |
|             context "when the user isn't blocked" do
 | |
|               it "responds with status 200" do
 | |
|                 get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|                 expect(response).to have_gitlab_http_status(200)
 | |
|               end
 | |
| 
 | |
|               it_behaves_like 'content not cached without revalidation' do
 | |
|                 subject do
 | |
|                   get :show, model: 'note', mounted_as: 'attachment', id: note.id, filename: 'image.png'
 | |
|                   response
 | |
|                 end
 | |
|               end
 | |
|             end
 | |
|           end
 | |
| 
 | |
|           context "when the user doesn't have access to the project" do
 | |
|             it "responds with status 404" do
 | |
|               get :show, model: "note", mounted_as: "attachment", id: note.id, filename: "image.png"
 | |
| 
 | |
|               expect(response).to have_gitlab_http_status(404)
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     context 'Appearance' do
 | |
|       context 'when viewing a custom header logo' do
 | |
|         let!(:appearance) { create :appearance, header_logo: fixture_file_upload(Rails.root.join('spec/fixtures/dk.png'), 'image/png') }
 | |
| 
 | |
|         context 'when not signed in' do
 | |
|           it 'responds with status 200' do
 | |
|             get :show, model: 'appearance', mounted_as: 'header_logo', id: appearance.id, filename: 'dk.png'
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'appearance', mounted_as: 'header_logo', id: appearance.id, filename: 'dk.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       context 'when viewing a custom logo' do
 | |
|         let!(:appearance) { create :appearance, logo: fixture_file_upload(Rails.root.join('spec/fixtures/dk.png'), 'image/png') }
 | |
| 
 | |
|         context 'when not signed in' do
 | |
|           it 'responds with status 200' do
 | |
|             get :show, model: 'appearance', mounted_as: 'logo', id: appearance.id, filename: 'dk.png'
 | |
| 
 | |
|             expect(response).to have_gitlab_http_status(200)
 | |
|           end
 | |
| 
 | |
|           it_behaves_like 'content not cached without revalidation' do
 | |
|             subject do
 | |
|               get :show, model: 'appearance', mounted_as: 'logo', id: appearance.id, filename: 'dk.png'
 | |
|               response
 | |
|             end
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| end
 |