feat(model-factory): add add post model and add term model to not erase all registered models in factory (#32)

Amorfx · web-flow · commit 6b0712adbbc9 · 2024-01-03T23:15:29.000+01:00
diff --git a/src/Model/ModelFactory.php b/src/Model/ModelFactory.php
@@ -51,6 +51,10 @@ public function registerPostModel(array $postModels): void {
         self::$postTypeModelClasses = $postModels;
     }
 
+    public function addPostModel(array $postModels): void {
+        self::$postTypeModelClasses = array_merge(self::$postTypeModelClasses, $postModels);
+    }
+
     /**
      * Use in Simply Plugin to register your term type models
      *
@@ -60,6 +64,10 @@ public function registerTermModel(array $termModels): void {
         self::$termTypeModelClasses = $termModels;
     }
 
+    public function addTermModel(array $termModels): void {
+        self::$termTypeModelClasses = array_merge(self::$termTypeModelClasses, $termModels);
+    }
+
     /**
      * Set key to have a clean mapping
      *
diff --git a/tests/unit/Model/ModelFactoryTest.php b/tests/unit/Model/ModelFactoryTest.php
@@ -78,4 +78,18 @@ public function testRegisterModels() {
         $this->assertSame($expected, $factoryReflection->getStaticPropertyValue('postTypeModelClasses'));
         $this->assertSame($expected, $factoryReflection->getStaticPropertyValue('termTypeModelClasses'));
     }
+    
+    public function testRegisterMultipleModelsWithMultipleCall() {
+        $factory = new ModelFactory();
+        $classA = 'ClassA';
+        $classB = 'ClassB';
+        $expected = array($classA, $classB);
+        $factory->registerPostModel(array($classA));
+        $factory->registerTermModel(array($classA));
+        $factory->addPostModel(array($classB));
+        $factory->addTermModel(array($classB));
+        $factoryReflection = new \ReflectionClass($factory);
+        $this->assertSame($expected, $factoryReflection->getStaticPropertyValue('postTypeModelClasses'));
+        $this->assertSame($expected, $factoryReflection->getStaticPropertyValue('termTypeModelClasses'));
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,10 @@ public function registerPostModel(array $postModels): void {`
`51`	`51`	`self::$postTypeModelClasses = $postModels;`
`52`	`52`	`}`
`53`	`53`
	`54`	`+ public function addPostModel(array $postModels): void {`
	`55`	`+ self::$postTypeModelClasses = array_merge(self::$postTypeModelClasses, $postModels);`
	`56`	`+ }`
	`57`	`+`
`54`	`58`	`/**`
`55`	`59`	`* Use in Simply Plugin to register your term type models`
`56`	`60`	`*`
`@@ -60,6 +64,10 @@ public function registerTermModel(array $termModels): void {`
`60`	`64`	`self::$termTypeModelClasses = $termModels;`
`61`	`65`	`}`
`62`	`66`
	`67`	`+ public function addTermModel(array $termModels): void {`
	`68`	`+ self::$termTypeModelClasses = array_merge(self::$termTypeModelClasses, $termModels);`
	`69`	`+ }`
	`70`	`+`
`63`	`71`	`/**`
`64`	`72`	`* Set key to have a clean mapping`
`65`	`73`	`*`