@dizzlkheinz/ynab-mcpb 0.18.4 → 0.19.0

package/src/tools/__tests__/monthTools.test.ts

@@ -1,523 +1,572 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import * as ynab from 'ynab';
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import type * as ynab from "ynab";
 import {
-  handleGetMonth,
-  handleListMonths,
-  GetMonthSchema,
-  ListMonthsSchema,
-} from '../monthTools.js';
-import { createDeltaFetcherMock, createRejectingDeltaFetcherMock } from './deltaTestUtils.js';
+	GetMonthSchema,
+	ListMonthsSchema,
+	handleGetMonth,
+	handleListMonths,
+} from "../monthTools.js";
+import {
+	createDeltaFetcherMock,
+	createRejectingDeltaFetcherMock,
+} from "./deltaTestUtils.js";
 
 // Mock the cache manager
-vi.mock('../../server/cacheManager.js', () => ({
-  cacheManager: {
-    wrap: vi.fn(),
-    has: vi.fn(),
-    delete: vi.fn(),
-    deleteMany: vi.fn(),
-    deleteByPrefix: vi.fn(),
-    deleteByBudgetId: vi.fn(),
-    clear: vi.fn(),
-  },
-  CacheManager: {
-    generateKey: vi.fn(),
-  },
-  CACHE_TTLS: {
-    MONTHS: 600000,
-  },
+vi.mock("../../server/cacheManager.js", () => ({
+	cacheManager: {
+		wrap: vi.fn(),
+		has: vi.fn(),
+		delete: vi.fn(),
+		deleteMany: vi.fn(),
+		deleteByPrefix: vi.fn(),
+		deleteByBudgetId: vi.fn(),
+		clear: vi.fn(),
+	},
+	CacheManager: {
+		generateKey: vi.fn(),
+	},
+	CACHE_TTLS: {
+		MONTHS: 600000,
+	},
 }));
 
 // Mock the YNAB API
 const mockYnabAPI = {
-  months: {
-    getBudgetMonth: vi.fn(),
-    getBudgetMonths: vi.fn(),
-  },
+	months: {
+		getBudgetMonth: vi.fn(),
+		getBudgetMonths: vi.fn(),
+	},
 } as unknown as ynab.API;
 
 // Import mocked cache manager
-const { cacheManager, CacheManager, CACHE_TTLS } = await import('../../server/cacheManager.js');
-
-describe('Month Tools', () => {
-  beforeEach(() => {
-    vi.resetAllMocks();
-    // Reset NODE_ENV to test to ensure cache bypassing in tests
-    process.env['NODE_ENV'] = 'test';
-    // Mock cache.wrap to call the loader directly (bypass cache)
-    (cacheManager.wrap as ReturnType<typeof vi.fn>).mockImplementation(
-      async (
-        _key: string,
-        options: {
-          loader: () => Promise<unknown>;
-        },
-      ) => {
-        return await options.loader();
-      },
-    );
-    (cacheManager.has as ReturnType<typeof vi.fn>).mockReturnValue(false);
-    (CacheManager.generateKey as ReturnType<typeof vi.fn>).mockImplementation(
-      (prefix: string, ...parts: (string | number | boolean | undefined)[]) =>
-        [prefix, ...parts.filter((part) => part !== undefined)].join(':'),
-    );
-  });
-
-  describe('handleGetMonth', () => {
-    it('should bypass cache in test environment', async () => {
-      const mockMonth = {
-        month: '2024-01-01',
-        note: 'January budget',
-        income: 500000,
-        budgeted: 450000,
-        activity: -400000,
-        to_be_budgeted: 50000,
-        age_of_money: 30,
-        deleted: false,
-        categories: [],
-      };
-
-      (mockYnabAPI.months.getBudgetMonth as any).mockResolvedValue({
-        data: { month: mockMonth },
-      });
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      // Verify the mock cache was called (and bypassed to call the loader)
-      expect(cacheManager.wrap).toHaveBeenCalled();
-      expect(mockYnabAPI.months.getBudgetMonth).toHaveBeenCalledTimes(1);
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.cached).toBe(false);
-      expect(parsedContent.cache_info).toBe('Fresh data retrieved from YNAB API');
-      expect(parsedContent.month.month).toBe('2024-01-01');
-    });
-
-    it.skip('should use cache when NODE_ENV is not test - obsolete test, caching now handled by DeltaFetcher', async () => {
-      // Temporarily set NODE_ENV to non-test
-      process.env['NODE_ENV'] = 'development';
-
-      const mockMonth = {
-        month: '2024-01-01',
-        note: 'January budget',
-        income: 500000,
-        budgeted: 450000,
-        activity: -400000,
-        to_be_budgeted: 50000,
-        age_of_money: 30,
-        deleted: false,
-        categories: [],
-      };
-
-      const mockCacheKey = 'month:get:budget-1:2024-01-01:generated-key';
-      (CacheManager.generateKey as any).mockReturnValue(mockCacheKey);
-      (cacheManager.wrap as any).mockResolvedValue(mockMonth);
-      (cacheManager.has as any).mockReturnValue(true);
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.cached).toBe(true);
-      expect(parsedContent.cache_info).toBe('Data retrieved from cache for improved performance');
-
-      // Verify cache was used
-      expect(CacheManager.generateKey).toHaveBeenCalledWith(
-        'month',
-        'get',
-        'budget-1',
-        '2024-01-01',
-      );
-      expect(cacheManager.wrap).toHaveBeenCalledWith(mockCacheKey, {
-        ttl: CACHE_TTLS.MONTHS,
-        loader: expect.any(Function),
-      });
-      expect(cacheManager.has).toHaveBeenCalledWith(mockCacheKey);
-
-      // Reset NODE_ENV
-      process.env['NODE_ENV'] = 'test';
-    });
-
-    it('should return formatted month data on success', async () => {
-      const mockMonth = {
-        month: '2024-01-01',
-        note: 'January budget',
-        income: 500000,
-        budgeted: 450000,
-        activity: -400000,
-        to_be_budgeted: 50000,
-        age_of_money: 30,
-        deleted: false,
-        categories: [
-          {
-            id: 'category-1',
-            category_group_id: 'group-1',
-            category_group_name: 'Monthly Bills',
-            name: 'Rent',
-            hidden: false,
-            original_category_group_id: null,
-            note: 'Monthly rent payment',
-            budgeted: 150000,
-            activity: -150000,
-            balance: 0,
-            goal_type: 'TB',
-            goal_creation_month: '2024-01-01',
-            goal_target: 150000,
-            goal_target_month: '2024-01-01',
-            goal_percentage_complete: 100,
-            goal_months_to_budget: 0,
-            goal_under_funded: 0,
-            goal_overall_funded: 150000,
-            goal_overall_left: 0,
-            deleted: false,
-          },
-        ],
-      };
-
-      (mockYnabAPI.months.getBudgetMonth as any).mockResolvedValue({
-        data: { month: mockMonth },
-      });
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      expect(result.content[0].type).toBe('text');
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.month.month).toBe('2024-01-01');
-      expect(parsedContent.month.note).toBe('January budget');
-      expect(parsedContent.month.income).toBe(500);
-      expect(parsedContent.month.budgeted).toBe(450);
-      expect(parsedContent.month.activity).toBe(-400);
-      expect(parsedContent.month.to_be_budgeted).toBe(50);
-      expect(parsedContent.month.age_of_money).toBe(30);
-      expect(parsedContent.month.categories).toHaveLength(1);
-      expect(parsedContent.month.categories[0].name).toBe('Rent');
-    });
-
-    it('should handle 401 authentication errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(new Error('401 Unauthorized'));
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Invalid or expired YNAB access token');
-    });
-
-    it('should handle 403 forbidden errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(new Error('403 Forbidden'));
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Insufficient permissions to access YNAB data');
-    });
-
-    it('should handle 404 not found errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(new Error('404 Not Found'));
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'invalid-budget',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Budget or month not found');
-    });
-
-    it('should handle 429 rate limit errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
-        new Error('429 Too Many Requests'),
-      );
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Rate limit exceeded. Please try again later');
-    });
-
-    it('should handle 500 server errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
-        new Error('500 Internal Server Error'),
-      );
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('YNAB service is currently unavailable');
-    });
-
-    it('should handle generic errors', async () => {
-      (mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(new Error('Network error'));
-
-      const result = await handleGetMonth(mockYnabAPI, {
-        budget_id: 'budget-1',
-        month: '2024-01-01',
-      });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Failed to get month data');
-    });
-  });
-
-  describe('handleListMonths', () => {
-    it('should include cache metadata from delta fetcher results', async () => {
-      const mockMonths = [
-        {
-          month: '2024-01-01',
-          note: 'January budget',
-          income: 500000,
-          budgeted: 450000,
-          activity: -400000,
-          to_be_budgeted: 50000,
-          age_of_money: 30,
-          deleted: false,
-        },
-      ];
-      const { fetcher, resolved } = createDeltaFetcherMock('fetchMonths', {
-        data: mockMonths,
-        wasCached: true,
-        usedDelta: true,
-      });
-
-      const result = await handleListMonths(mockYnabAPI, fetcher, { budget_id: 'budget-1' });
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.cached).toBe(resolved.wasCached);
-      expect(parsedContent.cache_info).toContain('delta merge applied');
-      expect(parsedContent.months).toHaveLength(1);
-    });
-
-    it.skip('should use cache when NODE_ENV is not test - obsolete test, caching now handled by DeltaFetcher', async () => {
-      // Temporarily set NODE_ENV to non-test
-      process.env['NODE_ENV'] = 'development';
-
-      const mockMonths = [
-        {
-          month: '2024-01-01',
-          note: 'January budget',
-          income: 500000,
-          budgeted: 450000,
-          activity: -400000,
-          to_be_budgeted: 50000,
-          age_of_money: 30,
-          deleted: false,
-        },
-      ];
-
-      const mockCacheKey = 'months:list:budget-1:generated-key';
-      (CacheManager.generateKey as any).mockReturnValue(mockCacheKey);
-      (cacheManager.wrap as any).mockResolvedValue(mockMonths);
-      (cacheManager.has as any).mockReturnValue(true);
-
-      const result = await handleListMonths(mockYnabAPI, { budget_id: 'budget-1' });
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.cached).toBe(true);
-      expect(parsedContent.cache_info).toBe('Data retrieved from cache for improved performance');
-
-      // Verify cache was used
-      expect(CacheManager.generateKey).toHaveBeenCalledWith('months', 'list', 'budget-1');
-      expect(cacheManager.wrap).toHaveBeenCalledWith(mockCacheKey, {
-        ttl: CACHE_TTLS.MONTHS,
-        loader: expect.any(Function),
-      });
-      expect(cacheManager.has).toHaveBeenCalledWith(mockCacheKey);
-
-      // Reset NODE_ENV
-      process.env['NODE_ENV'] = 'test';
-    });
-
-    it('should return formatted months list on success', async () => {
-      const mockMonths = [
-        {
-          month: '2024-01-01',
-          note: 'January budget',
-          income: 500000,
-          budgeted: 450000,
-          activity: -400000,
-          to_be_budgeted: 50000,
-          age_of_money: 30,
-          deleted: false,
-        },
-        {
-          month: '2024-02-01',
-          note: 'February budget',
-          income: 520000,
-          budgeted: 470000,
-          activity: -420000,
-          to_be_budgeted: 50000,
-          age_of_money: 32,
-          deleted: false,
-        },
-      ];
-      const { fetcher, resolved } = createDeltaFetcherMock('fetchMonths', {
-        data: mockMonths,
-        wasCached: false,
-      });
-
-      const result = await handleListMonths(mockYnabAPI, fetcher, { budget_id: 'budget-1' });
-
-      expect(result.content).toHaveLength(1);
-      expect(result.content[0].type).toBe('text');
-
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.cached).toBe(resolved.wasCached);
-      expect(parsedContent.cache_info).toBe('Fresh data retrieved from YNAB API');
-      expect(parsedContent.months).toHaveLength(2);
-      expect(parsedContent.months[0]).toEqual({
-        month: '2024-01-01',
-        note: 'January budget',
-        income: 500,
-        budgeted: 450,
-        activity: -400,
-        to_be_budgeted: 50,
-        age_of_money: 30,
-        deleted: false,
-      });
-      expect(parsedContent.months[1]).toEqual({
-        month: '2024-02-01',
-        note: 'February budget',
-        income: 520,
-        budgeted: 470,
-        activity: -420,
-        to_be_budgeted: 50,
-        age_of_money: 32,
-        deleted: false,
-      });
-    });
-
-    it('should handle authentication errors', async () => {
-      const { fetcher } = createRejectingDeltaFetcherMock(
-        'fetchMonths',
-        new Error('401 Unauthorized'),
-      );
-
-      const result = await handleListMonths(mockYnabAPI, fetcher, { budget_id: 'budget-1' });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Invalid or expired YNAB access token');
-    });
-
-    it('should handle not found errors', async () => {
-      const { fetcher } = createRejectingDeltaFetcherMock(
-        'fetchMonths',
-        new Error('404 Not Found'),
-      );
-
-      const result = await handleListMonths(mockYnabAPI, fetcher, { budget_id: 'invalid-budget' });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Budget or month not found');
-    });
-
-    it('should handle generic errors', async () => {
-      (mockYnabAPI.months.getBudgetMonths as any).mockRejectedValue(new Error('Network error'));
-
-      const result = await handleListMonths(mockYnabAPI, { budget_id: 'budget-1' });
-
-      expect(result.content).toHaveLength(1);
-      const parsedContent = JSON.parse(result.content[0].text);
-      expect(parsedContent.error.message).toBe('Failed to list months');
-    });
-  });
-
-  describe('GetMonthSchema', () => {
-    it('should validate valid parameters', () => {
-      const result = GetMonthSchema.parse({
-        budget_id: 'valid-budget-id',
-        month: '2024-01-01',
-      });
-      expect(result.budget_id).toBe('valid-budget-id');
-      expect(result.month).toBe('2024-01-01');
-    });
-
-    it('should reject empty budget_id', () => {
-      expect(() =>
-        GetMonthSchema.parse({
-          budget_id: '',
-          month: '2024-01-01',
-        }),
-      ).toThrow();
-    });
-
-    it('should reject missing budget_id', () => {
-      expect(() =>
-        GetMonthSchema.parse({
-          month: '2024-01-01',
-        }),
-      ).toThrow();
-    });
-
-    it('should reject invalid month format', () => {
-      expect(() =>
-        GetMonthSchema.parse({
-          budget_id: 'valid-budget-id',
-          month: '2024-1-1',
-        }),
-      ).toThrow();
-    });
-
-    it('should reject missing month', () => {
-      expect(() =>
-        GetMonthSchema.parse({
-          budget_id: 'valid-budget-id',
-        }),
-      ).toThrow();
-    });
-
-    it('should reject non-ISO date format', () => {
-      expect(() =>
-        GetMonthSchema.parse({
-          budget_id: 'valid-budget-id',
-          month: '01/01/2024',
-        }),
-      ).toThrow();
-    });
-  });
-
-  describe('ListMonthsSchema', () => {
-    it('should validate valid budget_id', () => {
-      const result = ListMonthsSchema.parse({ budget_id: 'valid-budget-id' });
-      expect(result.budget_id).toBe('valid-budget-id');
-    });
-
-    it('should reject empty budget_id', () => {
-      expect(() => ListMonthsSchema.parse({ budget_id: '' })).toThrow();
-    });
-
-    it('should reject missing budget_id', () => {
-      expect(() => ListMonthsSchema.parse({})).toThrow();
-    });
-
-    it('should reject non-string budget_id', () => {
-      expect(() => ListMonthsSchema.parse({ budget_id: 123 })).toThrow();
-    });
-  });
+const { cacheManager, CacheManager, CACHE_TTLS } = await import(
+	"../../server/cacheManager.js"
+);
+
+describe("Month Tools", () => {
+	beforeEach(() => {
+		vi.resetAllMocks();
+		// Reset NODE_ENV to test to ensure cache bypassing in tests
+		process.env.NODE_ENV = "test";
+		// Mock cache.wrap to call the loader directly (bypass cache)
+		(cacheManager.wrap as ReturnType<typeof vi.fn>).mockImplementation(
+			async (
+				_key: string,
+				options: {
+					loader: () => Promise<unknown>;
+				},
+			) => {
+				return await options.loader();
+			},
+		);
+		(cacheManager.has as ReturnType<typeof vi.fn>).mockReturnValue(false);
+		(CacheManager.generateKey as ReturnType<typeof vi.fn>).mockImplementation(
+			(prefix: string, ...parts: (string | number | boolean | undefined)[]) =>
+				[prefix, ...parts.filter((part) => part !== undefined)].join(":"),
+		);
+	});
+
+	describe("handleGetMonth", () => {
+		it("should bypass cache in test environment", async () => {
+			const mockMonth = {
+				month: "2024-01-01",
+				note: "January budget",
+				income: 500000,
+				budgeted: 450000,
+				activity: -400000,
+				to_be_budgeted: 50000,
+				age_of_money: 30,
+				deleted: false,
+				categories: [],
+			};
+
+			(mockYnabAPI.months.getBudgetMonth as any).mockResolvedValue({
+				data: { month: mockMonth },
+			});
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			// Verify the mock cache was called (and bypassed to call the loader)
+			expect(cacheManager.wrap).toHaveBeenCalled();
+			expect(mockYnabAPI.months.getBudgetMonth).toHaveBeenCalledTimes(1);
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.cached).toBe(false);
+			expect(parsedContent.cache_info).toBe(
+				"Fresh data retrieved from YNAB API",
+			);
+			expect(parsedContent.month.month).toBe("2024-01-01");
+		});
+
+		it.skip("should use cache when NODE_ENV is not test - obsolete test, caching now handled by DeltaFetcher", async () => {
+			// Temporarily set NODE_ENV to non-test
+			process.env.NODE_ENV = "development";
+
+			const mockMonth = {
+				month: "2024-01-01",
+				note: "January budget",
+				income: 500000,
+				budgeted: 450000,
+				activity: -400000,
+				to_be_budgeted: 50000,
+				age_of_money: 30,
+				deleted: false,
+				categories: [],
+			};
+
+			const mockCacheKey = "month:get:budget-1:2024-01-01:generated-key";
+			(CacheManager.generateKey as any).mockReturnValue(mockCacheKey);
+			(cacheManager.wrap as any).mockResolvedValue(mockMonth);
+			(cacheManager.has as any).mockReturnValue(true);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.cached).toBe(true);
+			expect(parsedContent.cache_info).toBe(
+				"Data retrieved from cache for improved performance",
+			);
+
+			// Verify cache was used
+			expect(CacheManager.generateKey).toHaveBeenCalledWith(
+				"month",
+				"get",
+				"budget-1",
+				"2024-01-01",
+			);
+			expect(cacheManager.wrap).toHaveBeenCalledWith(mockCacheKey, {
+				ttl: CACHE_TTLS.MONTHS,
+				loader: expect.any(Function),
+			});
+			expect(cacheManager.has).toHaveBeenCalledWith(mockCacheKey);
+
+			// Reset NODE_ENV
+			process.env.NODE_ENV = "test";
+		});
+
+		it("should return formatted month data on success", async () => {
+			const mockMonth = {
+				month: "2024-01-01",
+				note: "January budget",
+				income: 500000,
+				budgeted: 450000,
+				activity: -400000,
+				to_be_budgeted: 50000,
+				age_of_money: 30,
+				deleted: false,
+				categories: [
+					{
+						id: "category-1",
+						category_group_id: "group-1",
+						category_group_name: "Monthly Bills",
+						name: "Rent",
+						hidden: false,
+						original_category_group_id: null,
+						note: "Monthly rent payment",
+						budgeted: 150000,
+						activity: -150000,
+						balance: 0,
+						goal_type: "TB",
+						goal_creation_month: "2024-01-01",
+						goal_target: 150000,
+						goal_target_month: "2024-01-01",
+						goal_percentage_complete: 100,
+						goal_months_to_budget: 0,
+						goal_under_funded: 0,
+						goal_overall_funded: 150000,
+						goal_overall_left: 0,
+						deleted: false,
+					},
+				],
+			};
+
+			(mockYnabAPI.months.getBudgetMonth as any).mockResolvedValue({
+				data: { month: mockMonth },
+			});
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			expect(result.content[0].type).toBe("text");
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.month.month).toBe("2024-01-01");
+			expect(parsedContent.month.note).toBe("January budget");
+			expect(parsedContent.month.income).toBe(500);
+			expect(parsedContent.month.budgeted).toBe(450);
+			expect(parsedContent.month.activity).toBe(-400);
+			expect(parsedContent.month.to_be_budgeted).toBe(50);
+			expect(parsedContent.month.age_of_money).toBe(30);
+			expect(parsedContent.month.categories).toHaveLength(1);
+			expect(parsedContent.month.categories[0].name).toBe("Rent");
+		});
+
+		it("should handle 401 authentication errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("401 Unauthorized"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe(
+				"Invalid or expired YNAB access token",
+			);
+		});
+
+		it("should handle 403 forbidden errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("403 Forbidden"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe(
+				"Insufficient permissions to access YNAB data",
+			);
+		});
+
+		it("should handle 404 not found errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("404 Not Found"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "invalid-budget",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe("Budget or month not found");
+		});
+
+		it("should handle 429 rate limit errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("429 Too Many Requests"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe(
+				"Rate limit exceeded. Please try again later",
+			);
+		});
+
+		it("should handle 500 server errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("500 Internal Server Error"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe(
+				"YNAB service is currently unavailable",
+			);
+		});
+
+		it("should handle generic errors", async () => {
+			(mockYnabAPI.months.getBudgetMonth as any).mockRejectedValue(
+				new Error("Network error"),
+			);
+
+			const result = await handleGetMonth(mockYnabAPI, {
+				budget_id: "budget-1",
+				month: "2024-01-01",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe("Failed to get month data");
+		});
+	});
+
+	describe("handleListMonths", () => {
+		it("should include cache metadata from delta fetcher results", async () => {
+			const mockMonths = [
+				{
+					month: "2024-01-01",
+					note: "January budget",
+					income: 500000,
+					budgeted: 450000,
+					activity: -400000,
+					to_be_budgeted: 50000,
+					age_of_money: 30,
+					deleted: false,
+				},
+			];
+			const { fetcher, resolved } = createDeltaFetcherMock("fetchMonths", {
+				data: mockMonths,
+				wasCached: true,
+				usedDelta: true,
+			});
+
+			const result = await handleListMonths(mockYnabAPI, fetcher, {
+				budget_id: "budget-1",
+			});
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.cached).toBe(resolved.wasCached);
+			expect(parsedContent.cache_info).toContain("delta merge applied");
+			expect(parsedContent.months).toHaveLength(1);
+		});
+
+		it.skip("should use cache when NODE_ENV is not test - obsolete test, caching now handled by DeltaFetcher", async () => {
+			// Temporarily set NODE_ENV to non-test
+			process.env.NODE_ENV = "development";
+
+			const mockMonths = [
+				{
+					month: "2024-01-01",
+					note: "January budget",
+					income: 500000,
+					budgeted: 450000,
+					activity: -400000,
+					to_be_budgeted: 50000,
+					age_of_money: 30,
+					deleted: false,
+				},
+			];
+
+			const mockCacheKey = "months:list:budget-1:generated-key";
+			(CacheManager.generateKey as any).mockReturnValue(mockCacheKey);
+			(cacheManager.wrap as any).mockResolvedValue(mockMonths);
+			(cacheManager.has as any).mockReturnValue(true);
+
+			const result = await handleListMonths(mockYnabAPI, {
+				budget_id: "budget-1",
+			});
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.cached).toBe(true);
+			expect(parsedContent.cache_info).toBe(
+				"Data retrieved from cache for improved performance",
+			);
+
+			// Verify cache was used
+			expect(CacheManager.generateKey).toHaveBeenCalledWith(
+				"months",
+				"list",
+				"budget-1",
+			);
+			expect(cacheManager.wrap).toHaveBeenCalledWith(mockCacheKey, {
+				ttl: CACHE_TTLS.MONTHS,
+				loader: expect.any(Function),
+			});
+			expect(cacheManager.has).toHaveBeenCalledWith(mockCacheKey);
+
+			// Reset NODE_ENV
+			process.env.NODE_ENV = "test";
+		});
+
+		it("should return formatted months list on success", async () => {
+			const mockMonths = [
+				{
+					month: "2024-01-01",
+					note: "January budget",
+					income: 500000,
+					budgeted: 450000,
+					activity: -400000,
+					to_be_budgeted: 50000,
+					age_of_money: 30,
+					deleted: false,
+				},
+				{
+					month: "2024-02-01",
+					note: "February budget",
+					income: 520000,
+					budgeted: 470000,
+					activity: -420000,
+					to_be_budgeted: 50000,
+					age_of_money: 32,
+					deleted: false,
+				},
+			];
+			const { fetcher, resolved } = createDeltaFetcherMock("fetchMonths", {
+				data: mockMonths,
+				wasCached: false,
+			});
+
+			const result = await handleListMonths(mockYnabAPI, fetcher, {
+				budget_id: "budget-1",
+			});
+
+			expect(result.content).toHaveLength(1);
+			expect(result.content[0].type).toBe("text");
+
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.cached).toBe(resolved.wasCached);
+			expect(parsedContent.cache_info).toBe(
+				"Fresh data retrieved from YNAB API",
+			);
+			expect(parsedContent.months).toHaveLength(2);
+			expect(parsedContent.months[0]).toEqual({
+				month: "2024-01-01",
+				note: "January budget",
+				income: 500,
+				budgeted: 450,
+				activity: -400,
+				to_be_budgeted: 50,
+				age_of_money: 30,
+				deleted: false,
+			});
+			expect(parsedContent.months[1]).toEqual({
+				month: "2024-02-01",
+				note: "February budget",
+				income: 520,
+				budgeted: 470,
+				activity: -420,
+				to_be_budgeted: 50,
+				age_of_money: 32,
+				deleted: false,
+			});
+		});
+
+		it("should handle authentication errors", async () => {
+			const { fetcher } = createRejectingDeltaFetcherMock(
+				"fetchMonths",
+				new Error("401 Unauthorized"),
+			);
+
+			const result = await handleListMonths(mockYnabAPI, fetcher, {
+				budget_id: "budget-1",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe(
+				"Invalid or expired YNAB access token",
+			);
+		});
+
+		it("should handle not found errors", async () => {
+			const { fetcher } = createRejectingDeltaFetcherMock(
+				"fetchMonths",
+				new Error("404 Not Found"),
+			);
+
+			const result = await handleListMonths(mockYnabAPI, fetcher, {
+				budget_id: "invalid-budget",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe("Budget or month not found");
+		});
+
+		it("should handle generic errors", async () => {
+			(mockYnabAPI.months.getBudgetMonths as any).mockRejectedValue(
+				new Error("Network error"),
+			);
+
+			const result = await handleListMonths(mockYnabAPI, {
+				budget_id: "budget-1",
+			});
+
+			expect(result.content).toHaveLength(1);
+			const parsedContent = JSON.parse(result.content[0].text);
+			expect(parsedContent.error.message).toBe("Failed to list months");
+		});
+	});
+
+	describe("GetMonthSchema", () => {
+		it("should validate valid parameters", () => {
+			const result = GetMonthSchema.parse({
+				budget_id: "valid-budget-id",
+				month: "2024-01-01",
+			});
+			expect(result.budget_id).toBe("valid-budget-id");
+			expect(result.month).toBe("2024-01-01");
+		});
+
+		it("should reject empty budget_id", () => {
+			expect(() =>
+				GetMonthSchema.parse({
+					budget_id: "",
+					month: "2024-01-01",
+				}),
+			).toThrow();
+		});
+
+		it("should reject missing budget_id", () => {
+			expect(() =>
+				GetMonthSchema.parse({
+					month: "2024-01-01",
+				}),
+			).toThrow();
+		});
+
+		it("should reject invalid month format", () => {
+			expect(() =>
+				GetMonthSchema.parse({
+					budget_id: "valid-budget-id",
+					month: "2024-1-1",
+				}),
+			).toThrow();
+		});
+
+		it("should reject missing month", () => {
+			expect(() =>
+				GetMonthSchema.parse({
+					budget_id: "valid-budget-id",
+				}),
+			).toThrow();
+		});
+
+		it("should reject non-ISO date format", () => {
+			expect(() =>
+				GetMonthSchema.parse({
+					budget_id: "valid-budget-id",
+					month: "01/01/2024",
+				}),
+			).toThrow();
+		});
+	});
+
+	describe("ListMonthsSchema", () => {
+		it("should validate valid budget_id", () => {
+			const result = ListMonthsSchema.parse({ budget_id: "valid-budget-id" });
+			expect(result.budget_id).toBe("valid-budget-id");
+		});
+
+		it("should reject empty budget_id", () => {
+			expect(() => ListMonthsSchema.parse({ budget_id: "" })).toThrow();
+		});
+
+		it("should reject missing budget_id", () => {
+			expect(() => ListMonthsSchema.parse({})).toThrow();
+		});
+
+		it("should reject non-string budget_id", () => {
+			expect(() => ListMonthsSchema.parse({ budget_id: 123 })).toThrow();
+		});
+	});
 });