cond_tree

Author: raju-opti

lib/core/audience_evaluator/index.spec.ts

@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { beforeEach, afterEach, describe, it, vi, expect, afterAll } from 'vitest';
+import { beforeEach, afterEach, describe, it, vi, expect, afterAll, MockInstance } from 'vitest';
 
 import AudienceEvaluator, { createAudienceEvaluator } from './index';
 import * as conditionTreeEvaluator from '../condition_tree_evaluator';
@@ -23,6 +23,9 @@ import { getMockLogger } from '../../tests/mock/mock_logger';
 import { Audience, OptimizelyDecideOption, OptimizelyDecision } from '../../shared_types';
 import { IOptimizelyUserContext } from '../../optimizely_user_context';
 
+vi.mock('../condition_tree_evaluator', { spy: true });
+vi.mock('../custom_attribute_condition_evaluator', { spy: true });
+
 let mockLogger = getMockLogger();
 
 const getMockUserContext = (attributes?: unknown, segments?: string[]): IOptimizelyUserContext => ({
@@ -207,6 +210,9 @@ describe('lib/core/audience_evaluator', () => {
         });
 
         describe('integration with dependencies', () => {
+          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();
           });
@@ -220,7 +226,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('returns true if conditionTreeEvaluator.evaluate returns true', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockReturnValue(true);
+            evaluateSpy.mockReturnValue(true);
             const result = audienceEvaluator.evaluate(
               ['or', '0', '1'],
               audiencesById,
@@ -230,7 +236,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('returns false if conditionTreeEvaluator.evaluate returns false', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockReturnValue(false);
+            evaluateSpy.mockReturnValue(false);
             const result = audienceEvaluator.evaluate(
               ['or', '0', '1'],
               audiencesById,
@@ -240,7 +246,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('returns false if conditionTreeEvaluator.evaluate returns null', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockReturnValue(null);
+            evaluateSpy.mockReturnValue(null);
             const result = audienceEvaluator.evaluate(
               ['or', '0', '1'],
               audiencesById,
@@ -250,13 +256,13 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('calls customAttributeConditionEvaluator.evaluate in the leaf evaluator for audience conditions', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockImplementation((conditions: any, leafEvaluator) => {
+            evaluateSpy.mockImplementation((conditions: any, leafEvaluator: any) => {
               return leafEvaluator(conditions[1]);
             });
 
             const mockCustomAttributeConditionEvaluator = vi.fn().mockReturnValue(false);
 
-            vi.spyOn(customAttributeConditionEvaluator, 'getEvaluator').mockReturnValue({
+            getEvaluatorSpy.mockReturnValue({
               evaluate: mockCustomAttributeConditionEvaluator,
             });          
 
@@ -278,11 +284,15 @@ 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();
-            vi.spyOn(conditionTreeEvaluator, 'evaluate');
-            vi.spyOn(customAttributeConditionEvaluator, 'getEvaluator').mockReturnValue({
+            getEvaluatorSpy.mockReturnValue({
               evaluate: mockCustomAttributeConditionEvaluator,
             });
           });
@@ -292,7 +302,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('logs correctly when conditionTreeEvaluator.evaluate returns null', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockImplementationOnce((conditions: any, leafEvaluator) => {
+            evaluateSpy.mockImplementationOnce((conditions: any, leafEvaluator) => {
               return leafEvaluator(conditions[1]);
             });
 
@@ -319,7 +329,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('logs correctly when conditionTreeEvaluator.evaluate returns true', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockImplementationOnce((conditions: any, leafEvaluator) => {
+            evaluateSpy.mockImplementationOnce((conditions: any, leafEvaluator) => {
               return leafEvaluator(conditions[1]);
             });
 
@@ -345,7 +355,7 @@ describe('lib/core/audience_evaluator', () => {
           });
 
           it('logs correctly when conditionTreeEvaluator.evaluate returns false', () => {
-            vi.spyOn(conditionTreeEvaluator, 'evaluate').mockImplementationOnce((conditions: any, leafEvaluator) => {
+            evaluateSpy.mockImplementationOnce((conditions: any, leafEvaluator) => {
               return leafEvaluator(conditions[1]);
             });
 

vitest.browser.config.mts

@@ -113,7 +113,7 @@ export default defineConfig({
     // pool: 'forks', // Use forks pool to avoid threading issues with BrowserStack
     // bail: 1, // Stop on first failure to avoid cascading errors
     // Include all .spec.ts files in lib directory, but exclude react_native tests
-    include: ['lib/**/batch_event_processor.spec.ts'],
+    include: ['lib/**/audience_evaluator/index.spec.ts'],
     exclude: [
       'lib/**/*.react_native.spec.ts',
       'lib/**/*.node.spec.ts',