mirror of
https://github.com/zulip/zulip.git
synced 2025-11-20 06:28:23 +00:00
bot_storage: Fix missing test coverage.
Apparently, we didn't have coverage on setting a second value for a key.
This commit is contained in:
@@ -116,11 +116,13 @@ class TestServiceBotStateHandler(ZulipTestCase):
|
|||||||
self.assertTrue(storage.contains('some key'))
|
self.assertTrue(storage.contains('some key'))
|
||||||
self.assertFalse(storage.contains('nonexistent key'))
|
self.assertFalse(storage.contains('nonexistent key'))
|
||||||
self.assertRaises(BotUserStateData.DoesNotExist, lambda: storage.get('nonexistent key'))
|
self.assertRaises(BotUserStateData.DoesNotExist, lambda: storage.get('nonexistent key'))
|
||||||
|
storage.put('some key', 'a new value')
|
||||||
|
self.assertEqual(storage.get('some key'), 'a new value')
|
||||||
|
|
||||||
second_storage = StateHandler(self.second_bot_profile)
|
second_storage = StateHandler(self.second_bot_profile)
|
||||||
self.assertRaises(BotUserStateData.DoesNotExist, lambda: second_storage.get('some key'))
|
self.assertRaises(BotUserStateData.DoesNotExist, lambda: second_storage.get('some key'))
|
||||||
second_storage.put('some key', 'yet another value')
|
second_storage.put('some key', 'yet another value')
|
||||||
self.assertEqual(storage.get('some key'), 'some value')
|
self.assertEqual(storage.get('some key'), 'a new value')
|
||||||
self.assertEqual(second_storage.get('some key'), 'yet another value')
|
self.assertEqual(second_storage.get('some key'), 'yet another value')
|
||||||
|
|
||||||
def test_marshaling(self):
|
def test_marshaling(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user