Fix flaky test for dataSize RecordsQuery (#573)

TBD54566975 · Oct 24, 2023 · 62d5656 · 62d5656
1 parent dcf65d4
commit 62d5656
Showing 1 changed file with 27 additions and 10 deletions.
diff --git a/tests/handlers/records-query.spec.ts b/tests/handlers/records-query.spec.ts
@@ -223,8 +223,13 @@ export function testRecordsQueryHandler(): void {
         });
         const reply1 = await dwn.processMessage(alice.did, recordsQuery1.message);
         expect(reply1.entries?.length).to.equal(2);
-        expect(reply1.entries![0].encodedData).to.equal(Encoder.bytesToBase64Url(write2.dataBytes!));
-        expect(reply1.entries![1].encodedData).to.equal(Encoder.bytesToBase64Url(write3.dataBytes!));
+
+        expect(
+          reply1.entries?.map((entry) => entry.encodedData)
+        ).to.have.members([
+          Encoder.bytesToBase64Url(write2.dataBytes!),
+          Encoder.bytesToBase64Url(write3.dataBytes!)
+        ]);
 
         // testing lt
         const recordsQuery2 = await TestDataGenerator.generateRecordsQuery({
@@ -233,8 +238,12 @@ export function testRecordsQueryHandler(): void {
         });
         const reply2 = await dwn.processMessage(alice.did, recordsQuery2.message);
         expect(reply2.entries?.length).to.equal(2);
-        expect(reply2.entries![0].encodedData).to.equal(Encoder.bytesToBase64Url(write1.dataBytes!));
-        expect(reply2.entries![1].encodedData).to.equal(Encoder.bytesToBase64Url(write2.dataBytes!));
+        expect(
+          reply2.entries?.map((entry) => entry.encodedData)
+        ).to.have.members([
+          Encoder.bytesToBase64Url(write1.dataBytes!),
+          Encoder.bytesToBase64Url(write2.dataBytes!)
+        ]);
 
         // testing gte
         const recordsQuery3 = await TestDataGenerator.generateRecordsQuery({
@@ -243,9 +252,13 @@ export function testRecordsQueryHandler(): void {
         });
         const reply3 = await dwn.processMessage(alice.did, recordsQuery3.message);
         expect(reply3.entries?.length).to.equal(3);
-        expect(reply3.entries![0].encodedData).to.equal(Encoder.bytesToBase64Url(write1.dataBytes!));
-        expect(reply3.entries![1].encodedData).to.equal(Encoder.bytesToBase64Url(write2.dataBytes!));
-        expect(reply3.entries![2].encodedData).to.equal(Encoder.bytesToBase64Url(write3.dataBytes!));
+        expect(
+          reply3.entries?.map((entry) => entry.encodedData)
+        ).to.have.members([
+          Encoder.bytesToBase64Url(write1.dataBytes!),
+          Encoder.bytesToBase64Url(write2.dataBytes!),
+          Encoder.bytesToBase64Url(write3.dataBytes!)
+        ]);
 
         // testing lte
         const recordsQuery4 = await TestDataGenerator.generateRecordsQuery({
@@ -254,9 +267,13 @@ export function testRecordsQueryHandler(): void {
         });
         const reply4 = await dwn.processMessage(alice.did, recordsQuery4.message);
         expect(reply4.entries?.length).to.equal(3);
-        expect(reply4.entries![0].encodedData).to.equal(Encoder.bytesToBase64Url(write1.dataBytes!));
-        expect(reply4.entries![1].encodedData).to.equal(Encoder.bytesToBase64Url(write2.dataBytes!));
-        expect(reply4.entries![2].encodedData).to.equal(Encoder.bytesToBase64Url(write3.dataBytes!));
+        expect(
+          reply4.entries?.map((entry) => entry.encodedData)
+        ).to.have.members([
+          Encoder.bytesToBase64Url(write1.dataBytes!),
+          Encoder.bytesToBase64Url(write2.dataBytes!),
+          Encoder.bytesToBase64Url(write3.dataBytes!)
+        ]);
       });
 
       it('should be able to range query with `dataSize` filter (open & closed range)', async () => {