mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
49 lines
1.6 KiB
Ruby
49 lines
1.6 KiB
Ruby
require "tmpdir"
|
|
require "service/shared_service_tests"
|
|
|
|
class ActiveStorage::Service::MirrorServiceTest < ActiveSupport::TestCase
|
|
PRIMARY_DISK_SERVICE = ActiveStorage::Service.configure(:Disk, root: File.join(Dir.tmpdir, "active_storage"))
|
|
SECONDARY_DISK_SERVICE = ActiveStorage::Service.configure(:Disk, root: File.join(Dir.tmpdir, "active_storage_mirror"))
|
|
|
|
SERVICE = ActiveStorage::Service.configure :Mirror, services: [ PRIMARY_DISK_SERVICE, SECONDARY_DISK_SERVICE ]
|
|
|
|
include ActiveStorage::Service::SharedServiceTests
|
|
|
|
test "uploading to all services" do
|
|
begin
|
|
data = "Something else entirely!"
|
|
key = upload(data, to: @service)
|
|
|
|
assert_equal data, PRIMARY_DISK_SERVICE.download(key)
|
|
assert_equal data, SECONDARY_DISK_SERVICE.download(key)
|
|
ensure
|
|
@service.delete key
|
|
end
|
|
end
|
|
|
|
test "downloading from primary service" do
|
|
data = "Something else entirely!"
|
|
key = upload(data, to: PRIMARY_DISK_SERVICE)
|
|
|
|
assert_equal data, @service.download(key)
|
|
end
|
|
|
|
test "deleting from all services" do
|
|
@service.delete FIXTURE_KEY
|
|
assert_not PRIMARY_DISK_SERVICE.exist?(FIXTURE_KEY)
|
|
assert_not SECONDARY_DISK_SERVICE.exist?(FIXTURE_KEY)
|
|
end
|
|
|
|
test "URL generation in primary service" do
|
|
travel_to Time.now do
|
|
assert_equal PRIMARY_DISK_SERVICE.url(FIXTURE_KEY, expires_in: 2.minutes, disposition: :inline, filename: "test.txt"),
|
|
@service.url(FIXTURE_KEY, expires_in: 2.minutes, disposition: :inline, filename: "test.txt")
|
|
end
|
|
end
|
|
|
|
def upload(data, to:)
|
|
SecureRandom.base58(24).tap do |key|
|
|
@service.upload key, StringIO.new(data)
|
|
end
|
|
end
|
|
end
|