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:
Tim Abbott
2017-10-30 14:10:07 -07:00
parent 692db3f62b
commit c160c06f9c

View File

@@ -116,11 +116,13 @@ class TestServiceBotStateHandler(ZulipTestCase):
self.assertTrue(storage.contains('some key'))
self.assertFalse(storage.contains('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)
self.assertRaises(BotUserStateData.DoesNotExist, lambda: second_storage.get('some key'))
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')
def test_marshaling(self):