diff --git a/source/poodinis/container.d b/source/poodinis/container.d index ac5be1f..ee2d530 100644 --- a/source/poodinis/container.d +++ b/source/poodinis/container.d @@ -59,22 +59,26 @@ class Container { } } - public ClassType resolve(ClassType)() { - TypeInfo resolveType = typeid(ClassType); + public RegistrationType resolve(RegistrationType)() { + TypeInfo resolveType = typeid(RegistrationType); Registration* registration = resolveType in registrations; if (!registration) { throw new ResolveException("Type not registered.", resolveType); } - ClassType instance = cast(ClassType) registration.getInstance(); - this.autowire!(ClassType)(instance); + RegistrationType instance = cast(RegistrationType) registration.getInstance(); + this.autowire!(RegistrationType)(instance); return instance; } - public void clearRegistrations() { + public void clearAllRegistrations() { registrations.clear(); } + public void removeRegistration(RegistrationType)() { + registrations.remove(typeid(RegistrationType)); + } + public static Container getInstance() { if (instance is null) { instance = new Container(); diff --git a/test/poodinis/containertest.d b/test/poodinis/containertest.d index b5292a8..a10e3e5 100644 --- a/test/poodinis/containertest.d +++ b/test/poodinis/containertest.d @@ -90,7 +90,7 @@ version(unittest) { unittest { auto container = new Container(); container.register!(TestClass)(); - container.clearRegistrations(); + container.clearAllRegistrations(); assertThrown!ResolveException(container.resolve!(TestClass)(), "Resolving cleared type does not fail"); } @@ -157,5 +157,13 @@ version(unittest) { auto mouse = container.resolve!ComponentMouse; auto cat = container.resolve!ComponentCat; assert(mouse.cat is cat && cat.mouse is mouse, "Circular dependencies should be autowirable"); - } + } + + // Test remove registration + unittest { + auto container = new Container(); + container.register!TestClass; + container.removeRegistration!TestClass; + assertThrown!ResolveException(container.resolve!TestClass); + } } \ No newline at end of file