🐛 Fix issue Rareloop/lumberjack#26

Androlax2 · Aug 18, 2021 · 0b151d7 · 0b151d7
1 parent d877406
commit 0b151d7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/Providers/CustomPostTypesServiceProvider.php b/src/Providers/CustomPostTypesServiceProvider.php
@@ -8,10 +8,12 @@ class CustomPostTypesServiceProvider extends ServiceProvider
 {
     public function boot(Config $config)
     {
-        $postTypesToRegister = $config->get('posttypes.register');
+        add_action('init', function () use ($config) {
+            $postTypesToRegister = $config->get('posttypes.register');
 
-        foreach ($postTypesToRegister as $postType) {
-            $postType::register();
-        }
+            foreach ($postTypesToRegister as $postType) {
+                $postType::register();
+            }
+        });
     }
 }
diff --git a/tests/Unit/Providers/CustomPostTypesServiceProviderTest.php b/tests/Unit/Providers/CustomPostTypesServiceProviderTest.php
@@ -3,12 +3,14 @@
 namespace Rareloop\Lumberjack\Test;
 
 use Brain\Monkey\Functions;
+use Brain\Monkey\Actions;
 use Mockery;
 use PHPUnit\Framework\TestCase;
 use Rareloop\Lumberjack\Application;
 use Rareloop\Lumberjack\Config;
 use Rareloop\Lumberjack\Post;
 use Rareloop\Lumberjack\Providers\CustomPostTypesServiceProvider;
+use Rareloop\Lumberjack\Providers\ServiceProvider;
 use Rareloop\Lumberjack\Test\Unit\BrainMonkeyPHPUnitIntegration;
 
 class CustomPostTypesServiceProviderTest extends TestCase
@@ -30,6 +32,10 @@ public function should_call_register_post_type_for_each_configured_post_type()
         Functions\expect('register_post_type')
             ->times(2);
 
+        Actions\expectAdded('init')
+            ->once()
+            ->whenHappen(fn ($callback) => $callback());
+
         $provider = new CustomPostTypesServiceProvider($app);
         $provider->boot($config);
     }