From 4b974b038c80764691af27bbc8bf18ccdb1322ed Mon Sep 17 00:00:00 2001
From: Andrey Smirnov <Smirnov.Andrey@gmail.com>
Date: Sun, 1 Mar 2015 22:36:22 +0300
Subject: [PATCH] Test for snapshot publishing. #116

---
 system/t12_api/publish.py | 62 +++++++++++++++++++++++++++++++++++----
 1 file changed, 57 insertions(+), 5 deletions(-)

diff --git a/system/t12_api/publish.py b/system/t12_api/publish.py
index 3a02e1fb..7b27549e 100644
--- a/system/t12_api/publish.py
+++ b/system/t12_api/publish.py
@@ -27,6 +27,7 @@ class PublishAPITestRepo(APITest):
 
         self.check_equal(self.post("/api/repos/" + repo_name + "/file/" + d).status_code, 200)
 
+        # publishing under prefix, default distribution
         prefix = self.random_name()
         resp = self.post("/api/publish/" + prefix,
                          json={
@@ -34,7 +35,7 @@ class PublishAPITestRepo(APITest):
                              "Sources": [{"Name": repo_name}],
                              "Signing": DefaultSigningOptions,
                          })
-        self.check_equal(resp.status_code, 200)
+        self.check_equal(resp.status_code, 201)
         self.check_equal(resp.json(), {
             'Architectures': ['i386', 'source'],
             'Distribution': 'wheezy',
@@ -45,13 +46,64 @@ class PublishAPITestRepo(APITest):
             'Sources': [{'Component': 'main', 'Name': repo_name}],
             'Storage': ''})
 
+        # publishing under root, custom distribution, architectures
+        distribution = self.random_name()
+        resp = self.post("/api/publish",
+                         json={
+                             "SourceKind": "local",
+                             "Sources": [{"Name": repo_name}],
+                             "Signing": DefaultSigningOptions,
+                             "Distribution": distribution,
+                             "Architectures": ["i386", "amd64"],
+                         })
+        self.check_equal(resp.status_code, 201)
+        self.check_equal(resp.json(), {
+            'Architectures': ['amd64', 'i386'],
+            'Distribution': distribution,
+            'Label': '',
+            'Origin': '',
+            'Prefix': ".",
+            'SourceKind': 'local',
+            'Sources': [{'Component': 'main', 'Name': repo_name}],
+            'Storage': ''})
+
 
 class PublishSnapshotAPITest(APITest):
     """
-    POST /publish/:prefix/snapshot
-
-    XXX: test me when snapshot API becomes available
+    POST /publish/:prefix (snapshots)
     """
 
     def check(self):
-        pass
+        repo_name = self.random_name()
+        snapshot_name = self.random_name()
+        self.check_equal(self.post("/api/repos", json={"Name": repo_name}).status_code, 201)
+
+        resp = self.post("/api/repos/" + repo_name + '/snapshots', json={'Name': snapshot_name})
+        self.check_equal(resp.status_code, 400)
+
+        d = self.random_name()
+        self.check_equal(self.upload("/api/files/" + d,
+                         "libboost-program-options-dev_1.49.0.1_i386.deb").status_code, 200)
+
+        self.check_equal(self.post("/api/repos/" + repo_name + "/file/" + d).status_code, 200)
+
+        self.check_equal(self.post("/api/repos/" + repo_name + '/snapshots', json={'Name': snapshot_name}).status_code, 201)
+
+        prefix = self.random_name()
+        resp = self.post("/api/publish/" + prefix,
+                         json={
+                             "SourceKind": "snapshot",
+                             "Sources": [{"Name": snapshot_name}],
+                             "Signing": DefaultSigningOptions,
+                             "Distribution": "squeeze",
+                         })
+        self.check_equal(resp.status_code, 201)
+        self.check_equal(resp.json(), {
+            'Architectures': ['i386'],
+            'Distribution': 'squeeze',
+            'Label': '',
+            'Origin': '',
+            'Prefix': prefix,
+            'SourceKind': 'snapshot',
+            'Sources': [{'Component': 'main', 'Name': snapshot_name}],
+            'Storage': ''})