test: Make sure that fake date stays in the same month

TriPSs · Dec 1, 2023 · 3ceb52d · 3ceb52d
1 parent b103072
commit 3ceb52d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/examples/typeorm/e2e/fixtures.ts b/examples/typeorm/e2e/fixtures.ts
@@ -15,10 +15,14 @@ export const refresh = async (connection: Connection): Promise<void> => {
   const subTaskRepo = connection.getRepository(SubTaskEntity)
   const tagsRepo = connection.getRepository(TagEntity)
 
-  const yesterday = new Date()
-  yesterday.setDate(yesterday.getDate() - 1)
-
-  const urgentTag = await tagsRepo.save({ name: 'Urgent', fakeDate: yesterday })
+  const yesterdayOrTomorrow = new Date()
+  if (yesterdayOrTomorrow.getDate() === 1) {
+    yesterdayOrTomorrow.setDate(yesterdayOrTomorrow.getDate() + 1)
+  } else {
+    yesterdayOrTomorrow.setDate(yesterdayOrTomorrow.getDate() - 1)
+  }
+
+  const urgentTag = await tagsRepo.save({ name: 'Urgent', fakeDate: yesterdayOrTomorrow })
   const homeTag = await tagsRepo.save({ name: 'Home' })
   const workTag = await tagsRepo.save({ name: 'Work' })
   const questionTag = await tagsRepo.save({ name: 'Question' })

diff --git a/examples/typeorm/e2e/tag.resolver.spec.ts b/examples/typeorm/e2e/tag.resolver.spec.ts
@@ -443,8 +443,12 @@ describe('TagResolver (typeorm - e2e)', () => {
           .then(({ body }) => {
             const res: AggregateResponse<TodoItemDTO>[] = body.data.tagAggregate
             expect(res).toHaveLength(2)
-            expect(res[0].sum).toEqual({ id: 1 })
-            expect(res[1].sum).toEqual({ id: 14 })
+
+            // First of the month we switched the days so they stay in the same month
+            const isFirstOfMonth = new Date().getDate() === 1
+
+            expect(res[isFirstOfMonth ? 1 : 0].sum).toEqual({ id: 1 })
+            expect(res[isFirstOfMonth ? 0 : 1].sum).toEqual({ id: 14 })
           }))
     })
   })