Skip to content

Commit bbba96e

Browse files
committed
Rework tests from unittest to pytest
* Add frozen_time fixture to use freeze_time in every test
1 parent 5828c13 commit bbba96e

File tree

1 file changed

+12
-15
lines changed

1 file changed

+12
-15
lines changed

tests/messages/test_frontend.py

Lines changed: 12 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,12 @@
4848
from tests.messages.utils import CUSTOM_EXTRACTOR_COOKIE
4949

5050

51+
@pytest.fixture(autouse=True)
52+
def frozen_time():
53+
with freeze_time("1994-11-11"):
54+
yield
55+
56+
5157
def _po_file(locale):
5258
return os.path.join(i18n_dir, locale, 'LC_MESSAGES', 'messages.po')
5359

@@ -715,9 +721,9 @@ def test_supports_width(self):
715721
assert expected_content == actual_content
716722

717723

718-
class ConcatanateCatalogTestCase(unittest.TestCase):
724+
class TestConcatanateCatalog:
719725

720-
def setUp(self):
726+
def setup_method(self):
721727
self.olddir = os.getcwd()
722728
os.chdir(data_dir)
723729

@@ -747,7 +753,7 @@ def setUp(self):
747753
catalog.add(('plural', 'plurals'), string=('Plural', 'Plurals other'), locations=[('hard.py', 2000)])
748754
pofile.write_po(file, catalog)
749755

750-
def tearDown(self):
756+
def teardown_method(self):
751757
for file in [self.temp1, self.temp2, self.output_file]:
752758
if os.path.isfile(file):
753759
os.unlink(file)
@@ -783,7 +789,6 @@ def test_no_output_file(self):
783789
with pytest.raises(OptionError):
784790
self.cmd.finalize_options()
785791

786-
@freeze_time("1994-11-11")
787792
def test_default(self):
788793
self.cmd.input_files = [self.temp1, self.temp2]
789794
self.cmd.output_file = self.output_file
@@ -840,7 +845,6 @@ def test_default(self):
840845
actual_content = f.read()
841846
assert expected_content == actual_content
842847

843-
@freeze_time("1994-11-11")
844848
def test_use_first(self):
845849
self.cmd.input_files = [self.temp1, self.temp2]
846850
self.cmd.output_file = self.output_file
@@ -888,7 +892,6 @@ def test_use_first(self):
888892
actual_content = f.read()
889893
assert expected_content == actual_content
890894

891-
@freeze_time("1994-11-11")
892895
def test_unique(self):
893896
self.cmd.input_files = [self.temp1, self.temp2]
894897
self.cmd.output_file = self.output_file
@@ -928,7 +931,6 @@ def test_unique(self):
928931
actual_content = f.read()
929932
assert expected_content == actual_content
930933

931-
@freeze_time("1994-11-11")
932934
def test_more_than(self):
933935
self.cmd.input_files = [self.temp1, self.temp2]
934936
self.cmd.output_file = self.output_file
@@ -970,10 +972,9 @@ def test_more_than(self):
970972
assert expected_content == actual_content
971973

972974

973-
class MergeCatalogTestCase(unittest.TestCase):
975+
class TestMergeCatalog:
974976

975-
@freeze_time("1994-11-11")
976-
def setUp(self):
977+
def setup_method(self):
977978
self.olddir = os.getcwd()
978979
os.chdir(data_dir)
979980

@@ -1007,7 +1008,7 @@ def setUp(self):
10071008
catalog.add('word5', string='Word 5')
10081009
pofile.write_po(file, catalog)
10091010

1010-
def tearDown(self):
1011+
def teardown_method(self):
10111012
for file in [
10121013
self.temp_def,
10131014
self.temp_def + '~',
@@ -1066,8 +1067,6 @@ def test_no_output_file(self):
10661067
self.cmd.update = True
10671068
self.cmd.finalize_options()
10681069

1069-
1070-
@freeze_time("1994-11-11")
10711070
def test_default(self):
10721071
self.cmd.input_files = [self.temp_def, self.temp_ref]
10731072
self.cmd.output_file = self.output_file
@@ -1093,7 +1092,6 @@ def test_default(self):
10931092
actual_content = f.read()
10941093
assert expected_content == actual_content
10951094

1096-
@freeze_time("1994-11-11")
10971095
def test_compenidum(self):
10981096
self.cmd.input_files = [self.temp_def, self.temp_ref]
10991097
self.cmd.output_file = self.output_file
@@ -1121,7 +1119,6 @@ def test_compenidum(self):
11211119
actual_content = f.read()
11221120
assert expected_content == actual_content
11231121

1124-
@freeze_time("1994-11-11")
11251122
def test_compenidum_overwrite(self):
11261123
self.cmd.input_files = [self.temp_def, self.temp_ref]
11271124
self.cmd.output_file = self.output_file

0 commit comments

Comments
 (0)