From 26848535187821399b7b5800376161dff3ad0cc3 Mon Sep 17 00:00:00 2001
From: Guido Günther <agx@sigxcpu.org>
Date: Wed, 10 Sep 2014 13:02:27 +0200
Subject: Add copy and move methods

---
 industriart/artifactory.py | 40 ++++++++++++++++++++++++++++++++
 tests/test_artifactory.py  | 58 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 98 insertions(+)
 create mode 100644 tests/test_artifactory.py

diff --git a/industriart/artifactory.py b/industriart/artifactory.py
index 241d7ba..4a719a4 100644
--- a/industriart/artifactory.py
+++ b/industriart/artifactory.py
@@ -76,6 +76,46 @@ class Artifactory(object):
         url = posixpath.join(self.base, 'api/repositories')
         return self.get(url)
 
+    def _copy_or_move(self, action, source, target):
+        try:
+            if len(source) != 2:
+                raise ValueError("Source tuple not (repo, path)")
+        except TypeError:
+            raise TypeError("Source must be a tuple")
+
+        try:
+            if len(target) != 2:
+                raise ValueError("Target tuple not (repo, path)")
+        except TypeError:
+            raise TypeError("Source must be a tuple")
+
+        url = posixpath.join(self.base,
+                             'api', action,
+                             source[0], source[1])
+        dst = posixpath.join(target[0], target[1])
+
+        log.debug("%s %s", action.capitalize(), url)
+        # Parameter order matters for artifactory
+        return self.post(url, {'to': dst})
+
+    def copy(self, source, target):
+        """
+        Copy artifacts in the repository from soure to target
+
+        :arg tuple source: The source in the form (repo, path) to copy
+        :arg tuple target: The target in the form (repo, path) to copy
+        """
+        return self._copy_or_move('copy', source, target)
+
+    def move(self, source, target):
+        """
+        Move artifacts in the repository from soure to target
+
+        :arg tuple source: The source in the form (repo, path) to copy
+        :arg tuple target: The target in the form (repo, path) to copy
+        """
+        return self._copy_or_move('move', source, target)
+
     def _transform_url(self, url):
         """
         Helper to rewrite URLs. All artifactory URLs are passed through this method
diff --git a/tests/test_artifactory.py b/tests/test_artifactory.py
new file mode 100644
index 0000000..3e28e90
--- /dev/null
+++ b/tests/test_artifactory.py
@@ -0,0 +1,58 @@
+#!/usr/bin/python
+# vim: set fileencoding=utf-8 :
+#
+# (C) 2014 Guido Günther <agx@sigxcpu.org>
+#    This program is free software; you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation; either version 2 of the License, or
+#    (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program; if not, write to the Free Software
+#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+#
+
+import unittest
+from mock import patch
+
+from industriart.artifactory import Artifactory
+
+def request_mock(http_method, url, params):
+        return http_method, url, params
+
+class TestArtifactory(unittest.TestCase):
+    def test_copy(self):
+        a = Artifactory('http://a.exmple.com')
+        with patch.object(Artifactory, '_request') as mocked_request:
+            mocked_request.side_effect = request_mock
+            ret = a.copy(('unstable','foo'), ('release', 'bar'))
+            self.assertEquals(ret[0], 'POST')
+            self.assertEquals(ret[1], 'http://a.exmple.com/api/copy/unstable/foo')
+            self.assertDictEqual(ret[2], {'to': 'release/bar'})
+
+            # src needs to be a two element tuple
+            with self.assertRaises(ValueError):
+                a.copy('unstable', ('release', 'bar'))
+            with self.assertRaises(TypeError):
+                a.copy(10, ('release', 'bar'))
+
+            # dst needs to be a two element tuple
+            with self.assertRaises(ValueError):
+                a.copy(('unstable', 'foo'), 'release')
+            with self.assertRaises(TypeError):
+                a.copy(('unstable', 'foo'), 10)
+
+
+    def test_move(self):
+        a = Artifactory('http://a.exmple.com')
+        with patch.object(Artifactory, '_request') as mocked_request:
+            mocked_request.side_effect = request_mock
+            ret = a.move(('unstable','foo'), ('release', 'bar'))
+            self.assertEquals(ret[0], 'POST')
+            self.assertEquals(ret[1], 'http://a.exmple.com/api/move/unstable/foo')
+            self.assertDictEqual(ret[2], {'to': 'release/bar'})
-- 
cgit v1.2.3