Refactor to use a Post class

References #78

web/modules/custom/custom/custom.module

@@ -7,14 +7,15 @@
 
 declare(strict_types=1);
 
-use Drupal\custom\Entity\Node;
+use Drupal\discoverable_entity_bundle_classes\Storage\Node\NodeStorage;
 
 /**
  * Implements hook_entity_type_build().
  */
 function custom_entity_type_build(array &$entityTypes): void {
+	/** @var \Drupal\Core\Entity\EntityTypeInterface[] $entityTypes */
   if (isset($entityTypes['node'])) {
-    $entityTypes['node']->setClass(Node::class);
+    $entityTypes['node']->setStorageClass(NodeStorage::class);
   }
 }
 

web/modules/custom/custom/src/Entity/Node/Post.php

@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Drupal\custom\Entity\Node;
+
+use Drupal\node\Entity\Node;
+use Drupal\discoverable_entity_bundle_classes\ContentEntityBundleInterface;
+
+/**
+ * Defines an blog post node class.
+ *
+ * @ContentEntityBundleClass(
+ *   label = @Translation("Blog post"),
+ *   entity_type = "node",
+ *   bundle = "post"
+ * );
+ */
+class Post extends Node implements ContentEntityBundleInterface {
+
+  public function hasTweet(): bool {
+    return (bool) $this->get('field_has_tweet')->getString();
+  }
+
+}

web/modules/custom/custom/tests/src/Kernel/TalksTestBase.php

@@ -22,6 +22,7 @@ abstract class TalksTestBase extends EntityKernelTestBase {
     'datetime',
 
     // Contrib.
+    'discoverable_entity_bundle_classes',
     'entity_reference_revisions',
     'paragraphs',
     'hook_event_dispatcher',