Invalidate cache in rm_file (#762)

Author: ianthomas23

s3fs/core.py

@@ -1879,6 +1879,7 @@ async def _bulk_delete(self, pathlist, **kwargs):
 
     async def _rm_file(self, path, **kwargs):
         bucket, key, _ = self.split_path(path)
+        self.invalidate_cache(path)
 
         try:
             await self._call_s3("delete_object", Bucket=bucket, Key=key)

s3fs/tests/test_s3fs.py

@@ -2636,3 +2636,20 @@ async def read_stream():
 
     asyncio.run(read_stream())
     assert b"".join(out) == data
+
+
+def test_rm_invalidates_cache(s3):
+    # Issue 761: rm_file does not invalidate cache
+    fn = test_bucket_name + "/2014-01-01.csv"
+    assert s3.exists(fn)
+    assert fn in s3.ls(test_bucket_name)
+    s3.rm(fn)
+    assert not s3.exists(fn)
+    assert fn not in s3.ls(test_bucket_name)
+
+    fn = test_bucket_name + "/2014-01-02.csv"
+    assert s3.exists(fn)
+    assert fn in s3.ls(test_bucket_name)
+    s3.rm_file(fn)
+    assert not s3.exists(fn)
+    assert fn not in s3.ls(test_bucket_name)