From 4be2419349bfce525ab0186a12a0d0be864d0368 Mon Sep 17 00:00:00 2001 From: Dennis Kliban Date: Thu, 27 Aug 2015 16:31:15 +0000 Subject: [PATCH] Drops celery_taskmeta collection re #1084 https://pulp.plan.io/issues/1084 --- .../migrations/0020_drop_celery_taskmeta.py | 18 +++++++++++++++ .../test_0020_drop_celery_taskmeta.py | 23 +++++++++++++++++++ 2 files changed, 41 insertions(+) create mode 100644 server/pulp/server/db/migrations/0020_drop_celery_taskmeta.py create mode 100644 server/test/unit/server/db/migrations/test_0020_drop_celery_taskmeta.py diff --git a/server/pulp/server/db/migrations/0020_drop_celery_taskmeta.py b/server/pulp/server/db/migrations/0020_drop_celery_taskmeta.py new file mode 100644 index 0000000000..01e0cad9cb --- /dev/null +++ b/server/pulp/server/db/migrations/0020_drop_celery_taskmeta.py @@ -0,0 +1,18 @@ +""" +This migration removes the `celery_taskmeta` collection. +""" +from pulp.server.db import connection + + +def migrate(*args, **kwargs): + """ + Perform the migration as described in this module's docblock. + + :param args: unused + :type args: list + :param kwargs: unused + :type kwargs: dict + """ + db = connection.get_database() + collection = db['celery_taskmeta'] + collection.drop() diff --git a/server/test/unit/server/db/migrations/test_0020_drop_celery_taskmeta.py b/server/test/unit/server/db/migrations/test_0020_drop_celery_taskmeta.py new file mode 100644 index 0000000000..8c02b13cb6 --- /dev/null +++ b/server/test/unit/server/db/migrations/test_0020_drop_celery_taskmeta.py @@ -0,0 +1,23 @@ +""" +This module contains tests for pulp.server.db.migrations.0020_drop_celery_taskmeta.py +""" +import unittest + +from mock import patch + +from pulp.server.db.migrate.models import _import_all_the_way + +migration = _import_all_the_way('pulp.server.db.migrations.0020_drop_celery_taskmeta') + + +class TestMigrate(unittest.TestCase): + """ + Test the migrate() function. + """ + + @patch.object(migration.connection, 'get_database') + def test_celery_taskmeta_collection_dropped(self, mock_get_database): + mock_get_database.return_value.collection_names.return_value = [] + collection = mock_get_database.return_value['celery_taskmeta'] + migration.migrate() + collection.drop.assert_called_once()