register lambdas

lib/node_types/node.lua

@@ -10,7 +10,9 @@ function Node:initialize(config)
   end
 
   if self.name ~= nil then
-    Registry.register(self.name, self)
+    Registry.register(self.name, function()
+      return Node:new(config)
+    end)
   end
 end
 

lib/registry.lua

@@ -1,14 +1,18 @@
-local registeredNodes = {}
-
 local Registry = {}
+Registry.registeredNodes = {}
 
 function Registry.register(name, node)
-  registeredNodes[name] = node;
+  Registry.registeredNodes[name] = node;
 end
 
 function Registry.getNode(name)
   if type(name) == 'string' then
-    return registeredNodes[name]
+    local node = Registry.registeredNodes[name]
+    if type(node) == "function" then
+      return node()
+    else
+      return node
+    end
   else
     return name
   end

spec/functional/registry_spec.lua

@@ -0,0 +1,35 @@
+local BehaviourTree = require 'lib/behaviour_tree'
+local Registry = require 'lib/registry'
+local Task = BehaviourTree.Task
+
+describe('Registry', function() 
+  local subject, task1
+  before_each(function()
+	subject = Registry
+	task1 = BehaviourTree.Task:new()
+	Registry.register('node', task1)
+	Registry.register('func', function() return BehaviourTree.Task:new() end)
+	subject = Registry
+  end)
+
+  teardown(function()
+    Registry.registeredNodes = {}
+  end)
+
+  it('should be able to return a simple node', function()
+	assert.has_no.errors(function() subject:getNode('node'))
+  end)
+  it('should return the same simple node each time', function()
+    local node1 = subject:getNode('node')
+	local node2 = subject:getNode('node')
+	assert.are_equals(node1, node2)
+  end)
+  it('should be able to return a func node', function()
+	assert.has_no.errors(function() subject:getNode('func'))
+  end)
+   it('should return new instances of func nodes', function()
+    local node1 = subject:getNode('func')
+    local node2 = subject:getNode('func')
+	assert.are_not.equals(node1, node2)
+  end)
+end)