diff options
Diffstat (limited to 'test/scenario_test/rtc_test.py')
| -rw-r--r-- | test/scenario_test/rtc_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/scenario_test/rtc_test.py b/test/scenario_test/rtc_test.py index 13bc8b25..98a91c70 100644 --- a/test/scenario_test/rtc_test.py +++ b/test/scenario_test/rtc_test.py @@ -38,8 +38,8 @@ class GoBGPTestBase(unittest.TestCase): def assert_upd_count(self, src, dst, sent, received): messages = src.get_neighbor(dst)['state']['messages'] - self.assertEqual(messages['sent']['update'], sent) - self.assertEqual(messages['received']['update'], received) + self.assertEqual(messages['sent'].get('update', 0), sent) + self.assertEqual(messages['received'].get('update', 0), received) @classmethod def setUpClass(cls): |