def test_kvs(self):
with self.assertRaises(KeyError):
KVS.get('hogehoge')
KVS.put('hoge', 'fuga')
self.assertEqual('fuga', KVS.get('hoge'))
KVS.put('hoge', 'foo')
self.assertEqual('foo', KVS.get('hoge'))
def test_kvs(self):
with self.assertRaises(KeyError):
KVS.get('hogehoge')
KVS.put('hoge', 'fuga')
self.assertEqual('fuga', KVS.get('hoge'))
KVS.put('hoge', 'foo')
self.assertEqual('foo', KVS.get('hoge'))
def set(self, lobby):
KVS.put(self.active_lobby_key, lobby.pk)
def get(self):
try:
lobby_pk = KVS.get(self.active_lobby_key)
return Lobby.objects.get(pk=lobby_pk)
except KeyError:
return None
