🐛 Fix SequenceSet[input] for a SequenceSet input

I'm surprised there was no test for this!
ruby · Sep 24, 2024 · 0cfc69b · 0cfc69b
1 parent d8b68bc
commit 0cfc69b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/net/imap/sequence_set.rb b/lib/net/imap/sequence_set.rb
@@ -304,7 +304,7 @@ class << self
         # Use ::new to create a mutable or empty SequenceSet.
         def [](first, *rest)
           if rest.empty?
-            if first.is_a?(SequenceSet) && set.frozen? && set.valid?
+            if first.is_a?(SequenceSet) && first.frozen? && first.valid?
               first
             else
               new(first).validate.freeze

diff --git a/test/net/imap/test_sequence_set.rb b/test/net/imap/test_sequence_set.rb
@@ -101,6 +101,11 @@ def compare_to_reference_set(nums, set, seqset)
     assert_raise DataFormatError do SequenceSet.new Time.now     end
   end
 
+  test ".[frozen SequenceSet] returns that SequenceSet" do
+    frozen_seqset = SequenceSet[123..456]
+    assert_same frozen_seqset, SequenceSet[frozen_seqset]
+  end
+
   test ".new, input may be empty" do
     assert_empty SequenceSet.new
     assert_empty SequenceSet.new []