Bug 1872171 - add and support --restartAfterFailure for desktop mochi…

…tests. r=ahal

Differential Revision: https://phabricator.services.mozilla.com/D197365

UltraBlame original commit: 39c1f4cb2135dc97883ba694417a713f52325891

testing/mochitest/mochitest_options.py

@@ -6078,6 +6078,60 @@
 
 }
 
+]
+
+[
+
+[
+"
+-
+-
+restart
+-
+after
+-
+failure
+"
+]
+
+{
+
+"
+dest
+"
+:
+"
+restartAfterFailure
+"
+
+"
+default
+"
+:
+False
+
+"
+help
+"
+:
+"
+Terminate
+the
+session
+on
+first
+failure
+and
+restart
+where
+you
+left
+off
+.
+"
+
+}
+
 ]
 
 ]

testing/mochitest/runtests.py

@@ -18536,6 +18536,10 @@
 =
 None
 
+restartAfterFailure
+=
+False
+
 marionette_args
 =
 None
@@ -19045,6 +19049,10 @@
 bisectChunk
 =
 bisectChunk
+
+restartAfterFailure
+=
+restartAfterFailure
 
 )
 
@@ -20838,6 +20846,109 @@
 result
 )
 
+elif
+options
+.
+restartAfterFailure
+:
+
+if
+not
+self
+.
+expectedError
+:
+
+status
+=
+-
+1
+
+else
+:
+
+firstFail
+=
+len
+(
+testsToRun
+)
+
+for
+key
+in
+self
+.
+expectedError
+:
+
+full_key
+=
+[
+x
+for
+x
+in
+testsToRun
+if
+key
+in
+x
+]
+
+if
+full_key
+:
+
+if
+testsToRun
+.
+index
+(
+full_key
+[
+0
+]
+)
+<
+firstFail
+:
+
+firstFail
+=
+testsToRun
+.
+index
+(
+full_key
+[
+0
+]
+)
+
+testsToRun
+=
+testsToRun
+[
+firstFail
++
+1
+:
+]
+
+if
+testsToRun
+=
+=
+[
+]
+:
+
+status
+=
+-
+1
+
 else
 :
 
@@ -24756,6 +24867,12 @@
 .
 bisectChunk
 
+restartAfterFailure
+=
+options
+.
+restartAfterFailure
+
 marionette_args
 =
 marionette_args
@@ -25808,6 +25925,10 @@
 bisectChunk
 =
 None
+
+restartAfterFailure
+=
+None
 
 )
 :
@@ -25885,6 +26006,12 @@
 =
 bisectChunk
 
+self
+.
+restartAfterFailure
+=
+restartAfterFailure
+
 self
 .
 browserProcessId
@@ -26036,6 +26163,10 @@
 self
 .
 bisectChunk
+or
+self
+.
+restartAfterFailure
 :
 
 handlers
@@ -27444,6 +27575,18 @@
 =
 True
 
+if
+options
+.
+restartAfterFailure
+:
+
+options
+.
+runUntilFailure
+=
+True
+
 if
 options
 .

testing/mochitest/runtestsremote.py

@@ -2201,6 +2201,10 @@
 =
 None
 
+restartAfterFailure
+=
+False
+
 marionette_args
 =
 None

testing/mochitest/tests/SimpleTest/TestRunner.js

@@ -980,6 +980,11 @@ _numTimeouts
 TestRunner
 .
 maxTimeouts
+|
+|
+TestRunner
+.
+runUntilFailure
 )
 {
 TestRunner