← Back to branch summary

~jcsackett/charmworld/bac-tag-constraints

« back to all changes in this revision

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

  • Committer: Tarmac
  • Author(s): Aaron Bentley
  • Date: 2013-08-06 14:09:18 UTC
  • mfrom: (328.1.7 restore-questions)
  • Revision ID: tarmac-20130806140918-d8nkgw3oynufkb7p
[r=bac][bug=][author=abentley] Restore QA questions to migrations.

Show diffs side-by-side

added added

removed removed

Lines of Context:
charmworld/migrations/versions/tests/test_migrations.py
6
6
)
7
7
from charmworld.models import CharmSource
8
8
from charmworld.testing import MongoTestBase
9
 
import migrations.migrate
10
 
from migrations.migrate import Versions
 
9
from charmworld.migrations import migrate
 
10
from charmworld.migrations.migrate import Versions
11
11
 
12
12
 
13
13
def fake():
18
18
 
19
19
    def setUp(self):
20
20
        super(MigrationTestBase, self).setUp()
21
 
        self.versions = Versions('migrations/versions/')
22
 
        true_configure_logging = migrations.migrate.configure_logging
23
 
        migrations.migrate.configure_logging = fake
 
21
        self.versions = Versions(migrate.get_migration_path())
 
22
        true_configure_logging = migrate.configure_logging
 
23
        migrate.configure_logging = fake
24
24
        self.addCleanup(
25
 
            setattr, migrations.migrate,
26
 
            'comfigure_logging', true_configure_logging)
 
25
            setattr, migrate,
 
26
            'configure_logging', true_configure_logging)
27
27
 
28
28
 
29
29
class TestMigration010(MigrationTestBase):
78
78
            self.index_client.get('e')
79
79
        self.assertEqual({'_id': 'e', 'config': ['a']},
80
80
                         self.db.charms.find_one({'_id': 'e'}))
 
81
 
 
82
 
 
83
class TestMigration011(MigrationTestBase):
 
84
 
 
85
    def test_migration(self):
 
86
        self.use_index_client()
 
87
        source = CharmSource.from_request(self)
 
88
        self.assertItemsEqual([], self.db.qa.find())
 
89
        self.versions.run_migration(self.db, self.index_client,
 
90
                                    '011_ensure_category_questions.py')
 
91
        self.assertIn('reliable', [cat['name'] for cat in self.db.qa.find()])