lib/snapshot/collector.rb in snapshot-1.13.1 vs lib/snapshot/collector.rb in snapshot-1.14.0

@@ -71,11 +71,14 @@
       UI.verbose "Found #{to_store.join(', ')}"
       return to_store
     end
 
     def self.check_activity(activity, to_store)
-      # We now check if it's the rotation gesture, because that's the only thing we care about
-      if activity["Title"] == "Set device orientation to Unknown"
+      # On iOS, we look for the "Unknown" rotation gesture that signals a snapshot was taken here.
+      # On tvOS, we look for "Browser" count.
+      # These are both events that are not normally triggered by UI testing, making it easy for us to
+      # locate where snapshot() was invoked.
+      if activity["Title"] == "Set device orientation to Unknown" || activity["Title"] == "Get number of matches for: Children matching type Browser"
         if activity["Attachments"]
           to_store << activity["Attachments"].last["FileName"]
         else # Xcode 7.3 has stopped including 'Attachments', so we synthesize the filename manually
           to_store << "Screenshot_#{activity['UUID']}.png"
         end