cmdlib: Fix typo, s/nick/NIC/

[ganeti-local] / lib / utils / io.py

diff --git a/lib/utils/io.py b/lib/utils/io.py
index a1ef785..08c9990 100644 (file)
--- 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: