register IntegrityService correctly (#16543)

Signed-off-by: William Woodruff <[email protected]>

Author: woodruffw

tests/unit/attestations/test_init.py

@@ -0,0 +1,31 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pretend
+
+from warehouse import attestations
+from warehouse.attestations.interfaces import IIntegrityService
+from warehouse.attestations.services import IntegrityService
+
+
+def test_includeme():
+    config = pretend.stub(
+        register_service_factory=pretend.call_recorder(
+            lambda factory, iface, name=None: None
+        ),
+    )
+
+    attestations.includeme(config)
+
+    assert config.register_service_factory.calls == [
+        pretend.call(IntegrityService.create_service, IIntegrityService),
+    ]

warehouse/attestations/__init__.py

@@ -25,3 +25,7 @@
     "IntegrityService",
     "UnsupportedPublisherError",
 ]
+
+
+def includeme(config):
+    config.register_service_factory(IntegrityService.create_service, IIntegrityService)