~jcsackett/charmworld/bac-tag-constraints

« back to all changes in this revision

Viewing changes to charmworld/migrations/versions/tests/test_migrations.py

[r=sinzui][bug=1208477][author=abentley] Remove doctype attribute.

Show diffs side-by-side

added added

removed removed

Lines of Context:
3
3
 
4
4
from charmworld.models import (
5
5
    CharmSource,
6
 
    FeaturedSource,
7
6
    QADataSource,
8
7
)
9
8
from charmworld.testing import (
161
160
                                    '012_migrate_qa_representation.py')
162
161
        for charm_data in source._get_all(charm['_id']):
163
162
            self.assertNotIn('qa', charm_data)
164
 
 
165
 
 
166
 
class TestMigration013(MigrationTestBase):
167
 
 
168
 
    def test_migration(self):
169
 
        self.use_index_client()
170
 
        featured_source = FeaturedSource.from_db(self.db)
171
 
        charm_data = factory.get_charm_json(name='featured-charm')
172
 
        charm_data['is_featured'] = True
173
 
        source = CharmSource.from_request(self).save(charm_data)
174
 
        self.assertEqual([], list(self.db.featured.find()))
175
 
        self.versions.run_migration(
176
 
            self.db, self.index_client, '013_migrate_featured.py')
177
 
        self.assertIn(
178
 
            'featured-charm',
179
 
            [featured['name'] for featured in self.db.featured.find()])
180
 
        self.assertEqual(len(list(self.db.featured.find())), 1)