14484: Adds test for pdh grouping functionality in the container model

[arvados.git] / services / api / test / unit / container_test.rb

diff --git a/services/api/test/unit/container_test.rb b/services/api/test/unit/container_test.rb
index 2b7fda8d7f1493725c773ecd0ca9051634cdc9fd..783d2a985fa569d8a2cd0275a28441e60636c2c9 100644 (file)
--- a/services/api/test/unit/container_test.rb
+++ b/services/api/test/unit/container_test.rb
@@ -962,8 +962,14 @@ class ContainerTest < ActiveSupport::TestCase
   test "pdh_grouping_by_manifest_size" do
     batch_size_max = 200
     pdhs_in = ['x1+30', 'x2+30', 'x3+201', 'x4+100', 'x5+100']
+    pdh_lambda = lambda { |last_pdh, &block|
+      pdhs = pdhs_in.select{|pdh| pdh > last_pdh} 
+      pdhs.each do |p|
+        block.call(p)
+      end
+    }
     batched_pdhs = []
-    Container.group_pdhs_by_manifest_size(pdhs_in, batch_size_max) do |pdhs|
+    Container.group_pdhs_for_multiple_transactions(pdh_lambda, pdhs_in.size, batch_size_max, "") do |pdhs|
       batched_pdhs << pdhs
     end
     expected = [['x1+30', 'x2+30'], ['x3+201'], ['x4+100', 'x5+100']]