nirbarazida
/
promptfoo
mirror of https://github.com/promptfoo/promptfoo


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
46

	
47

	
48

	
49

	
50

	
51

	
52

	
53

	
54

	
            import {
  setLogCallback,
  getLogLevel,
  setLogLevel,
  LOG_LEVELS,
  globalLogCallback,
} from '../src/logger';

describe('logger', () => {
  beforeEach(() => {
    setLogCallback(null);
  });

  describe('setLogCallback', () => {
    it('should set the global log callback', () => {
      const callback = jest.fn();
      setLogCallback(callback);
      expect(globalLogCallback).toBe(callback);
    });

    it('should allow setting null callback', () => {
      setLogCallback(null);
      expect(globalLogCallback).toBeNull();
    });
  });

  describe('getLogLevel', () => {
    it('should return current log level', () => {
      const level = getLogLevel();
      expect(Object.keys(LOG_LEVELS)).toContain(level);
    });
  });

  describe('setLogLevel', () => {
    it('should set valid log levels', () => {
      setLogLevel('debug');
      expect(getLogLevel()).toBe('debug');

      setLogLevel('info');
      expect(getLogLevel()).toBe('info');

      setLogLevel('warn');
      expect(getLogLevel()).toBe('warn');

      setLogLevel('error');
      expect(getLogLevel()).toBe('error');
    });

    it('should throw error for invalid log level', () => {
      // @ts-expect-error Testing invalid input
      expect(() => setLogLevel('invalid')).toThrow('Invalid log level: invalid');
    });
  });
});