mirror of
https://github.com/ManchildProductions/UXP-Fixed.git
synced 2026-06-26 10:18:33 +00:00
Matt A. Tobin
31 lines
964 B
Python
31 lines
964 B
Python
#!/usr/bin/env python
|
|
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
# You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import unittest
|
|
import os
|
|
from mozprofile import Profile
|
|
|
|
|
|
class TestProfile(unittest.TestCase):
|
|
|
|
def test_with_profile_should_cleanup(self):
|
|
with Profile() as profile:
|
|
self.assertTrue(os.path.exists(profile.profile))
|
|
# profile is cleaned
|
|
self.assertFalse(os.path.exists(profile.profile))
|
|
|
|
def test_with_profile_should_cleanup_even_on_exception(self):
|
|
with self.assertRaises(ZeroDivisionError):
|
|
with Profile() as profile:
|
|
self.assertTrue(os.path.exists(profile.profile))
|
|
1 / 0 # will raise ZeroDivisionError
|
|
# profile is cleaned
|
|
self.assertFalse(os.path.exists(profile.profile))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|