analogjs · brandonroberts · Oct 27, 2025 · Oct 27, 2025 · Oct 27, 2025
diff --git a/apps/analog-app/src/app/pages/package.page.ts b/apps/analog-app/src/app/pages/package.page.ts
@@ -1,8 +1,8 @@
 import { Component } from '@angular/core';
-import { MyPackageComponent } from 'my-package';
+import { MyPackage } from 'my-package';
 
 @Component({
-  imports: [MyPackageComponent],
+  imports: [MyPackage],
   template: ` <lib-my-package /> `,
 })
 export default class PackagePageComponent {}
diff --git a/libs/my-package/src/index.ts b/libs/my-package/src/index.ts
@@ -1 +1 @@
-export * from './lib/my-package/my-package.component';
+export * from './lib/my-package/my-package';
diff --git a/libs/my-package/src/lib/my-package/allow-overrides-with-aot.spec.ts b/libs/my-package/src/lib/my-package/allow-overrides-with-aot.spec.ts
@@ -0,0 +1,16 @@
+import { TestBed } from '@angular/core/testing';
+import { MyPackage } from './my-package';
+import { test } from 'vitest';
+
+test('overrides should work even when AOT is enabled', () => {
+  /* AOT is enabled through `angular({jit: false})` in `../../../vite.config.ts`. */
+
+  TestBed.overrideComponent(MyPackage, {
+    set: {
+      template: 'My Overriden Package',
+    },
+  });
+
+  const fixture = TestBed.createComponent(MyPackage);
+  expect(fixture.nativeElement.textContent).toBe('My Overriden Package');
+});
diff --git a/libs/my-package/src/lib/my-package/my-package.component.css b/libs/my-package/src/lib/my-package/my-package.component.css
diff --git a/libs/my-package/src/lib/my-package/my-package.component.html b/libs/my-package/src/lib/my-package/my-package.component.html
diff --git a/libs/my-package/src/lib/my-package/my-package.component.spec.ts b/libs/my-package/src/lib/my-package/my-package.component.spec.ts
diff --git a/libs/my-package/src/lib/my-package/my-package.component.ts b/libs/my-package/src/lib/my-package/my-package.component.ts
diff --git a/libs/my-package/src/lib/my-package/my-package.ts b/libs/my-package/src/lib/my-package/my-package.ts
@@ -0,0 +1,8 @@
+import { ChangeDetectionStrategy, Component } from '@angular/core';
+import { CommonModule } from '@angular/common';
+
+@Component({
+  selector: 'lib-my-package',
+  template: `My Package`,
+})
+export class MyPackage {}
diff --git a/libs/my-package/vite.config.ts b/libs/my-package/vite.config.ts
@@ -8,7 +8,7 @@ export default defineConfig(({ mode }) => ({
   root: __dirname,
   cacheDir: '../../node_modules/.vite/libs/my-package',
   plugins: [
-    angular(),
+    angular({ jit: false }),
     nxViteTsPaths(),
     nxCopyAssetsPlugin(['*.md', 'package.json']),
   ],

diff --git a/packages/vite-plugin-angular/src/lib/angular-vite-plugin.ts b/packages/vite-plugin-angular/src/lib/angular-vite-plugin.ts
@@ -787,6 +787,11 @@ export function angular(options?: PluginOptions): Plugin[] {
       tsCompilerOptions['inlineSources'] = true;
     }
 
+    if (isTest) {
+      // Allow `TestBed.overrideXXX()` APIs.
+      tsCompilerOptions['supportTestBed'] = true;
+    }
+
     const replacements: string[] = pluginOptions.fileReplacements.map((rp) =>
       join(
         pluginOptions.workspaceRoot,
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		export * from './lib/my-package/my-package.component';
		export * from './lib/my-package/my-package';