From 0535ee631f0df494ba76cf82c46fdec3b0141a97 Mon Sep 17 00:00:00 2001
From: David de Boer <david@ddeboer.nl>
Date: Wed, 10 Jul 2024 12:13:55 +0200
Subject: [PATCH] test: Fix test

---
 test/iterator.test.ts | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/test/iterator.test.ts b/test/iterator.test.ts
index b969b48..1ef3252 100644
--- a/test/iterator.test.ts
+++ b/test/iterator.test.ts
@@ -10,6 +10,7 @@ import {fileURLToPath} from 'url';
 import removeDirectory from '../src/utils/removeDir.js';
 import {NamedNode} from '@rdfjs/types';
 import getSPARQLQuery from '../src/utils/getSPARQLQuery.js';
+import getEndpoint from '../src/utils/getEndpoint.js';
 chai.use(chaiAsPromised);
 
 describe('Iterator Class', () => {
@@ -57,8 +58,10 @@ describe('Iterator Class', () => {
       };
       const pipeline = new Pipeline(configuration, {silent: true});
       const stageConfig = configuration.stages[0];
-      const stage = new Stage(pipeline, stageConfig);
-      const iterator = new Iterator(stage);
+      const iterator = new Iterator(
+        Query.from(getSPARQLQuery(stageConfig.iterator.query, 'select')),
+        getEndpoint(new Stage(pipeline, stageConfig))
+      );
       chai.expect(iterator).to.be.an.instanceOf(Iterator);
       chai.expect(iterator).to.be.an.instanceOf(EventEmitter);
       chai.expect(iterator).to.have.property('query');
@@ -68,7 +71,7 @@ describe('Iterator Class', () => {
       chai.expect(iterator).to.have.property('totalResults', 0);
     });
   });
-  describe.skip('run', () => {
+  describe('run', () => {
     it('should emit "data" and "end" events with the correct $this and numResults', async () => {
       const configuration: Configuration = {
         name: 'Example Pipeline',
@@ -104,8 +107,10 @@ describe('Iterator Class', () => {
       };
       const pipeline = new Pipeline(configuration, {silent: true});
       const stageConfig = configuration.stages[0];
-      const stage = new Stage(pipeline, stageConfig);
-      const iterator = new Iterator(stage);
+      const iterator = new Iterator(
+        Query.from(getSPARQLQuery(stageConfig.iterator.query, 'select')),
+        getEndpoint(new Stage(pipeline, stageConfig))
+      );
       const emittedEvents: {event: string; bindings?: NamedNode}[] = [];
       async function runIteratorWithPromise(): Promise<boolean> {
         return new Promise((resolve, reject) => {