fix yaml load

zed-0xff · Jan 29, 2022 · 76a45f7 · 76a45f7
1 parent 07b1d8d
commit 76a45f7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/spec/sections_spec.rb b/spec/sections_spec.rb
@@ -5,7 +5,20 @@
 ['calc.exe', 'bad/data_dir_15_entries.exe'].each do |fname|
   describe fname do
     it "should match saved sections info" do
-      sample.sections.should == YAML::load_file(File.join(DATA_DIR,"#{File.basename(fname)}_sections.yml"))
+      sections = nil
+      begin
+        # tested with ruby 3.0.3
+        sections = YAML::load_file(
+          File.join(DATA_DIR,"#{File.basename(fname)}_sections.yml"),
+          permitted_classes: [Symbol, PEdump::IMAGE_SECTION_HEADER]
+        )
+      rescue
+        # old ruby?
+        sections = YAML::load_file(
+          File.join(DATA_DIR,"#{File.basename(fname)}_sections.yml")
+        )
+      end
+      sample.sections.should == sections
     end
   end
 end