Fix coverage

TiMESPLiNTER · Jan 31, 2024 · e9442a9 · e9442a9
1 parent e2953c7
commit e9442a9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/jest.config.ts b/jest.config.ts
@@ -10,7 +10,10 @@ const config: Config = {
             "lines": 93,
             "statements": 93
         }
-    }
+    },
+    collectCoverageFrom: [
+        "src/**/*"
+    ],
 };
 
 export default config;
diff --git a/src/pimple.ts b/src/pimple.ts
@@ -4,7 +4,7 @@ import Container, {ServiceKey} from "./container";
 import ServiceProvider from "./serviceProvider";
 
 /** Declaration types */
-type ProviderDeclaration<T> = Function|ServiceProvider<T>;
+type ProviderDeclaration<T> = (container: Pimple<T>) => void|ServiceProvider<T>;
 type LazyServiceDefinition<T, S> = (container: Pimple<T>) => S;
 type ProtectedServiceDefinition<T, S> = () => LazyServiceDefinition<T, S>;
 type ServiceDefinition<T, S> = LazyServiceDefinition<T, S>|ProtectedServiceDefinition<T, S>|S;
@@ -160,12 +160,8 @@ export default class Pimple<T> implements Container<T>
     public register(provider: ProviderDeclaration<T>): Pimple<T> {
         if (this.instanceOfServiceProvider(provider) && provider.register instanceof Function) {
             provider.register(this);
-            return this;
-        }
-
-        if (provider instanceof Function) {
+        } else {
             provider(this);
-            return this;
         }
 
         return this;