X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/437c3e77fb35bec7ab902873fa31d6225f8602f4..958d01f8f739093f752bff4af259f3e9bb7ff4c7:/lib/utils/io.py

diff --git a/lib/utils/io.py b/lib/utils/io.py
index a1ef785..08c9990 100644
--- a/lib/utils/io.py
+++ b/lib/utils/io.py
@@ -135,11 +135,15 @@ def WriteFile(file_name, fn=None, data=None,
       if callable(prewrite):
         prewrite(fd)
       if data is not None:
+        if isinstance(data, unicode):
+          data = data.encode()
+        assert isinstance(data, str)
         to_write = len(data)
         offset = 0
         while offset < to_write:
           written = os.write(fd, buffer(data, offset))
           assert written >= 0
+          assert written <= to_write - offset
           offset += written
         assert offset == to_write
       else: