xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Rename bot storage endpoint from state to storage.

Browse source
This commit is contained in:
Robert Hönig 2017-11-28 20:34:25 +01:00
parent 80e4ef9f72
commit d6a4c02ba5
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
zulip_bots/zulip_bots/lib.py
View file

@ -66,14 +66,14 @@ class StateHandler(object):
self.demarshal = lambda obj: json.loads(obj)
response = self._client.get_storage()
if response['result'] == 'success':
self.state_ = response['state']
self.state_ = response['storage']
else:
raise StateHandlerError("Error initializing state: {}".format(str(response)))
def put(self, key, value):
# type: (Text, Text) -> None
self.state_[key] = self.marshal(value)
response = self._client.update_storage({'state': {key: self.state_[key]}})
response = self._client.update_storage({'storage': {key: self.state_[key]}})
if response['result'] != 'success':
raise StateHandlerError("Error updating state: {}".format(str(response)))

2
zulip_bots/zulip_bots/test_lib.py
View file

@ -46,7 +46,7 @@ class BotTestCaseBase(TestCase):
self.MockClass = self.patcher.start()
self.mock_bot_handler = self.MockClass(None, None, None, None)
self.mock_client = MagicMock()
self.mock_client.get_storage.return_value = {'result': 'success', 'state': {}}
self.mock_client.get_storage.return_value = {'result': 'success', 'storage': {}}
self.mock_client.update_storage.return_value = {'result': 'success'}
self.mock_bot_handler.storage = StateHandler(self.mock_client)
self.mock_bot_handler.send_message.return_value = {'id': 42}