|
1 | 1 | import { describe, it, expect } from 'vitest'; |
2 | | -import { calculateStreak } from './calculate'; |
| 2 | +import { calculateStreak, calculateMonthlyStats } from './calculate'; |
3 | 3 | import type { ContributionCalendar } from '../types'; |
4 | 4 |
|
5 | 5 | // Turns a flat array of daily counts into the ContributionCalendar shape, |
@@ -259,6 +259,105 @@ describe('calculateStreak — timezone awareness', () => { |
259 | 259 | }); |
260 | 260 | }); |
261 | 261 |
|
| 262 | +describe('calculateMonthlyStats', () => { |
| 263 | + it('calculates monthly stats correctly when both months have commits', () => { |
| 264 | + const calendar = { |
| 265 | + totalContributions: 15, |
| 266 | + weeks: [ |
| 267 | + { |
| 268 | + contributionDays: [ |
| 269 | + { contributionCount: 5, date: '2024-05-15' }, |
| 270 | + { contributionCount: 10, date: '2024-06-10' }, |
| 271 | + ], |
| 272 | + }, |
| 273 | + ], |
| 274 | + }; |
| 275 | + const now = new Date('2024-06-15T12:00:00Z'); |
| 276 | + const result = calculateMonthlyStats(calendar, 'UTC', now); |
| 277 | + |
| 278 | + expect(result.currentMonthTotal).toBe(10); |
| 279 | + expect(result.previousMonthTotal).toBe(5); |
| 280 | + expect(result.deltaAbsolute).toBe(5); |
| 281 | + expect(result.deltaPercentage).toBe(100); |
| 282 | + expect(result.currentMonthName).toBe('June'); |
| 283 | + }); |
| 284 | + |
| 285 | + it('handles zero previous month contributions', () => { |
| 286 | + const calendar = { |
| 287 | + totalContributions: 10, |
| 288 | + weeks: [ |
| 289 | + { |
| 290 | + contributionDays: [{ contributionCount: 10, date: '2024-06-10' }], |
| 291 | + }, |
| 292 | + ], |
| 293 | + }; |
| 294 | + const now = new Date('2024-06-15T12:00:00Z'); |
| 295 | + const result = calculateMonthlyStats(calendar, 'UTC', now); |
| 296 | + |
| 297 | + expect(result.previousMonthTotal).toBe(0); |
| 298 | + expect(result.currentMonthTotal).toBe(10); |
| 299 | + expect(result.deltaPercentage).toBe(100); |
| 300 | + }); |
| 301 | + |
| 302 | + it('handles zero current month contributions', () => { |
| 303 | + const calendar = { |
| 304 | + totalContributions: 5, |
| 305 | + weeks: [ |
| 306 | + { |
| 307 | + contributionDays: [{ contributionCount: 5, date: '2024-05-10' }], |
| 308 | + }, |
| 309 | + ], |
| 310 | + }; |
| 311 | + const now = new Date('2024-06-15T12:00:00Z'); |
| 312 | + const result = calculateMonthlyStats(calendar, 'UTC', now); |
| 313 | + |
| 314 | + expect(result.previousMonthTotal).toBe(5); |
| 315 | + expect(result.currentMonthTotal).toBe(0); |
| 316 | + expect(result.deltaPercentage).toBe(-100); |
| 317 | + }); |
| 318 | + |
| 319 | + it('handles negative delta correctly', () => { |
| 320 | + const calendar = { |
| 321 | + totalContributions: 15, |
| 322 | + weeks: [ |
| 323 | + { |
| 324 | + contributionDays: [ |
| 325 | + { contributionCount: 10, date: '2024-05-10' }, |
| 326 | + { contributionCount: 5, date: '2024-06-10' }, |
| 327 | + ], |
| 328 | + }, |
| 329 | + ], |
| 330 | + }; |
| 331 | + const now = new Date('2024-06-15T12:00:00Z'); |
| 332 | + const result = calculateMonthlyStats(calendar, 'UTC', now); |
| 333 | + |
| 334 | + expect(result.previousMonthTotal).toBe(10); |
| 335 | + expect(result.currentMonthTotal).toBe(5); |
| 336 | + expect(result.deltaPercentage).toBe(-50); |
| 337 | + expect(result.deltaAbsolute).toBe(-5); |
| 338 | + }); |
| 339 | + |
| 340 | + it('handles year boundary correctly (Jan vs Dec)', () => { |
| 341 | + const calendar = { |
| 342 | + totalContributions: 15, |
| 343 | + weeks: [ |
| 344 | + { |
| 345 | + contributionDays: [ |
| 346 | + { contributionCount: 10, date: '2023-12-15' }, |
| 347 | + { contributionCount: 5, date: '2024-01-15' }, |
| 348 | + ], |
| 349 | + }, |
| 350 | + ], |
| 351 | + }; |
| 352 | + const now = new Date('2024-01-15T12:00:00Z'); |
| 353 | + const result = calculateMonthlyStats(calendar, 'UTC', now); |
| 354 | + |
| 355 | + expect(result.previousMonthTotal).toBe(10); |
| 356 | + expect(result.currentMonthTotal).toBe(5); |
| 357 | + expect(result.currentMonthName).toBe('January'); |
| 358 | + }); |
| 359 | +}); |
| 360 | + |
262 | 361 | describe('calculateStreak — empty and sparse year edge cases', () => { |
263 | 362 | it('returns stable output when all weeks have zero-contribution days', () => { |
264 | 363 | const calendar = buildCalendar([ |
|
0 commit comments