From 600adb3dd7db2656858e46b00d3844e2aee2d6d8 Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Tue, 28 Mar 2023 02:07:26 -0500
Subject: [PATCH] test: improve coverage on `@affine/debug` (#1723)

---
 packages/debug/src/__tests__/index.spec.ts | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 packages/debug/src/__tests__/index.spec.ts

diff --git a/packages/debug/src/__tests__/index.spec.ts b/packages/debug/src/__tests__/index.spec.ts
new file mode 100644
index 0000000000..0c26fe3a60
--- /dev/null
+++ b/packages/debug/src/__tests__/index.spec.ts
@@ -0,0 +1,33 @@
+/**
+ * @vitest-environment happy-dom
+ */
+import { DebugLogger } from '@affine/debug';
+import { describe, expect, test, vi } from 'vitest';
+
+describe('debug', () => {
+  test('disabled', () => {
+    const logger = new DebugLogger('test');
+    logger.enabled = false;
+    for (const level of ['debug', 'info', 'warn', 'error'] as const) {
+      const fn = vi.fn();
+      vi.spyOn(globalThis.console, level).mockImplementation(fn);
+      expect(logger.enabled).toBe(false);
+      expect(fn).not.toBeCalled();
+      logger[level]('test');
+      expect(fn, level).not.toBeCalled();
+    }
+  });
+
+  test('log', () => {
+    const logger = new DebugLogger('test');
+    logger.enabled = true;
+    for (const level of ['debug', 'info', 'warn', 'error'] as const) {
+      const fn = vi.fn();
+      vi.spyOn(globalThis.console, level).mockImplementation(fn);
+      expect(logger.enabled).toBe(true);
+      expect(fn).not.toBeCalled();
+      logger[level]('test');
+      expect(fn, level).toBeCalled();
+    }
+  });
+});