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