Check integrity after uploads

1 files changed, 15 insertions, 2 deletions

diff --git a/test/service/shared_service_tests.rb b/test/service/shared_service_tests.rb
index 3676272e27..b4c888e77c 100644
--- a/test/service/shared_service_tests.rb
+++ b/test/service/shared_service_tests.rb
@@ -26,11 +26,11 @@ module ActiveStorage::Service::SharedServiceTests
       FIXTURE_FILE.rewind
     end
 
-    test "uploading" do
+    test "uploading with integrity" do
       begin
         key  = SecureRandom.base58(24)
         data = "Something else entirely!"
-        @service.upload(key, StringIO.new(data))
+        @service.upload(key, StringIO.new(data), checksum: Digest::MD5.base64digest(data))
 
         assert_equal data, @service.download(key)
       ensure
@@ -38,6 +38,19 @@ module ActiveStorage::Service::SharedServiceTests
       end
     end
 
+    test "upload without integrity" do
+      begin
+        key  = SecureRandom.base58(24)
+        data = "Something else entirely!"
+
+        assert_raises(ActiveStorage::IntegrityError) do
+          @service.upload(key, StringIO.new(data), checksum: "BAD_CHECKSUM")
+        end
+      ensure
+        @service.delete key
+      end
+    end
+
     test "downloading" do
       assert_equal FIXTURE_FILE.read, @service.download(FIXTURE_KEY)
     end