Adds valid tests

OnedocLabs · May 26, 2024 · e5c9c74 · e5c9c74
1 parent 9d217a1
commit e5c9c74
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 24 deletions.
diff --git a/output.pdf b/output.pdf
diff --git a/src/Client.ts b/src/Client.ts
@@ -22,10 +22,6 @@ export declare namespace FileForgeClient {
         maxRetries?: number;
     }
 
-    interface Response{
-        file?: string;
-        url?:string;
-    }
 }
 
 export class FileForgeClient {
@@ -42,7 +38,7 @@ export class FileForgeClient {
         files: File[] | fs.ReadStream[],
         request: FileForge.GenerateRequest,
         requestOptions?: FileForgeClient.RequestOptions
-    ): Promise<FileForgeClient.Response>{
+    ): Promise<stream.Readable>{
         const _request = core.newFormData();
         const options = await serializers.GenerateRequestOptions.jsonOrThrow(request.options, {
             unrecognizedObjectKeys: "passthrough",
@@ -76,19 +72,8 @@ export class FileForgeClient {
         });
 
         if (_response.ok) {
-                const chunks: any[] = [];
-
-                for await (let chunk of _response.body) {
-                    chunks.push(chunk);
-                }
-
-                const buffer: Buffer = Buffer.concat(chunks);
-
-                if (request.options?.host !== true){
-                    return {"file": buffer.toString()};
-                }else{
-                    return JSON.parse(buffer.toString())
-                }
+
+                return _response.body
         }
 
         if (_response.error.reason === "status-code") {

diff --git a/tests/custom.test.ts b/tests/custom.test.ts
@@ -46,14 +46,22 @@ describe("test", () => {
             apiKey: FILEFORGE_API_KEY
         });
 
-        const pdf:FileForgeClient.Response = await ff.generate(
+        const pdf = await ff.generate(
             [htmlFile, cssFile], 
             {
                 options: {}
             }
-        );
+        );       
+
+        const chunks: any[] = [];
+
+        for await (let chunk of pdf) {
+            chunks.push(chunk);
+        }
+
+        const buffer: Buffer = Buffer.concat(chunks);
 
-        await writeFile("output.pdf", pdf.file!);
+        await writeFile("output.pdf", buffer.toString()!);
     }, 10_000_000);
 
 
@@ -71,16 +79,23 @@ describe("test", () => {
             apiKey: FILEFORGE_API_KEY
         });
 
-        const pdf:FileForgeClient.Response = await ff.generate(
+        const pdf = await ff.generate(
             [htmlFile, cssFile], 
             {
                 options: {
                     host: true,
                 }
             }
         );
-
-        expect(pdf.url).not.toBeNull();
+        const chunks: any[] = [];
+
+        for await (let chunk of pdf) {
+            chunks.push(chunk);
+        }
+
+        const buffer: Buffer = Buffer.concat(chunks);
+
+        expect(JSON.parse(buffer.toString()).url).not.toBeNull();
 
     }, 10_000_000);