Prechádzať zdrojové kódy

Fixes beat tests on Py3

Ask Solem 8 rokov pred
rodič
commit
a566fb2291
1 zmenil súbory, kde vykonal 6 pridanie a 6 odobranie
  1. 6 6
      t/unit/app/test_beat.py

+ 6 - 6
t/unit/app/test_beat.py

@@ -398,17 +398,17 @@ class test_PersistentScheduler:
         s.setup_schedule()
         s._remove_db.assert_called_with()
 
-        s._store = {'__version__': 1}
+        s._store = {b'__version__': 1}
         s.setup_schedule()
 
         s._store.clear = Mock()
         op = s.persistence.open = Mock()
         op.return_value = s._store
-        s._store['tz'] = 'FUNKY'
+        s._store[b'tz'] = 'FUNKY'
         s.setup_schedule()
         op.assert_called_with(s.schedule_filename, writeback=True)
         s._store.clear.assert_called_with()
-        s._store['utc_enabled'] = False
+        s._store[b'utc_enabled'] = False
         s._store.clear = Mock()
         s.setup_schedule()
         s._store.clear.assert_called_with()
@@ -417,10 +417,10 @@ class test_PersistentScheduler:
         s = create_persistent_scheduler()[0](
             schedule_filename='schedule', app=self.app,
         )
-        s._store = {'entries': {}}
+        s._store = {b'entries': {}}
         s.schedule = {'foo': 'bar'}
         assert s.schedule == {'foo': 'bar'}
-        assert s._store['entries'] == s.schedule
+        assert s._store[b'entries'] == s.schedule
 
 
 class test_Service:
@@ -439,7 +439,7 @@ class test_Service:
         assert isinstance(schedule, dict)
         assert isinstance(s.scheduler, beat.Scheduler)
         scheduled = list(schedule.keys())
-        for task_name in keys(sh['entries']):
+        for task_name in keys(sh[b'entries']):
             assert task_name in scheduled
 
         s.sync()