test: resolves broken tests

Author: JasonTheAdams

tests/_support/Helper/MockModelWithAfterConstruct.php

@@ -13,7 +13,7 @@ class MockModelWithAfterConstruct extends Model {
 	public bool $afterConstructCalled = false;
 	public array $constructedAttributes = [];
 
-	protected function afterConstruct() {
+	protected function afterConstruct(): void {
 		$this->afterConstructCalled = true;
 		$this->constructedAttributes = $this->toArray();
 	}

tests/_support/Helper/MockModelWithRelationship.php

@@ -3,8 +3,8 @@
 namespace StellarWP\Models\Tests;
 
 use StellarWP\DB\DB;
-use StellarWP\DB\QueryBuilder\QueryBuilder;
 use StellarWP\Models\Model;
+use StellarWP\Models\ModelQueryBuilder;
 use StellarWP\Models\ValueObjects\Relationship;
 
 class MockModelWithRelationship extends Model {
@@ -19,16 +19,16 @@ class MockModelWithRelationship extends Model {
 	];
 
 	/**
-	 * @return QueryBuilder
+	 * @return ModelQueryBuilder<MockModel>
 	 */
-	public function relatedAndCallableHasOne() {
-		return DB::table( 'posts' );
+	public function relatedAndCallableHasOne(): ModelQueryBuilder {
+		return ( new ModelQueryBuilder( MockModel::class ) )->from( 'posts' );
 	}
 
 	/**
-	 * @return QueryBuilder
+	 * @return ModelQueryBuilder<MockModel>
 	 */
-	public function relatedAndCallableHasMany() {
-		return DB::table( 'posts' );
+	public function relatedAndCallableHasMany(): ModelQueryBuilder {
+		return ( new ModelQueryBuilder( MockModel::class ) )->from( 'posts' );
 	}
 }