Setup test fixture for Plone 5

src/sc/embedder/testing.py

@@ -1,11 +1,24 @@
 # -*- coding: utf-8 -*-
+"""Setup testing infrastructure.
+
+For Plone 5 we need to install plone.app.contenttypes.
+"""
 from plone.app.robotframework.testing import AUTOLOGIN_LIBRARY_FIXTURE
 from plone.app.testing import FunctionalTesting
 from plone.app.testing import IntegrationTesting
-from plone.app.testing import PLONE_FIXTURE
 from plone.app.testing import PloneSandboxLayer
 from plone.testing import z2
 
+import pkg_resources
+
+
+try:
+    pkg_resources.get_distribution('plone.app.contenttypes')
+except pkg_resources.DistributionNotFound:
+    from plone.app.testing import PLONE_FIXTURE
+else:
+    from plone.app.contenttypes.testing import PLONE_APP_CONTENTTYPES_FIXTURE as PLONE_FIXTURE
+
 
 class Fixture(PloneSandboxLayer):
 
@@ -22,14 +35,14 @@ def setUpPloneSite(self, portal):
         self.applyProfile(portal, 'sc.embedder:default')
 
 FIXTURE = Fixture()
+
 INTEGRATION_TESTING = IntegrationTesting(
-    bases=(FIXTURE,),
-    name='sc.embedder:Integration',
-)
+    bases=(FIXTURE,), name='sc.embedder:Integration')
+
 FUNCTIONAL_TESTING = FunctionalTesting(
-    bases=(FIXTURE, z2.ZSERVER_FIXTURE),
-    name='sc.embedder:Functional',
-)
+    bases=(FIXTURE, z2.ZSERVER_FIXTURE), name='sc.embedder:Functional')
+
 ROBOT_TESTING = FunctionalTesting(
     bases=(FIXTURE, AUTOLOGIN_LIBRARY_FIXTURE, z2.ZSERVER_FIXTURE),
-    name='sc.embedder:Robot')
+    name='sc.embedder:Robot',
+)