fix tests

SNEWS2 · Aug 14, 2024 · 71fb3e1 · 71fb3e1
1 parent 7817ce3
commit 71fb3e1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/snews_pt/test/test_coincidence_tier.py b/snews_pt/test/test_coincidence_tier.py
@@ -8,8 +8,8 @@ def test_coincidence_expected():
                                neutrino_time='2012-06-09T15:31:08.891011',
                                firedrill_mode=False, is_test=True)
     # Check that message has expected structure.
-    assert coin.selected_tiers == ['SNEWSCoincidenceTierMessage']
-    assert len(coin.messages) == 1, f"Expected 1 CoincidenceTier Message got {len(coin.messages)}!"
+    assert coin.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSHeartbeatMessage']
+    assert len(coin.messages) == 2, f"Expected 1(+1 HB) CoincidenceTier Message got {len(coin.messages)}!"
 
     assert coin.messages[0].message_data == {'_id': 'KamLAND_CoincidenceTier_2012-06-09T15:30:00.000501000',
                                              'detector_name': 'KamLAND',

diff --git a/snews_pt/test/test_significance_tier.py b/snews_pt/test/test_significance_tier.py
@@ -15,7 +15,7 @@ def test_significance_expected():
                                is_test=True)
 
     # Check that message has expected structure.
-    assert sign.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSSignificanceTierMessage']
+    assert sign.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSSignificanceTierMessage', 'SNEWSHeartbeatMessage']
     assert sign.messages[0].message_data == {'_id': 'DS-20K_CoincidenceTier_2012-06-09T15:30:00.000501000',
                                              'schema_version': '1.3.1',
                                              'detector_name': 'DS-20K',

diff --git a/snews_pt/test/test_timing_tier.py b/snews_pt/test/test_timing_tier.py
@@ -15,7 +15,7 @@ def test_timing_expected():
                                firedrill_mode=False, is_test=True)
 
     # # Check that message has expected structure.
-    assert tims.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSTimingTierMessage']
+    assert tims.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSTimingTierMessage', 'SNEWSHeartbeatMessage']
     assert tims.messages[1].message_data == {'_id': 'XENONnT_TimingTier_2012-06-09T15:30:00.009876000',
                                              'schema_version': '1.3.1',
                                              'detector_name': 'XENONnT',
@@ -47,7 +47,7 @@ def test_timing_expected_with_floats():
                                firedrill_mode=False, is_test=True)
 
     # # Check that message has expected structure.
-    assert tims.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSTimingTierMessage']
+    assert tims.selected_tiers == ['SNEWSCoincidenceTierMessage', 'SNEWSTimingTierMessage', 'SNEWSHeartbeatMessage']
     assert tims.messages[1].message_data == {'_id': 'XENONnT_TimingTier_2012-06-09T15:30:00.009876000',
                                              'schema_version': '1.3.1',
                                              'detector_name': 'XENONnT',