ZJIT: Add insns keyword to test that a certain opcode is present

Author: XrXr

test/ruby/test_zjit.rb

@@ -488,11 +488,11 @@ def fib(n)
   end
 
   def test_opt_aref_with
-    assert_compiles ':ok', %q{
+    assert_compiles ':ok', <<~RUBY, insns: [:opt_aref_with]
       def aref_with(hash) = hash["key"]
 
       aref_with({ "key" => :ok })
-    }
+    RUBY
   end
 
   # tool/ruby_vm/views/*.erb relies on the zjit instructions a) being contiguous and
@@ -512,26 +512,47 @@ def test_instruction_order
 
   # Assert that every method call in `test_script` can be compiled by ZJIT
   # at a given call_threshold
-  def assert_compiles(expected, test_script, **opts)
+  def assert_compiles(expected, test_script, insns: [], **opts)
     pipe_fd = 3
 
     script = <<~RUBY
       _test_proc = -> {
         RubyVM::ZJIT.assert_compiles
         #{test_script}
       }
-      result = _test_proc.call
-      IO.open(#{pipe_fd}).write(result.inspect)
+      ret_val = _test_proc.call
+      result = {
+        ret_val:,
+        #{ unless insns.empty?
+          'insns: RubyVM::InstructionSequence.of(_test_proc).enum_for(:each_child).map(&:to_a)'
+        end}
+      }
+      IO.open(#{pipe_fd}).write(Marshal.dump(result))
     RUBY
 
-    status, out, err, actual = eval_with_jit(script, pipe_fd:, **opts)
+    status, out, err, result = eval_with_jit(script, pipe_fd:, **opts)
 
     message = "exited with status #{status.to_i}"
     message << "\nstdout:\n```\n#{out}```\n" unless out.empty?
     message << "\nstderr:\n```\n#{err}```\n" unless err.empty?
     assert status.success?, message
 
-    assert_equal expected, actual
+    result = Marshal.load(result)
+    assert_equal expected, result.fetch(:ret_val).inspect
+
+    unless insns.empty?
+      iseqs = result.fetch(:insns)
+      iseqs.filter! { it[9] == :method } # ISeq type
+      assert_equal 1, iseqs.size, "Opcode assertions tests must define exactly one method"
+      iseq_insns = iseqs.first.last
+
+      expected_insns = Set.new(insns)
+      iseq_insns.each do
+        next unless it.is_a?(Array)
+        expected_insns.delete(it.first)
+      end
+      assert(expected_insns.empty?, -> { "Not present in ISeq: #{expected_insns.to_a}" })
+    end
   end
 
   # Run a Ruby process with ZJIT options and a pipe for writing test results