ci: Test DuckDB (#12)

DuckDB may not return results sorted by `id`. So we need to sort
returned results manually for stable result.

Author: kou

.github/workflows/test.yaml

@@ -58,25 +58,51 @@ jobs:
       - name: Install dependencies
         run: |
           MAKEFLAGS="-j$(nproc)" bundle install --jobs=$(nproc)
-      - name: Prepare test environment
+      - name: "Prepare test environment: SQLite3"
         run: |
           sudo apt install -y -V \
-            libadbc-driver-postgresql-dev \
-            libadbc-driver-sqlite-dev \
-            postgresql
-          sudo systemctl restart postgresql
-          sudo -u postgres -H psql -c "CREATE ROLE ${USER} SUPERUSER LOGIN;"
+            libadbc-driver-sqlite-dev
+          latest_version=
       - name: "Test: SQLite3"
         # We can remove this once ADBC 20 is released.
         continue-on-error: true
         run: |
           bundle exec rake
+      - name: "Prepare test environment: PostgreSQL"
+        run: |
+          sudo apt install -y -V \
+            libadbc-driver-postgresql-dev \
+            postgresql
+          sudo systemctl restart postgresql
+          sudo -u postgres -H psql -c "CREATE ROLE ${USER} SUPERUSER LOGIN;"
       - name: "Test: PostgreSQL"
         env:
           ACTIVERECORD_ADBC_ADAPTER_BACKEND: postgresql
           PGHOST: /var/run/postgresql
         run: |
           bundle exec rake
+      - name: "Prepare test environment: DuckDB"
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: |
+          latest_tag=$(gh release list \
+                         --exclude-drafts \
+                         --exclude-pre-releases \
+                         --jq '.[].tagName' \
+                         --json tagName \
+                         --limit 1 \
+                         --repo duckdb/duckdb)
+          wget \
+            --output-document libduckdb.zip \
+            --no-verbose \
+            https://github.com/duckdb/duckdb/releases/download/${latest_tag}/libduckdb-linux-amd64.zip
+          unzip libduckdb.zip
+          sudo mv libduckdb.so /usr/lib/
+      - name: "Test: DuckDB"
+        env:
+          ACTIVERECORD_ADBC_ADAPTER_BACKEND: duckdb
+        run: |
+          bundle exec rake
       - name: Install benchmark dependencies
         run: |
           # We can use this with ubuntu-26.04.

test/test_model.rb

@@ -20,7 +20,7 @@ def test_all
                    User.new(id: 2),
                    User.new(id: 3),
                  ],
-                 User.select(:id).all)
+                 User.select(:id).all.order(:id))
   end
 
   def test_count
@@ -36,47 +36,57 @@ def test_record_batch
       record_batch = Arrow::RecordBatch.new(id: id_array)
       User.ingest(record_batch)
       assert_equal((1..6).collect {|id| User.new(id: id)},
-                   User.select(:id).all)
+                   User.select(:id).all.order(:id))
     end
 
     def test_table
       table = Arrow::Table.new(id: id_array)
       User.ingest(table)
       assert_equal((1..6).collect {|id| User.new(id: id)},
-                   User.select(:id).all)
+                   User.select(:id).all.order(:id))
     end
 
     def test_record_batch_reader
       record_batch = Arrow::RecordBatch.new(id: id_array)
       User.ingest(Arrow::RecordBatchReader.new([record_batch]))
       assert_equal((1..6).collect {|id| User.new(id: id)},
-                   User.select(:id).all)
+                   User.select(:id).all.order(:id))
     end
   end
 
   sub_test_case("#to_arrow") do
+    def sort_table(table)
+      table.take(table.sort_indices(:id))
+    end
+
     def test_model
       assert_equal(Arrow::Table.new(id: Arrow::Int64Array.new([1, 2, 3])),
-                   User.to_arrow)
+                   sort_table(User.to_arrow))
     end
 
     def test_relation
       assert_equal(Arrow::Table.new(id: Arrow::Int64Array.new([1, 2, 3])),
-                   User.all.to_arrow)
+                   User.all.order(:id).to_arrow)
     end
   end
 
   sub_test_case("#each_record_batch") do
+    def sort_record_batches(record_batches)
+      record_batches.collect do |record_batch|
+        record_batch.take(record_batch.sort_indices(:id))
+      end
+    end
+
     def test_model
       record_batch = Arrow::RecordBatch.new(id: Arrow::Int64Array.new([1, 2, 3]))
       assert_equal([record_batch],
-                   User.each_record_batch.to_a)
+                   sort_record_batches(User.each_record_batch))
     end
 
     def test_relation
       record_batch = Arrow::RecordBatch.new(id: Arrow::Int64Array.new([1, 2, 3]))
       assert_equal([record_batch],
-                   User.all.each_record_batch.to_a)
+                   User.all.order(:id).each_record_batch.to_a)
     end
   end
 end