mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-06-25 20:29:33 +00:00
30 lines
961 B
Python
30 lines
961 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()
|