✨ Coerce search args with SequenceSet.try_convert

ruby · Nov 8, 2024 · 4372efb · 4372efb
1 parent 24faaac
commit 4372efb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/net/imap.rb b/lib/net/imap.rb
@@ -1948,6 +1948,7 @@ def uid_expunge(uid_set)
     #     * +Set+
     #     * +Range+
     #     * <tt>-1</tt> and +:*+ -- both translate to <tt>*</tt>
+    #     * responds to +#to_sequence_set+
     #     * nested +Array+
     #   * Any +String+ is sent verbatim when it is a valid \IMAP atom,
     #     and encoded as an \IMAP quoted or literal string otherwise.
@@ -3200,7 +3201,7 @@ def normalize_searching_criteria(criteria)
         when Range       then SequenceSet[arg]
         when Array       then SequenceSet[arg]
         else
-          arg
+          SequenceSet.try_convert(arg) || arg
         end
       }
     end

diff --git a/test/net/imap/test_imap.rb b/test/net/imap/test_imap.rb
@@ -1230,6 +1230,13 @@ def test_unselect
       imap.search([:*])
       assert_equal "*", server.commands.pop.args
 
+      seqset_coercible = Object.new
+      def seqset_coercible.to_sequence_set
+        Net::IMAP::SequenceSet[1..9]
+      end
+      imap.search([seqset_coercible])
+      assert_equal "1:9", server.commands.pop.args
+
       server.on "UID SEARCH", &search_resp
       assert_equal search_result, imap.uid_search(["subject", "hello",
                                                    [1..22, 30..-1]])