Fix default cache dataloader raise key error on non-existing key

RELEASE.md

@@ -0,0 +1,16 @@
+Release type: patch
+
+Calling `.clear(key)` on default dataloader with non-existing `key` will not throw `KeyError` anymore. Example:
+```python
+from strawberry.dataloader import DataLoader
+
+
+async def load_data(keys):
+    return [str(key) for key in keys]
+
+
+dataloader = DataLoader(load_fn=load_data)
+dataloader.clear(42)  # does not throw KeyError anymore
+```
+
+This is a patch release with no breaking changes.

strawberry/dataloader.py

@@ -84,7 +84,9 @@ def set(self, key: K, value: Future[T]) -> None:
         self.cache_map[self.cache_key_fn(key)] = value
 
     def delete(self, key: K) -> None:
-        del self.cache_map[self.cache_key_fn(key)]
+        cache_key = self.cache_key_fn(key)
+        if cache_key in self.cache_map:
+            del self.cache_map[cache_key]
 
     def clear(self) -> None:
         self.cache_map.clear()

tests/test_dataloaders.py

@@ -288,6 +288,10 @@ async def idx(keys: List[int]) -> List[Tuple[int, int]]:
 
     assert await loader.load_many([1, 2, 3]) == [(1, 4), (2, 4), (3, 4)]
 
+    loader.clear(-1)
+
+    assert await loader.load_many([1, 2, 3]) == [(1, 4), (2, 4), (3, 4)]
+
 
 @pytest.mark.asyncio
 async def test_clear_nocache():
@@ -315,6 +319,10 @@ async def idx(keys: List[int]) -> List[Tuple[int, int]]:
 
     assert await loader.load_many([1, 2, 3]) == [(1, 4), (2, 4), (3, 4)]
 
+    loader.clear(-1)
+
+    assert await loader.load_many([1, 2, 3]) == [(1, 5), (2, 5), (3, 5)]
+
 
 @pytest.mark.asyncio
 async def test_dont_dispatch_cancelled():