Add an index for db.charms.

charmworld/testing/factory.py

 # Copyright 2012, 2013 Canonical Ltd.  This software is licensed under the
 # GNU Affero General Public License version 3 (see the file LICENSE).
 
 """Helpers to generate test data for use."""
 
 from bzrlib.bzrdir import BzrDir
 from contextlib import contextmanager
 from calendar import timegm
 from datetime import (
     date,
@@ skipping 238 matching lines @@
     if payload is None:
         charm = get_payload_json(name, bname, series, owner, promulgated,
                                  branch_deleted)
     else:
         for key in ['name', 'bname', 'series', 'owner']:
             if locals()[key] is not None:
                 raise ValueError(
                     'Cannot specify "%s" if payload supplied.' % key)
         charm = dict(payload)
     add_store_data(charm, promulgated, charm_error, store_revision)
+    _id = construct_charm_id(charm, True)
     charm.update({
-        '_id': construct_charm_id(charm, revision),
+        '_id': _id,
         'address': get_address(charm, promulgated),
         "branch_dir": os.path.join(branch_root, charm['name']),
         "categories": categories,
         "changes": changes,
         "description": description,
         "hooks": [
             "install",
             "start",
             "stop",
-            "website-relation-joined"
+            "website-relation-joined",
         ],
         "maintainer": maintainer,
         "proof": proof,
         "provides": provides,
         'requires': requires,
         "revision": revision,
         "summary": summary,
         'config': {
             'options': options,
         },
         'files': files,
         'downloads': downloads,
         'downloads_in_past_30_days': downloads_in_past_30_days,
         'date_created': date_created.isoformat(),
     })
     if empty_branch:
         charm["first_change"] = None
         charm["last_change"] = None
     else:
         charm["first_change"] = {
             "committer": "Jane Doe <jane@sample.com>",
             "created": 1308093138.892,
             "message": "initial checkin\n",
-            "revno": 1
+            "revno": 1,
         }
         charm["last_change"] = {
             "committer": "John Doe <jdoe@example.com>",
             "created": 1343082725.06,
             "message": commit_message,
             "revno": revno,
         }
     if tests is not None:
         charm['tests'] = tests
         charm['test_results'] = {}
         for provider, status in tests.items():
             charm['test_results'][provider] = makeTestResult(
                 charm['name'], charm['branch_spec'], status, provider)
     if subordinate:
         charm['subordinate'] = True
     if hash_ is not None:
         charm['hash'] = hash_
     return process_charm(charm)
 
 
 def makeCharm(db, *args, **kwargs):
     charm = get_charm_json(*args, **kwargs)
     record_id = db.charms.insert(charm)
     return record_id, charm
 
 
+def makeCharms(db, num, revisions=1, *args, **kwargs):
+    """Make a bunch of charms.  Overwrite"""

[j.c.sackett, 2014/02/10 22:28:57]

Overwrite what?

[bac, 2014/02/11 15:41:00]

Done.

+    charms = dict()
+    name = kwargs.get('name')
+    for i in xrange(num):
+        if name is None:
+            kwargs['name'] = random_string(20)
+        kwargs['owner'] = random_string(10)
+        for revision in xrange(revisions):
+            kwargs['store_revision'] = kwargs['revision'] = revision + 1
+            record_id, charm = makeCharm(db, *args, **kwargs)
+            charms[record_id] = charm
+    return charms
+
+
 def make_charm(db, **kwargs):
     """Create a new charm and add it to the database.
 
     This function returns a charm model.
     """
     charm_data = get_charm_json(**kwargs)
     db.charms.insert(charm_data)
     return Charm(charm_data)
 
 
@@ skipping 469 matching lines @@
 
 @contextmanager
 def bundle_branch(bundle_path=None, extra=[]):
     """Create a Bazaar branch from the files in bundle_path."""
     # Avoid circular imports.
     if bundle_path is None:
         bundle_path = os.path.split(__file__)[0]
         bundle_path = os.path.join(bundle_path, 'data/sample_bundle')
     with _sample_branch(bundle_path, extra) as (branch_dir, rev):
         yield branch_dir, rev, bundle_path