Merge pull request #303 from alleyinteractive/hotfix/IRV-909/theme-co…

…nfig-callback

Fix ability to override theme in a config callback

inc/components/class-component.php

@@ -199,7 +199,7 @@ private function process_args( array $args ): self {
 			->apply_context()
 			->set_config( $args['config'] )
 			->hydrate_config()
-			->set_theme( 'irving/site-theme' !== $this->get_name() ? $args['config']['theme'] : $args['theme'] )
+			->set_theme( 'irving/site-theme' !== $this->get_name() ? $this->get_config( 'theme' ) : $args['theme'] )
 			->set_children( $args['children'] );
 
 		return $this;

tests/components/test-class-component.php

@@ -183,10 +183,11 @@ public function get_config_for_test_schema(): array {
 			[
 				[
 					'test-default' => 'overridden',
+					'theme'        => 'overridden',
 				],
 				[
 					'test-default' => 'overridden',
-					'theme'        => 'default',
+					'theme'        => 'overridden',
 				],
 				'Default config values not overridden.',
 			],
@@ -475,6 +476,36 @@ public function test_config_callback() {
 		);
 	}
 
+	/**
+	 * Test overriding `theme` in a config callback.
+	 */
+	public function test_config_callback_for_theme() {
+		$name = 'example/component';
+
+		/*
+		 * Register component with callback.
+		 *
+		 * We're using a callback that uses the config in order
+		 * to ensure that the callback is passed an array.
+		 */
+		get_registry()->register_component(
+			$name,
+			[
+				'config_callback' => function ( array $config ) {
+					$config['theme'] = 'fred';
+					return $config;
+				},
+			]
+		);
+
+		$component = new Component( $name );
+
+		// Clean up.
+		get_registry()->unregister_component( $name );
+
+		$this->assertEquals( 'fred', $component->get_theme() );
+	}
+
 	/**
 	 * Test children set from config.
 	 */