diff --git a/spec/data/samples/autoscale_font.rb.txt b/spec/data/samples/autoscale_font.rb.txt
index 715f5e8..127b870 100644
--- a/spec/data/samples/autoscale_font.rb.txt
+++ b/spec/data/samples/autoscale_font.rb.txt
@@ -76,3 +76,6 @@ cairo: set_source_color([:red])
 cairo: set_line_width([2.0])
 cairo: stroke([])
 cairo: restore([])
+surface: write_to_png(["_output/autoscale_0.png"])
+surface: write_to_png(["_output/autoscale_1.png"])
+surface: write_to_png(["_output/autoscale_2.png"])
diff --git a/spec/data/samples/basic.rb.txt b/spec/data/samples/basic.rb.txt
index 8d4cd53..c8e5631 100644
--- a/spec/data/samples/basic.rb.txt
+++ b/spec/data/samples/basic.rb.txt
@@ -189,7 +189,7 @@ cairo: save([])
 cairo: translate([620, 218])
 cairo: rotate([0])
 cairo: translate([-620, -218])
-cairo: set_source([ImageSurface, 620, 218])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 620, 218])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: scale([1.0, 1.0])
@@ -198,6 +198,9 @@ cairo: save([])
 cairo: translate([620, 218])
 cairo: rotate([0])
 cairo: translate([-620, -218])
-cairo: set_source([ImageSurface, 620, 218])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 620, 218])
 cairo: paint([1.0])
 cairo: restore([])
+surface: write_to_png(["_output/basic_0.png"])
+surface: write_to_png(["_output/basic_1.png"])
+surface: write_to_png(["_output/basic_2.png"])
diff --git a/spec/data/samples/cairo_access.rb.txt b/spec/data/samples/cairo_access.rb.txt
index 191113b..77f8ed2 100644
--- a/spec/data/samples/cairo_access.rb.txt
+++ b/spec/data/samples/cairo_access.rb.txt
@@ -21,3 +21,5 @@ cairo: circle([50, 50, 50])
 cairo: set_source_color([:red])
 cairo: fill([])
 cairo: restore([])
+surface: write_to_png(["_output/cairo_access_0.png"])
+surface: write_to_png(["_output/cairo_access_1.png"])
diff --git a/spec/data/samples/colors.rb.txt b/spec/data/samples/colors.rb.txt
index 109595e..82c6286 100644
--- a/spec/data/samples/colors.rb.txt
+++ b/spec/data/samples/colors.rb.txt
@@ -121,3 +121,4 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
+surface: write_to_png(["_output/colors_0.png"])
diff --git a/spec/data/samples/custom_config.rb.txt b/spec/data/samples/custom_config.rb.txt
index 4df90f2..31c114d 100644
--- a/spec/data/samples/custom_config.rb.txt
+++ b/spec/data/samples/custom_config.rb.txt
@@ -37,8 +37,9 @@ cairo: save([])
 cairo: translate([620, 218])
 cairo: rotate([0])
 cairo: translate([-620, -218])
-cairo: set_source([ImageSurface, 620, 218])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 620, 218])
 cairo: paint([1.0])
 cairo: restore([])
-cairo: set_source([ImageSurface, 75, 75])
+surface: write_to_png(["_output/custom-config_0.png"])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 75, 75])
 cairo: paint([])
