aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGuido Günther <agx@sigxcpu.org>2014-09-10 13:02:27 +0200
committerGuido Günther <agx@sigxcpu.org>2014-09-10 14:50:11 +0200
commit26848535187821399b7b5800376161dff3ad0cc3 (patch)
tree90b379adb24f9134b04c2c52b345ad2dac5d4fc0
parent370cda042d68706af27115893d062e42f8480424 (diff)
Add copy and move methods
-rw-r--r--industriart/artifactory.py40
-rw-r--r--tests/test_artifactory.py58
2 files changed, 98 insertions, 0 deletions
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'})