Attempt to remove Firefox profile when using Marionette

Author: davehunt

py/selenium/webdriver/firefox/webdriver.py

@@ -99,12 +99,12 @@ def quit(self):
             self.service.stop()
         else:
             self.binary.kill()
-            try:
-                shutil.rmtree(self.profile.path)
-                if self.profile.tempfolder is not None:
-                    shutil.rmtree(self.profile.tempfolder)
-            except Exception as e:
-                print(str(e))
+        try:
+            shutil.rmtree(self.profile.path)
+            if self.profile.tempfolder is not None:
+                shutil.rmtree(self.profile.tempfolder)
+        except Exception as e:
+            print(str(e))
 
     @property
     def firefox_profile(self):

py/test/selenium/webdriver/firefox/mn_profile_tests.py

@@ -15,6 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import os
+
 import pytest
 from selenium import webdriver
 from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
@@ -47,3 +49,9 @@ def profile():
 
 def test_profile_is_used(driver):
     assert 'about:' == driver.current_url
+
+
+def test_profile_is_deleted(driver, profile):
+    assert os.path.exists(profile.path)
+    driver.quit()
+    assert not os.path.exists(profile.path)