153 lines
4.0 KiB
Ruby
153 lines
4.0 KiB
Ruby
require 'rails_helper'
|
||
require "s3_helper"
|
||
|
||
require_dependency 'backup'
|
||
|
||
describe Backup do
|
||
|
||
let(:b1) { Backup.new('backup1') }
|
||
let(:b2) { Backup.new('backup2') }
|
||
let(:b3) { Backup.new('backup3') }
|
||
|
||
before do
|
||
Backup.stubs(:all).returns([b1, b2, b3])
|
||
end
|
||
|
||
context '#remove_old' do
|
||
it "does nothing if there aren't more backups than the setting" do
|
||
SiteSetting.maximum_backups = 3
|
||
Backup.any_instance.expects(:remove).never
|
||
Backup.remove_old
|
||
end
|
||
|
||
it "calls remove on the backups over our limit" do
|
||
SiteSetting.maximum_backups = 1
|
||
b1.expects(:remove).never
|
||
b2.expects(:remove).once
|
||
b3.expects(:remove).once
|
||
Backup.remove_old
|
||
end
|
||
end
|
||
|
||
shared_context 's3 helpers' do
|
||
let(:client) { Aws::S3::Client.new(stub_responses: true) }
|
||
let(:resource) { Aws::S3::Resource.new(client: client) }
|
||
let!(:s3_bucket) { resource.bucket("s3-upload-bucket") }
|
||
let(:s3_helper) { b1.s3 }
|
||
|
||
before(:each) do
|
||
SiteSetting.s3_backup_bucket = "s3-upload-bucket"
|
||
SiteSetting.s3_access_key_id = "s3-access-key-id"
|
||
SiteSetting.s3_secret_access_key = "s3-secret-access-key"
|
||
end
|
||
end
|
||
|
||
context ".after_create_hook" do
|
||
context "when SiteSetting is true" do
|
||
include_context "s3 helpers"
|
||
|
||
before do
|
||
SiteSetting.enable_s3_backups = true
|
||
end
|
||
|
||
it "should upload the backup to S3 with the right paths" do
|
||
b1.path = 'some/path/backup.gz'
|
||
File.expects(:open).with(b1.path).yields(stub)
|
||
|
||
s3_helper.expects(:s3_bucket).returns(s3_bucket)
|
||
s3_object = stub
|
||
|
||
s3_bucket.expects(:object).with(b1.filename).returns(s3_object)
|
||
s3_object.expects(:upload_file)
|
||
|
||
b1.after_create_hook
|
||
end
|
||
|
||
context "when s3_backup_bucket includes folders path" do
|
||
before do
|
||
SiteSetting.s3_backup_bucket = "s3-upload-bucket/discourse-backups"
|
||
end
|
||
|
||
it "should upload the backup to S3 with the right paths" do
|
||
b1.path = 'some/path/backup.gz'
|
||
File.expects(:open).with(b1.path).yields(stub)
|
||
|
||
s3_helper.expects(:s3_bucket).returns(s3_bucket)
|
||
s3_object = stub
|
||
|
||
s3_bucket.expects(:object).with("discourse-backups/#{b1.filename}").returns(s3_object)
|
||
s3_object.expects(:upload_file)
|
||
|
||
b1.after_create_hook
|
||
end
|
||
end
|
||
end
|
||
|
||
it "calls upload_to_s3 if the SiteSetting is false" do
|
||
SiteSetting.enable_s3_backups = false
|
||
b1.expects(:upload_to_s3).never
|
||
b1.after_create_hook
|
||
end
|
||
end
|
||
|
||
context ".after_remove_hook" do
|
||
include_context "s3 helpers"
|
||
|
||
context "when SiteSetting is true" do
|
||
before do
|
||
SiteSetting.enable_s3_backups = true
|
||
end
|
||
|
||
it "should upload the backup to S3 with the right paths" do
|
||
s3_helper.expects(:s3_bucket).returns(s3_bucket)
|
||
s3_object = stub
|
||
|
||
s3_bucket.expects(:object).with(b1.filename).returns(s3_object)
|
||
s3_object.expects(:delete)
|
||
|
||
expect {
|
||
b1.after_remove_hook
|
||
}.to change {
|
||
Jobs::UpdateDiskSpace.jobs.size
|
||
}.by(0)
|
||
end
|
||
|
||
context "when s3_backup_bucket includes folders path" do
|
||
before do
|
||
SiteSetting.s3_backup_bucket = "s3-upload-bucket/discourse-backups"
|
||
end
|
||
|
||
it "should upload the backup to S3 with the right paths" do
|
||
s3_helper.expects(:s3_bucket).returns(s3_bucket)
|
||
s3_object = stub
|
||
|
||
s3_bucket.expects(:object).with("discourse-backups/#{b1.filename}").returns(s3_object)
|
||
s3_object.expects(:delete)
|
||
|
||
b1.after_remove_hook
|
||
end
|
||
end
|
||
end
|
||
|
||
context "when SiteSetting is false" do
|
||
before do
|
||
SiteSetting.enable_s3_backups = false
|
||
end
|
||
|
||
it "doesn’t call remove_from_s3" do
|
||
b1.expects(:remove_from_s3).never
|
||
b1.after_remove_hook
|
||
end
|
||
|
||
it "refreshes disk space if not backuping on s3" do
|
||
expect {
|
||
b1.after_remove_hook
|
||
}.to change {
|
||
Jobs::UpdateDiskSpace.jobs.size
|
||
}.by(1)
|
||
end
|
||
end
|
||
end
|
||
|
||
end
|