Add test

Author: elpete

tests/resources/app/models/InventoryBatch.cfc

@@ -0,0 +1,10 @@
+component table="inventory_batches" extends="quick.models.BaseEntity" accessors="true" {
+
+    property name="id";
+
+    function scopeAddTradeCount( qb ) {
+        qb.selectRaw(" (SELECT count(id) FROM trades WHERE trades.inventoryBatchId = inventory_batches.id) as tradeCount");
+        appendVirtualAttribute( "tradeCount" );
+    }
+
+}

tests/resources/database/migrations/2025_06_11_104851_create_inventory_batches_table.cfc

@@ -0,0 +1,35 @@
+component {
+
+    function up( schema, qb ) {
+        schema.create( "inventory_batches", function( t ) {
+            t.unsignedInteger( "id" ).primaryKey();
+            t.string( "name" );
+        } );
+
+        schema.create( "trades", function( t ) {
+            t.increments( "id" );
+            t.unsignedInteger( "inventoryBatchId" ).references( "id" ).onTable( "inventory_batches" );
+        } );
+
+        qb.newQuery().table( "inventory_batches" ).insert( [
+            { "id": 1, "name": "Batch 1" },
+            { "id": 2, "name": "Batch 2" },
+            { "id": 3, "name": "Batch 3" }
+        ] );
+
+        qb.table( "trades" ).insert( [
+            { "id": 1, "inventoryBatchId": 1 },
+            { "id": 2, "inventoryBatchId": 1 },
+            { "id": 3, "inventoryBatchId": 2 },
+            { "id": 4, "inventoryBatchId": 3 },
+            { "id": 5, "inventoryBatchId": 3 },
+            { "id": 6, "inventoryBatchId": 3 }
+        ] );
+    }
+
+    function down( schema, qb ) {
+        schema.drop( "trades" );
+        schema.drop( "inventory_batches" );
+    }
+
+}

tests/specs/integration/BaseEntity/AttributeSpec.cfc

@@ -17,6 +17,25 @@ component extends="tests.resources.ModuleIntegrationSpec" {
 				expect( attributesFromFresh ).notToInclude( "latestPostId" );
 			} );
 
+			it( "can load virtual attributes from a query", () => {
+				var batches = getInstance( "InventoryBatch" )
+					.addTradeCount()
+					.orderByAsc( "id" )
+					.get();
+
+				expect( batches ).toBeArray();
+				expect( batches ).toHaveLength( 3 );
+
+				expect( batches[ 1 ].getId() ).toBe( 1 );
+				expect( batches[ 1 ].getTradeCount() ).toBe( 2 );
+
+				expect( batches[ 2 ].getId() ).toBe( 2 );
+				expect( batches[ 2 ].getTradeCount() ).toBe( 1 );
+
+				expect( batches[ 3 ].getId() ).toBe( 3 );
+				expect( batches[ 3 ].getTradeCount() ).toBe( 3 );
+			} );
+
 			it( "can get any attribute using the `getColumnName` magic methods", function() {
 				var user = getInstance( "User" ).find( 1 );
 				expect( user.getId() ).toBe( 1 );