1 files changed, 6 insertions, 2 deletions
diff --git a/spec/support/matchers/pushed_frontend_feature_flags_matcher.rb b/spec/support/matchers/pushed_frontend_feature_flags_matcher.rb
index b49d4da8cda..ecd174edec9 100644
--- a/spec/support/matchers/pushed_frontend_feature_flags_matcher.rb
+++ b/spec/support/matchers/pushed_frontend_feature_flags_matcher.rb
@@ -5,15 +5,19 @@ RSpec::Matchers.define :have_pushed_frontend_feature_flags do |expected|
     "\"#{key}\":#{value}"
   end
 
+  def html(actual)
+    actual.try(:html) || actual
+  end
+
   match do |actual|
     expected.all? do |feature_flag_name, enabled|
-      page.html.include?(to_js(feature_flag_name, enabled))
+      html(actual).include?(to_js(feature_flag_name, enabled))
     end
   end
 
   failure_message do |actual|
     missing = expected.select do |feature_flag_name, enabled|
-      !page.html.include?(to_js(feature_flag_name, enabled))
+      !html(actual).include?(to_js(feature_flag_name, enabled))
     end
 
     formatted_missing_flags = missing.map { |feature_flag_name, enabled| to_js(feature_flag_name, enabled) }.join("\n")