Delete DIRECT_ONLY_CATEGORIES from published and draft branches.

This fixes https://openedx.atlassian.net/browse/PLAT-858.
2015-09-24 15:27:33 -04:00
parent 1f7082ebfe
commit 92eadf1afb
1 changed files with 3 additions and 0 deletions
--- a/common/lib/xmodule/xmodule/modulestore/split_mongo/split_draft.py
+++ b/common/lib/xmodule/xmodule/modulestore/split_mongo/split_draft.py
@@ -198,6 +198,9 @@ class DraftVersioningModuleStore(SplitMongoModuleStore, ModuleStoreDraftAndPubli
                branches_to_delete = [ModuleStoreEnum.BranchName.published, ModuleStoreEnum.BranchName.draft]
            elif revision is None:
                branches_to_delete = [ModuleStoreEnum.BranchName.draft]
+
+                if location.category in DIRECT_ONLY_CATEGORIES:
+                    branches_to_delete.append(ModuleStoreEnum.BranchName.published)
            else:
                raise UnsupportedRevisionError(
                    [