diff --git a/spec/data/samples/draw_shapes.rb.txt b/spec/data/samples/draw_shapes.rb.txt
index fa09b94..7ce650c 100644
--- a/spec/data/samples/draw_shapes.rb.txt
+++ b/spec/data/samples/draw_shapes.rb.txt
@@ -32,3 +32,4 @@ cairo: set_source_color([#<Cairo::Color::RGB: @alpha=1.0, @red=0.0, @green=0.0,
 cairo: set_line_width([25.0])
 cairo: stroke([])
 cairo: restore([])
+surface: write_to_png(["_output/shape_0.png"])
diff --git a/spec/data/samples/excel.rb.txt b/spec/data/samples/excel.rb.txt
index 00c8d48..91d0dda 100644
--- a/spec/data/samples/excel.rb.txt
+++ b/spec/data/samples/excel.rb.txt
@@ -163,3 +163,6 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
+surface: write_to_png(["_output/sample_excel_0.png"])
+surface: write_to_png(["_output/sample_excel_1.png"])
+surface: write_to_png(["_output/sample_excel_2.png"])
diff --git a/spec/data/samples/hello_world.rb.txt b/spec/data/samples/hello_world.rb.txt
index 3b0ccc7..c613edb 100644
--- a/spec/data/samples/hello_world.rb.txt
+++ b/spec/data/samples/hello_world.rb.txt
@@ -32,3 +32,5 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
+surface: write_to_png(["_output/card_0.png"])
+surface: write_to_png(["_output/card_1.png"])
diff --git a/spec/data/samples/load_images.rb.txt b/spec/data/samples/load_images.rb.txt
index 80b7dfd..6f6b0ea 100644
--- a/spec/data/samples/load_images.rb.txt
+++ b/spec/data/samples/load_images.rb.txt
@@ -24,7 +24,7 @@ cairo: save([])
 cairo: translate([620, 218])
 cairo: rotate([0])
 cairo: translate([-620, -218])
-cairo: set_source([ImageSurface, 620, 218])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 620, 218])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: scale([1.953125, 1.953125])
@@ -33,7 +33,7 @@ cairo: save([])
 cairo: translate([50, 50])
 cairo: rotate([0])
 cairo: translate([-50, -50])
-cairo: set_source([ImageSurface, 50, 50])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 50, 50])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: scale([0.5859375, 0.5859375])
@@ -42,7 +42,7 @@ cairo: save([])
 cairo: translate([50, 350])
 cairo: rotate([0])
 cairo: translate([-50, -350])
-cairo: set_source([ImageSurface, 50, 350])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 50, 350])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: scale([0.9765625, 0.9765625])
@@ -51,7 +51,7 @@ cairo: save([])
 cairo: translate([50, 450])
 cairo: rotate([0])
 cairo: translate([-50, -450])
-cairo: set_source([ImageSurface, 50, 450])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 50, 450])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: scale([0.8021390374331551, 0.8032128514056225])
@@ -60,7 +60,7 @@ cairo: save([])
 cairo: translate([0, 0])
 cairo: rotate([0])
 cairo: translate([0, 0])
-cairo: set_source([ImageSurface, 0, 0])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 0, 0])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: save([])
@@ -98,6 +98,7 @@ cairo: save([])
 cairo: translate([300, 900])
 cairo: rotate([1.4707963267948965])
 cairo: translate([-300, -900])
-cairo: set_source([ImageSurface, 300, 900])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 300, 900])
 cairo: paint([1.0])
 cairo: restore([])
+surface: write_to_png(["_output/load_images_0.png"])
diff --git a/spec/data/samples/portrait-landscape.rb.txt b/spec/data/samples/portrait-landscape.rb.txt
index 6bd7e88..7a4dd8c 100644
--- a/spec/data/samples/portrait-landscape.rb.txt
+++ b/spec/data/samples/portrait-landscape.rb.txt
@@ -19,6 +19,7 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
+surface: write_to_png(["_output/portrait_0.png"])
 cairo: save([])
 cairo: set_source_color([#<Cairo::Color::RGB: @alpha=1.0, @red=0.6666666666666666, @green=0.6666666666666666, @blue=0.6666666666666666>])
 cairo: paint([])
@@ -43,6 +44,6 @@ cairo: restore([])
 cairo: translate([412.5, 562.5])
 cairo: rotate([1.5707963267948966])
 cairo: translate([-562.5, -412.5])
-cairo: set_source([ImageSurface])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>])
 cairo: paint([])
 surface: write_to_png(["_output/landscape_0.png"])
