From deffd9703d2a42b079db87dfff3633a63d268a9b Mon Sep 17 00:00:00 2001
From: "P. Christeas" <p_christ@hol.gr>
Date: Wed, 3 Nov 2010 13:25:34 +0200
Subject: [PATCH] doc webdav: support for GET and PUT at yaml tests

bzr revid: p_christ@hol.gr-20101103112534-9xyjp4xup0pjoxwm
---
 addons/document_webdav/test/webdav_test1.yml |   20 +++++++
 addons/document_webdav/test_davclient.py     |   79 ++++++++++++++++++++++----
 2 files changed, 88 insertions(+), 11 deletions(-)

diff --git a/addons/document_webdav/test/webdav_test1.yml b/addons/document_webdav/test/webdav_test1.yml
index c9384a5..a64c2ef 100644
--- a/addons/document_webdav/test/webdav_test1.yml
+++ b/addons/document_webdav/test/webdav_test1.yml
@@ -37,3 +37,23 @@
         res = dc.gd_lsl(path=cr.dbname+'/Documents/')
         for lin in res:
             print "%(type)s\t%(uid)s\t%(gid)s\t%(size)s\t%(mtime)s\t%(name)s" % lin
+-
+    I will put a file to the server
+-
+    !python {model: ir.attachment}: |
+        from document_webdav import test_davclient as te
+        import addons
+        dc = te.DAVClient()
+        dc.get_creds(self, cr, uid)
+        tdp = addons.get_module_resource('document_webdav', 'test_davclient.py')
+        res = dc.gd_put(path=cr.dbname+'/Documents/test_davclient.py', srcpath=tdp)
+-
+    I will try to get the file from the root
+-
+    !python {model: ir.attachment}: |
+        from document_webdav import test_davclient as te
+        import addons
+        dc = te.DAVClient()
+        dc.get_creds(self, cr, uid)
+        tdp = addons.get_module_resource('document_webdav', 'test_davclient.py')
+        res = dc.gd_get(path=cr.dbname+'/Documents/test_davclient.py', compare=tdp)
diff --git a/addons/document_webdav/test_davclient.py b/addons/document_webdav/test_davclient.py
index 0af83f4..35c8708 100644
--- a/addons/document_webdav/test_davclient.py
+++ b/addons/document_webdav/test_davclient.py
@@ -252,6 +252,8 @@ class addAuthTransport:
                 if 'www-authenticate' in resp.msg:
                     (atype,realm) = resp.msg.getheader('www-authenticate').split(' ',1)
                     data1 = resp.read()
+                    if data1:
+                        log.warning("Why have data on a 401 auth. message?")
                     if realm.startswith('realm="') and realm.endswith('"'):
                         realm = realm[7:-1]
                     log.debug("Resp: %r %r", resp.version,resp.isclosed(), resp.will_close)
@@ -421,17 +423,18 @@ class DAVClient(object):
 
         log.debug("Reponse: %s %s",r1.status, r1.reason)
         data1 = r1.read()
-        log.debug("Body:\n%s\nEnd of body", data1)
-        try:
-            ctype = r1.msg.getheader('content-type')
-            if ctype and ';' in ctype:
-                ctype, encoding = ctype.split(';',1)
-            if ctype == 'text/xml':
-                doc = xml.dom.minidom.parseString(data1)
-                log.debug("XML Body:\n %s", doc.toprettyxml(indent="\t"))
-        except Exception:
-            log.warning("could not print xml", exc_info=True)
-            pass
+        if method != 'GET':
+            log.debug("Body:\n%s\nEnd of body", data1)
+            try:
+                ctype = r1.msg.getheader('content-type')
+                if ctype and ';' in ctype:
+                    ctype, encoding = ctype.split(';',1)
+                if ctype == 'text/xml':
+                    doc = xml.dom.minidom.parseString(data1)
+                    log.debug("XML Body:\n %s", doc.toprettyxml(indent="\t"))
+            except Exception:
+                log.warning("could not print xml", exc_info=True)
+                pass
         conn.close()
         return r1.status, r1.msg, data1
 
@@ -636,4 +639,58 @@ class DAVClient(object):
             
         return res
 
+    def gd_get(self, path, crange=None, mime=None, compare=None):
+        """ HTTP GET for path, supporting Partial ranges
+        """
+        hdrs = { 'Accept': mime or '*/*', }
+        if crange:
+            if isinstance(crange, tuple):
+                crange = [crange,]
+            if not isinstance(crange, list):
+                raise TypeError("Range must be a tuple or list of tuples")
+            rs = []
+            for r in crange:
+                rs.append('%d-%d' % r)
+            hdrs['Range'] = 'bytes='+ (','.join(rs))
+        s, m, d = self._http_request(self.davpath + path, method='GET', hdrs=hdrs)
+        assert s in (200, 206), "Bad status: %s" % s
+        ctype = m.getheader('Content-Type').split(';',1)[0]
+        if mime:
+            assert ctype == mime, m.getheader('Content-Type')
+        rrange = None
+        rrh = m.getheader('Content-Range')
+        if rrh:
+            assert rrh.startswith('bytes '), rrh
+            rrh=rrh[6:].split('/',1)[0]
+            rrange = map(int, rrh.split('-',1))
+        if compare:
+            # we need to compare the returned data with that of compare
+            fd = open(compare, 'rb')
+            d2 = fd.read()
+            fd.close()
+            assert d2 == d, "Data does not match"
+        return ctype, rrange, d
+
+    def gd_put(self, path, body=None, srcpath=None, mime=None, noclobber=False, ):
+        """ HTTP PUT 
+            @param noclobber will prevent overwritting a resource (If-None-Match)
+            @param mime will set the content-type
+        """
+        hdrs = { }
+        if not (body or srcpath):
+            raise ValueError("PUT must have something to send")
+        if (not body) and srcpath:
+            fd = open(srcpath, 'rb')
+            body = fd.read()
+            fd.close()
+        if mime:
+            hdrs['Content-Type'] = mime
+        if noclobber:
+            hdrs['If-None-Match'] = '*'
+        s, m, d = self._http_request(self.davpath + path, method='PUT', 
+                            hdrs=hdrs, body=body)
+        assert s == (201), "Bad status: %s" % s
+        etag = m.getheader('ETag')
+        return etag or True
+
 #eof
\ No newline at end of file
-- 
1.7.10.4