From 8d247b71c0131a47db33930497ddd81b4a06632c Mon Sep 17 00:00:00 2001
From: Carson Full <carson_full@tsco.org>
Date: Wed, 24 Jan 2024 11:11:35 -0600
Subject: [PATCH] Switch tests to use new file urls

---
 test/file.e2e-spec.ts     | 20 ++++++++------------
 test/utility/fragments.ts |  6 +++---
 2 files changed, 11 insertions(+), 15 deletions(-)

diff --git a/test/file.e2e-spec.ts b/test/file.e2e-spec.ts
index c125bd8cd1..bcd12d0ceb 100644
--- a/test/file.e2e-spec.ts
+++ b/test/file.e2e-spec.ts
@@ -1,5 +1,5 @@
 import { faker } from '@faker-js/faker';
-import { bufferFromStream } from '@seedcompany/common';
+import got from 'got';
 import { startCase, times } from 'lodash';
 import {
   DateTime,
@@ -104,14 +104,10 @@ function resetNow() {
   Settings.now = () => Date.now();
 }
 
-const expectEqualContent = async (
-  bucket: LocalBucket,
-  url: string,
-  expected: FakeFile,
-) => {
-  const actualFile = await bucket.download(url);
-  const contents = await bufferFromStream(actualFile.Body);
-  expect(contents.toString()).toEqual(expected.content.toString());
+const expectEqualContent = async (url: string, expected: FakeFile) => {
+  const expectedContents = expected.content.toString();
+  const actualContents = await got(url).buffer();
+  expect(expectedContents).toEqual(actualContents);
 };
 
 describe('File e2e', () => {
@@ -157,7 +153,7 @@ describe('File e2e', () => {
       expect(modifiedAt.diffNow().as('seconds')).toBeGreaterThan(-30);
       const createdAt = DateTime.fromISO(file.createdAt);
       expect(createdAt.diffNow().as('seconds')).toBeGreaterThan(-30);
-      await expectEqualContent(bucket, file.downloadUrl, fakeFile);
+      await expectEqualContent(file.url, fakeFile);
       expect(file.parents[0].id).toEqual(root.id);
     }
   });
@@ -178,7 +174,7 @@ describe('File e2e', () => {
     expect(version.createdBy.id).toEqual(me.id);
     const createdAt = DateTime.fromISO(version.createdAt);
     expect(createdAt.diffNow().as('seconds')).toBeGreaterThan(-30);
-    await expectEqualContent(bucket, file.downloadUrl, fakeFile);
+    await expectEqualContent(file.url, fakeFile);
     expect(version.parents[0].id).toEqual(file.id);
   });
 
@@ -217,7 +213,7 @@ describe('File e2e', () => {
     input: FakeFile,
   ) {
     expect(updated.id).toEqual(initial.id);
-    await expectEqualContent(bucket, updated.downloadUrl, input);
+    await expectEqualContent(updated.url, input);
     expect(updated.size).toEqual(input.size);
     expect(updated.mimeType).toEqual(input.mimeType);
     const createdAt = DateTime.fromISO(updated.createdAt);
diff --git a/test/utility/fragments.ts b/test/utility/fragments.ts
index af47ca0ccf..6e5747ceb2 100644
--- a/test/utility/fragments.ts
+++ b/test/utility/fragments.ts
@@ -394,12 +394,12 @@ export const fileNode = gql`
     ... on FileVersion {
       mimeType
       size
-      downloadUrl
+      url
     }
     ... on File {
       mimeType
       size
-      downloadUrl
+      url
       modifiedAt
       modifiedBy {
         ...user
@@ -421,7 +421,7 @@ export type RawBaseFileNode = RawNode<
 >;
 export type RawDirectory = RawBaseFileNode;
 export type RawFileVersion = RawBaseFileNode &
-  RawNode<FileVersion, keyof IFileNode, { downloadUrl: string }>;
+  RawNode<FileVersion, keyof IFileNode, { url: string }>;
 export type RawFile = RawFileVersion &
   RawNode<
     File,