fix some unit tests

Author: raju-opti

lib/core/audience_evaluator/index.spec.ts

@@ -114,7 +114,7 @@ describe('lib/core/audience_evaluator', () => {
   });
 
   afterEach(() => {
-    vi.restoreAllMocks();
+    vi.resetAllMocks();
   });
 
   describe('APIs', () => {
@@ -217,10 +217,6 @@ describe('lib/core/audience_evaluator', () => {
             vi.clearAllMocks();
           });
 
-          afterEach(() => {
-            vi.resetAllMocks();
-          });
-
           afterAll(() => { 
             vi.resetAllMocks();
           });
@@ -283,30 +279,28 @@ describe('lib/core/audience_evaluator', () => {
         });
 
         describe('Audience evaluation logging', () => {
-          let mockCustomAttributeConditionEvaluator: ReturnType<typeof vi.fn>;
-          
           const evaluateSpy = conditionTreeEvaluator.evaluate as unknown as MockInstance<typeof conditionTreeEvaluator.evaluate>;
           const getEvaluatorSpy = customAttributeConditionEvaluator.getEvaluator as unknown as MockInstance<typeof customAttributeConditionEvaluator.getEvaluator>;
 
           beforeEach(() => {
             vi.clearAllMocks();
-
-            mockCustomAttributeConditionEvaluator = vi.fn();
-            getEvaluatorSpy.mockReturnValue({
-              evaluate: mockCustomAttributeConditionEvaluator,
-            });
           });
 
-          afterEach(() => {
-            vi.restoreAllMocks();
+          afterAll(() => {
+            vi.resetAllMocks();
           });
 
           it('logs correctly when conditionTreeEvaluator.evaluate returns null', () => {
             evaluateSpy.mockImplementationOnce((conditions: any, leafEvaluator) => {
               return leafEvaluator(conditions[1]);
             });
 
-            mockCustomAttributeConditionEvaluator.mockReturnValue(null);
+            const mockCustomAttributeConditionEvaluator = vi.fn().mockReturnValue(null);
+
+            getEvaluatorSpy.mockReturnValue({
+              evaluate: mockCustomAttributeConditionEvaluator,
+            });
+
             const userAttributes = { device_model: 5.5 };
             const user = getMockUserContext(userAttributes);
 
@@ -333,7 +327,11 @@ describe('lib/core/audience_evaluator', () => {
               return leafEvaluator(conditions[1]);
             });
 
-            mockCustomAttributeConditionEvaluator.mockReturnValue(true);
+            const mockCustomAttributeConditionEvaluator = vi.fn().mockReturnValue(true);
+
+            getEvaluatorSpy.mockReturnValue({
+              evaluate: mockCustomAttributeConditionEvaluator,
+            });
 
             const userAttributes = { device_model: 'iphone' };
             const user = getMockUserContext(userAttributes);
@@ -356,10 +354,16 @@ describe('lib/core/audience_evaluator', () => {
 
           it('logs correctly when conditionTreeEvaluator.evaluate returns false', () => {
             evaluateSpy.mockImplementationOnce((conditions: any, leafEvaluator) => {
+              console.log('evaluateSpy leafEvaluator called with conditions:', conditions);
               return leafEvaluator(conditions[1]);
             });
 
-            mockCustomAttributeConditionEvaluator.mockReturnValue(false);
+            const mockCustomAttributeConditionEvaluator = vi.fn().mockReturnValue(false);
+
+            getEvaluatorSpy.mockReturnValue({
+              evaluate: mockCustomAttributeConditionEvaluator,
+            });
+
 
             const userAttributes = { device_model: 'android' };
             const user = getMockUserContext(userAttributes);

lib/core/audience_evaluator/index.ts

@@ -39,6 +39,8 @@ export class AudienceEvaluator {
    * @constructor
    */
   constructor(UNSTABLE_conditionEvaluators: unknown, logger?: LoggerFacade) {
+
+    // console.log('AudienceEvaluator constructor called', customAttributeConditionEvaluator.getEvaluator());
     this.logger = logger;
     this.typeToEvaluatorMap = {
       ...UNSTABLE_conditionEvaluators as any,

vitest.config.mts

@@ -26,7 +26,7 @@ export default defineConfig({
   test: {
     onConsoleLog: () => true,
     environment: 'happy-dom',
-    include: ['**/*.spec.ts'],
+    include: ['**/audience_evaluator/*.spec.ts'],
     typecheck: {
       enabled: true,
       tsconfig: 'tsconfig.spec.json',