Set engine of text bubbles in Target

Author: adazem009

include/scratchcpp/drawable.h

@@ -29,7 +29,7 @@ class LIBSCRATCHCPP_EXPORT Drawable
         virtual void setLayerOrder(int newLayerOrder);
 
         IEngine *engine() const;
-        void setEngine(IEngine *engine);
+        virtual void setEngine(IEngine *engine);
 
     private:
         spimpl::unique_impl_ptr<DrawablePrivate> impl;

include/scratchcpp/target.h

@@ -103,6 +103,8 @@ class LIBSCRATCHCPP_EXPORT Target : public Drawable
         TextBubble *bubble();
         const TextBubble *bubble() const;
 
+        void setEngine(IEngine *engine) override final;
+
     protected:
         /*! Override this method to set a custom data source for blocks, assets, comments, etc. */
         virtual Target *dataSource() const { return nullptr; }

src/scratch/target.cpp

@@ -572,3 +572,10 @@ const TextBubble *Target::bubble() const
 {
     return &impl->bubble;
 }
+
+/*! Overrides Drawable#setEngine(). */
+void Target::setEngine(IEngine *engine)
+{
+    Drawable::setEngine(engine);
+    impl->bubble.setEngine(engine);
+}

test/scratch_classes/target_test.cpp

@@ -797,6 +797,18 @@ TEST(SpriteTest, BubbleTextRedraw)
     target.bubble()->setText("");
 }
 
+TEST(TargetTest, Engine)
+{
+    Target target;
+    ASSERT_EQ(target.engine(), nullptr);
+    ASSERT_EQ(target.bubble()->engine(), nullptr);
+
+    EngineMock engine;
+    target.setEngine(&engine);
+    ASSERT_EQ(target.engine(), &engine);
+    ASSERT_EQ(target.bubble()->engine(), &engine);
+}
+
 TEST(TargetTest, DataSource)
 {
     TargetMock target;