diff --git a/spec/data/samples/ranges.rb.txt b/spec/data/samples/ranges.rb.txt
index 9e59b52..4effd03 100644
--- a/spec/data/samples/ranges.rb.txt
+++ b/spec/data/samples/ranges.rb.txt
@@ -421,3 +421,6 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
+surface: write_to_png(["_output/ranges_0.png"])
+surface: write_to_png(["_output/ranges_1.png"])
+surface: write_to_png(["_output/ranges_2.png"])
diff --git a/spec/data/samples/save_pdf.rb.txt b/spec/data/samples/save_pdf.rb.txt
index 362d9c9..7d3cb64 100644
--- a/spec/data/samples/save_pdf.rb.txt
+++ b/spec/data/samples/save_pdf.rb.txt
@@ -478,69 +478,68 @@ cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: update_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: show_pango_layout([#<RSpec::Mocks::Double: @name=Pango::Layout>])
 cairo: restore([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 75, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 75, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 831, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 180, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 1587, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 285, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 2343, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 390, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 75, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 495, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 831, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 600, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 1587, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 705, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 2343, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 810, 75])
 cairo: paint([])
-cairo: show_page([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 75, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 915, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 831, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1020, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 1587, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1125, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 2343, 75])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1230, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 75, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1335, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 831, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1440, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 1587, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1545, 75])
 cairo: paint([])
-cairo: set_source([ImageSurface, -37, -37])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, -37, -37])
 cairo: paint([])
-cairo: set_source([ImageSurface, 2343, 1131])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 1650, 75])
 cairo: paint([])
-cairo: show_page([])
+surface: write_to_png(["_output/save_pdf_1.png"])
diff --git a/spec/data/samples/tgc_proofs.rb.txt b/spec/data/samples/tgc_proofs.rb.txt
index bc20b18..5ea8b08 100644
--- a/spec/data/samples/tgc_proofs.rb.txt
+++ b/spec/data/samples/tgc_proofs.rb.txt
@@ -68,7 +68,7 @@ cairo: save([])
 cairo: translate([620, 218])
 cairo: rotate([0])
 cairo: translate([-620, -218])
-cairo: set_source([ImageSurface, 620, 218])
+cairo: set_source([#<RSpec::Mocks::Double: @name=Cairo::ImageSurface>, 620, 218])
 cairo: paint([1.0])
 cairo: restore([])
 cairo: save([])
@@ -78,3 +78,4 @@ cairo: translate([0, 0])
 cairo: set_source([ImageSurface, 0, 0])
 cairo: paint([0.5])
 cairo: restore([])
+surface: write_to_png(["_output/tgc_sample_0.png"])
diff --git a/spec/data/samples/units.rb.txt b/spec/data/samples/units.rb.txt
index aef0898..b9ee00c 100644
--- a/spec/data/samples/units.rb.txt
+++ b/spec/data/samples/units.rb.txt
@@ -11,3 +11,4 @@ cairo: rounded_rectangle([37.5, 37.5, 750.0, 1050.0, 25, 25])
 cairo: set_source_color([#<Cairo::Color::RGB: @alpha=0.0, @red=0.0, @green=0.0, @blue=0.0>])
 cairo: fill([])
 cairo: restore([])
+surface: write_to_png(["_output/units_0.png"])
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 893ea45..0238128 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -50,6 +50,8 @@ def mock_cairo(strio)
   surface = double(Cairo::ImageSurface)
   pango   = double(Pango::Layout)
   allow(ProgressBar).to receive(:create).and_return(Squib::DoNothing.new)
+  allow(Cairo::ImageSurface).to receive(:new).and_return(surface)
+  allow(surface).to receive(:width).and_return(100)
   allow(Cairo::Context).to receive(:new).and_return(cxt)
   allow(cxt).to receive(:create_pango_layout).and_return(pango)
   allow(cxt).to receive(:target).and_return(surface)