remove monkey_patch, add srem

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "znsocket"
-version = "0.1.0"
+version = "0.1.1"
 description = "Create objects with shared attributes through a socket connection."
 authors = ["Fabian Zills <[email protected]>"]
 license = "Apache-2.0"

tests/test_connection.py

@@ -186,3 +186,12 @@ def test_rdlete(client, request):
  c.set("name", "Alice")
  c.delete("name")
  assert c.get("name") is None
+
+
+@pytest.mark.parametrize("client", ["znsclient", "redisclient"])
+def test_srem(client, request):
+ c = request.getfixturevalue(client)
+ c.sadd("set", "member1")
+ c.sadd("set", "member2")
+ c.srem("set", "member1")
+ assert c.smembers("set") == {"member2"}

znsocket/__init__.py

@@ -1,7 +1,3 @@
-import eventlet
-
-eventlet.monkey_patch()
-
 from .client import Client
 from .server import get_sio
 

znsocket/client.py

@@ -78,5 +78,8 @@ def lrem(self, name: str, count: int, value: str):
  def sadd(self, name, value):
  return self.sio.call("sadd", {"name": name, "value": value})
 
+ def srem(self, name, value):
+ return self.sio.call("srem", {"name": name, "value": value})
+
  def flushall(self):
  return self.sio.call("flushall", {})

znsocket/server.py

@@ -180,4 +180,13 @@ def sadd(sid, data):
  def flushall(sid, data):
  storage.clear()
 
+ @sio.event
+ def srem(sid, data):
+ name = data.pop("name")
+ value = data.pop("value")
+ try:
+ storage[name].remove(value)
+ except KeyError:
+ pass
+
  return sio