Add PenExtension class

Author: adazem009

src/CMakeLists.txt

@@ -55,6 +55,8 @@ qt_add_qml_module(scratchcpp-render
 	penlayerpainter.cpp
 	penlayerpainter.h
 	penattributes.h
+	blocks/penextension.cpp
+	blocks/penextension.h
 	blocks/penblocks.cpp
 	blocks/penblocks.h
 )

src/blocks/penextension.cpp

@@ -0,0 +1,24 @@
+// SPDX-License-Identifier: LGPL-3.0-or-later
+
+#include <scratchcpp/iengine.h>
+
+#include "penextension.h"
+#include "penblocks.h"
+
+using namespace scratchcpprender;
+using namespace libscratchcpp;
+
+std::string PenExtension::name() const
+{
+    return "pen";
+}
+
+std::string PenExtension::description() const
+{
+    return "Pen extension";
+}
+
+void PenExtension::registerSections(IEngine *engine)
+{
+    engine->registerSection(std::make_shared<PenBlocks>());
+}

src/blocks/penextension.h

@@ -0,0 +1,19 @@
+// SPDX-License-Identifier: LGPL-3.0-or-later
+
+#pragma once
+
+#include <scratchcpp/iextension.h>
+
+namespace scratchcpprender
+{
+
+class PenExtension : public libscratchcpp::IExtension
+{
+    public:
+        std::string name() const override;
+        std::string description() const override;
+
+        void registerSections(libscratchcpp::IEngine *engine) override;
+};
+
+} // namespace scratchcpprender

test/blocks/CMakeLists.txt

@@ -15,3 +15,21 @@ target_link_libraries(
 
 add_test(pen_blocks_test)
 gtest_discover_tests(pen_blocks_test)
+
+# penextension_test
+add_executable(
+  penextension_test
+  penextension_test.cpp
+)
+
+target_link_libraries(
+  penextension_test
+  GTest::gtest_main
+  GTest::gmock_main
+  scratchcpp-render
+  scratchcpprender_mocks
+  ${QT_LIBS}
+)
+
+add_test(penextension_test)
+gtest_discover_tests(penextension_test)

test/blocks/penextension_test.cpp

@@ -0,0 +1,39 @@
+#include <blocks/penextension.h>
+#include <blocks/penblocks.h>
+#include <enginemock.h>
+
+#include "../common.h"
+
+using namespace scratchcpprender;
+using namespace libscratchcpp;
+
+using ::testing::WithArgs;
+using ::testing::Invoke;
+using ::testing::_;
+
+TEST(PenExtensionTest, Name)
+{
+    PenExtension ext;
+    ASSERT_EQ(ext.name(), "pen");
+}
+
+TEST(PenExtensionTest, Description)
+{
+    PenExtension ext;
+    ASSERT_EQ(ext.description(), "Pen extension");
+}
+
+TEST(PenExtensionTest, IncludeByDefault)
+{
+    PenExtension ext;
+    ASSERT_FALSE(ext.includeByDefault());
+}
+
+TEST(PenExtensionTest, RegisterSections)
+{
+    PenExtension ext;
+    EngineMock engine;
+
+    EXPECT_CALL(engine, registerSection(_)).WillOnce(WithArgs<0>(Invoke([](std::shared_ptr<IBlockSection> section) { ASSERT_TRUE(dynamic_cast<PenBlocks *>(section.get())); })));
+    ext.registerSections(&engine);
+}