[py-svn] r37779 - in py/trunk/py/execnet: . testing

[fijal at codespeak.net]

Author: fijal
Date: Fri Feb  2 00:40:49 2007
New Revision: 37779

Modified:
   py/trunk/py/execnet/rsync.py
   py/trunk/py/execnet/rsync_remote.py
   py/trunk/py/execnet/testing/test_rsync.py
Log:
Fix and a test for disappearing files.


Modified: py/trunk/py/execnet/rsync.py
==============================================================================
--- py/trunk/py/execnet/rsync.py	(original)
+++ py/trunk/py/execnet/rsync.py	Fri Feb  2 00:40:49 2007
@@ -85,26 +85,33 @@
                 elif command == "send":
                     modified_rel_path, checksum = data
                     modifiedpath = os.path.join(self.sourcedir, *modified_rel_path)
-                    f = open(modifiedpath, 'rb')
-                    data = f.read()
+                    try:
+                        f = open(modifiedpath, 'rb')
+                        data = f.read()
+                    except IOError:
+                        data = None
 
                     # provide info to progress callback function
                     modified_rel_path = "/".join(modified_rel_path)
-                    self.paths[modified_rel_path] = len(data)
+                    if data is not None:
+                        self.paths[modified_rel_path] = len(data)
+                    else:
+                        self.paths[modified_rel_path] = 0
                     if channel not in self.to_send:
                         self.to_send[channel] = []
                     self.to_send[channel].append(modified_rel_path)
 
-                    f.close()
-                    if checksum is not None and checksum == md5.md5(data).digest():
-                        data = None     # not really modified
-                    else:
-                        # ! there is a reason for the interning:
-                        # sharing multiple copies of the file's data
-                        data = intern(data)
-                        print '%s <= %s' % (
-                            channel.gateway._getremoteaddress(),
-                            modified_rel_path)
+                    if data is not None:
+                        f.close()
+                        if checksum is not None and checksum == md5.md5(data).digest():
+                            data = None     # not really modified
+                        else:
+                            # ! there is a reason for the interning:
+                            # sharing multiple copies of the file's data
+                            data = intern(data)
+                            print '%s <= %s' % (
+                                channel.gateway._getremoteaddress(),
+                                modified_rel_path)
                     channel.send(data)
                     del data
                 else:
@@ -118,7 +125,11 @@
         self.links.append(("link", basename, linkpoint))
 
     def _send_directory_structure(self, path):
-        st = os.lstat(path)
+        try:
+            st = os.lstat(path)
+        except OSError:
+            self._broadcast((0, 0))
+            return
         if stat.S_ISREG(st.st_mode):
             # regular file: send a timestamp/size pair
             self._broadcast((st.st_mtime, st.st_size))

Modified: py/trunk/py/execnet/rsync_remote.py
==============================================================================
--- py/trunk/py/execnet/rsync_remote.py	(original)
+++ py/trunk/py/execnet/rsync_remote.py	Fri Feb  2 00:40:49 2007
@@ -65,7 +65,10 @@
             f = open(path, 'wb')
             f.write(data)
             f.close()
-        os.utime(path, (time, time))
+        try:
+            os.utime(path, (time, time))
+        except OSError:
+            pass
         del data
     channel.send(("links", None))
 

Modified: py/trunk/py/execnet/testing/test_rsync.py
==============================================================================
--- py/trunk/py/execnet/testing/test_rsync.py	(original)
+++ py/trunk/py/execnet/testing/test_rsync.py	Fri Feb  2 00:40:49 2007
@@ -78,3 +78,24 @@
 
     assert total == {("list", 110):True, ("ack", 100):True, ("ack", 10):True}
 
+def test_file_disappearing():
+    base = py.test.ensuretemp("file_disappearing")
+    dest = base.join("dest")
+    source = base.join("source")
+    source.ensure("ex").write("a" * 100)
+    source.ensure("ex2").write("a" * 100)
+
+    class DRsync(RSync):
+        def filter(self, x):
+            if x.endswith("ex2"):
+                self.x = 1
+                source.join("ex2").remove()
+            return True
+
+    rsync = DRsync()
+    rsync.add_target(gw, dest)
+    rsync.send(source)
+    assert rsync.x == 1
+    assert len(dest.listdir()) == 1
+    assert len(source.listdir()) == 1
+