OLD | NEW |
1 # Copyright 2012, 2013 Canonical Ltd. This software is licensed under the | 1 # Copyright 2012, 2013 Canonical Ltd. This software is licensed under the |
2 # GNU Affero General Public License version 3 (see the file LICENSE). | 2 # GNU Affero General Public License version 3 (see the file LICENSE). |
3 | 3 |
4 import textwrap | 4 import textwrap |
5 import yaml | 5 import yaml |
6 | 6 |
7 from charmworld.models import ( | 7 from charmworld.models import ( |
8 Bundle, | 8 Bundle, |
9 CharmSource, | 9 CharmSource, |
10 store_bundles, | 10 store_bundles, |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
53 parsed = yaml.safe_load(deployer_config) | 53 parsed = yaml.safe_load(deployer_config) |
54 owner = 'owner' | 54 owner = 'owner' |
55 bundle_name = 'wordpress-stage' | 55 bundle_name = 'wordpress-stage' |
56 basket_name = 'wordpress-basket' | 56 basket_name = 'wordpress-basket' |
57 basket_rev = 3 | 57 basket_rev = 3 |
58 basket_id = "%s/%d" % (basket_name, basket_rev) | 58 basket_id = "%s/%d" % (basket_name, basket_rev) |
59 self.search_id = Bundle.construct_id( | 59 self.search_id = Bundle.construct_id( |
60 owner, basket_name, bundle_name) | 60 owner, basket_name, bundle_name) |
61 store_bundles( | 61 store_bundles( |
62 self.db.bundles, parsed, owner, basket_id, | 62 self.db.bundles, parsed, owner, basket_id, |
63 None, None, None, | 63 None, None, None, None, |
64 index_client=self.index_client) | 64 index_client=self.index_client) |
65 | 65 |
66 def test_bundles_are_removed_from_elastic_search(self): | 66 def test_bundles_are_removed_from_elastic_search(self): |
67 assert self.index_client.get(self.search_id, BUNDLE), ( | 67 assert self.index_client.get(self.search_id, BUNDLE), ( |
68 'the test bundle is missing') | 68 'the test bundle is missing') |
69 self.versions.run_migration( | 69 self.versions.run_migration( |
70 self.db, self.index_client, '018_delete_all_bundles.py') | 70 self.db, self.index_client, '018_delete_all_bundles.py') |
71 self.assertIsNone(self.index_client.get(self.search_id, BUNDLE)) | 71 self.assertIsNone(self.index_client.get(self.search_id, BUNDLE)) |
72 | 72 |
73 def test_bundles_are_removed_from_mongo(self): | 73 def test_bundles_are_removed_from_mongo(self): |
(...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
159 | 159 |
160 def test_fails_on_unknown(self): | 160 def test_fails_on_unknown(self): |
161 review_entries = [factory.make_review_entry()] | 161 review_entries = [factory.make_review_entry()] |
162 review_entries[0]['item'] = 'example.com' | 162 review_entries[0]['item'] = 'example.com' |
163 self.db.review_queue.insert(review_entries) | 163 self.db.review_queue.insert(review_entries) |
164 self.assertRaises( | 164 self.assertRaises( |
165 Exception, | 165 Exception, |
166 self.versions.run_migration, | 166 self.versions.run_migration, |
167 self.db, self.index_client, | 167 self.db, self.index_client, |
168 '020_adds_type_field_to_existing_review_items.py') | 168 '020_adds_type_field_to_existing_review_items.py') |
OLD | NEW |