Add a package listing API to cipd.

appengine/chrome_infra_packages/cipd/test/api_test.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import datetime
 
 from testing_utils import testing
 
 from components import auth
 from components import auth_testing
@@ skipping 49 matching lines @@
     resp = self.call_api('fetch_package', {'package_name': 'good/name'})
     self.assertEqual({'status': 'PACKAGE_NOT_FOUND'}, resp.json_body)
 
   def test_fetch_package_bad_name(self):
     resp = self.call_api('fetch_package', {'package_name': 'bad name'})
     self.assertEqual({
       'status': 'ERROR',
       'error_message': 'Invalid package name',
     }, resp.json_body)
 
+  def test_list_packages_no_results(self):
+    resp = self.call_api('list_packages', {})
+    self.assertEqual({
+      'status': 'SUCCESS',
+    }, resp.json_body)
+
+  def test_list_packages_all_packages(self):
+    resp = self.call_api('register_package', {'package_name': 'good/name'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+
+    resp = self.call_api('list_packages', {'recursive': True})
+    self.assertEqual({
+      'status': 'SUCCESS',
+      'packages': ['good/name'],
+      'directories': ['good'],
+    }, resp.json_body)
+
+  def test_list_packages_filter_no_access(self):
+    resp = self.call_api('register_package', {'package_name': 'good/name'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+
+    self.mock(api.acl, 'can_fetch_package', lambda *_: False)
+
+    resp = self.call_api('list_packages', {})
+    self.assertEqual({
+      'status': 'SUCCESS',
+    }, resp.json_body)
+
+  def test_list_packages_in_path(self):
+    resp = self.call_api('register_package', {'package_name': 'p/a'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+    resp = self.call_api('register_package', {'package_name': 'p/y'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+    resp = self.call_api('register_package', {'package_name': 'p/z/z'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+    resp = self.call_api('register_package', {'package_name': 'pp'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+    resp = self.call_api('register_package', {'package_name': 'q'})
+    self.assertEqual('REGISTERED', resp.json_body['status'])
+
+    resp = self.call_api('list_packages', {
+      'path': 'p',
+      'recursive': False,
+    })
+    self.assertEqual({
+      'status': 'SUCCESS',
+      'packages': [
+        'p/a',
+        'p/y',
+      ],
+      'directories': [
+        'p/z',
+      ],
+    }, resp.json_body)
+
+    resp = self.call_api('list_packages', {
+      'path': 'p',
+      'recursive': True,
+    })
+    self.assertEqual({
+      'status': 'SUCCESS',
+      'packages': [
+        'p/a',
+        'p/y',
+        'p/z/z',
+      ],
+      'directories': [
+        'p/z',
+      ],
+    }, resp.json_body)
+
   def test_register_package(self):
     self.mock(utils, 'utcnow', lambda: datetime.datetime(2014, 1, 1))
 
     # Not yet registered.
     resp = self.call_api('register_package', {'package_name': 'good/name'})
     self.assertEqual(200, resp.status_code)
     self.assertEqual({
       'status': 'REGISTERED',
       'package': {
         'package_name': 'good/name',
@@ skipping 920 matching lines @@
     return True
 
   def generate_fetch_url(self, instance):
     return 'http://signed-url/%s' % instance.instance_id
 
   def is_instance_file_uploaded(self, package_name, instance_id):
     return instance_id in self.uploaded
 
   def create_upload_session(self, package_name, instance_id, caller):
     return 'http://upload_url', 'upload_session_id'