Refactor EnvironmentArranger

Author: rsaladocid

src/apps/mooc_backend/config/dependency-injection/application_test.yaml

@@ -3,5 +3,5 @@ imports:
 
 services:
   Mooc.EnvironmentArranger:
-    class: ../../../../../tests/Contexts/Shared/infrastructure/arranger/MongoEnvironmentArranger
+    class: ../../../../../tests/Contexts/Shared/infrastructure/mongo/MongoEnvironmentArranger
     arguments: ['@Mooc.ConnectionManager']

tests/Contexts/Mooc/Courses/infrastructure/persistence/CourseRepository.test.ts

@@ -7,7 +7,7 @@ const repository: CourseRepository = container.get('Mooc.courses.CourseRepositor
 const environmentArranger: Promise<EnvironmentArranger> = container.get('Mooc.EnvironmentArranger');
 
 beforeEach(async () => {
-  await (await environmentArranger).arrange('courses');
+  await (await environmentArranger).arrange();
 });
 
 afterAll(async () => {

tests/Contexts/Shared/infrastructure/arranger/EnvironmentArranger.ts

@@ -1,9 +1,5 @@
 export abstract class EnvironmentArranger {
-  public async arrange(name: string): Promise<void> {
-    await this.clean(name);
-  }
-
-  protected abstract clean(name: string): Promise<void>;
+  public abstract arrange(): Promise<void>;
 
   public abstract close(): Promise<void>;
 }

tests/Contexts/Shared/infrastructure/arranger/MongoEnvironmentArranger.ts

@@ -0,0 +1,38 @@
+import { MongoClient } from 'mongodb';
+import { EnvironmentArranger } from '../arranger/EnvironmentArranger';
+
+export class MongoEnvironmentArranger extends EnvironmentArranger {
+  constructor(private _client: Promise<MongoClient>) {
+    super();
+  }
+
+  public async arrange(): Promise<void> {
+    await this.cleanDatabase();
+  }
+
+  protected async cleanDatabase(): Promise<void> {
+    const collections = await this.collections();
+
+    for (const collection of collections) {
+      await (await this.client()).db().dropCollection(collection);
+    }
+  }
+
+  private async collections(): Promise<string[]> {
+    const client = await this.client();
+    const collections = await client
+      .db()
+      .listCollections(undefined, { nameOnly: true })
+      .toArray();
+
+    return collections.map(collection => collection.name);
+  }
+
+  protected client(): Promise<MongoClient> {
+    return this._client;
+  }
+
+  public async close(): Promise<void> {
+    return (await this.client()).close();
+  }
+}