Add cleanup function

README.md

@@ -191,7 +191,7 @@ new B(); // "Hello A2!"
 
 ## Unit testing
 
-You can use `override` or `assign` for provide mocks in you dependencies.
+You can use `assign` for provide mocks into you dependencies.
 
 ```typescript
 // world.ts
@@ -212,11 +212,13 @@ export class Hello {
 }
 
 // hello.test.ts
-import { assign } from "node-provide";
+import { assign, cleanup } from "node-provide";
 import { World } from "./world";
 import { Hello } from "./hello";
 // ...
 
+after(cleanup);
+
 it("It works!", () => {
   const worldMock = {
     hello: jest.fn(),
@@ -420,17 +422,24 @@ class {
 
 **isolate**
 
+**cleanup**
 
-**reset**
-
-Clean all cached dependency instances and overrides. Its needed for testing. Has no parameters.
+Clean all cached dependency instances. Its needed for testing. Has no parameters.
 
 ```javascript
 // ...
-after(reset);
+after(cleanup);
 // ...
 ```
 
+**reset**
+
+Clean all cached dependency instances and overrides. Has no parameters.
+
+```javascript
+reset()
+```
+
 ---
 
 If you have questions or something else for me or this project, maybe architectures questions, improvement ideas or anything else, please make issue.

index.test.ts

@@ -11,6 +11,7 @@ import {
   assign,
   inject,
   reset,
+  cleanup,
   getZoneId,
   RootZoneId,
   zoneIndex,
@@ -101,6 +102,18 @@ test("Should cache override", () => {
   expect(instances[RootZoneId].get(A2)).toBeInstanceOf(A3);
 });
 
+test("Should work cleanup", () => {
+  class A {}
+  class B {}
+  const m = {};
+  expect(resolve(A)).toBeInstanceOf(A);
+  assign(B, m);
+  expect(resolve(B)).toBe(m);
+  expect(instances[RootZoneId].size).toBe(2);
+  cleanup();
+  expect(instances[RootZoneId]).toBeUndefined();
+});
+
 test("Should work reset", () => {
   class A {}
   class A2 extends A {}

index.ts

@@ -288,21 +288,25 @@ export function assign(...depOrDepInstPairs: any[]) {
   }
 }
 
-export function reset() {
+export function cleanup() {
   Object.keys(instances).forEach((id) => {
     instances[id].clear();
     delete instances[id];
   });
-  Object.keys(overrides).forEach((id) => {
-    overrides[id].clear();
-    delete overrides[id];
-  });
   Object.keys(resolvePhases).forEach((id) => {
     resolvePhases[id].clear();
     delete resolvePhases[id];
   });
 }
 
+export function reset() {
+  cleanup();
+  Object.keys(overrides).forEach((id) => {
+    overrides[id].clear();
+    delete overrides[id];
+  });
+}
+
 class Chan {
   private signal: () => void;
   private next: Promise<void>;

jest-cleanup-after-each.js

@@ -1 +1 @@
-afterEach(require(".").reset);
+afterEach(require(".").cleanup);

package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-provide",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Async context based Dependency Injection for Node.JS",
   "repository": {
     "url": "https://github.com/betula/node-provide"