1 files changed, 30 insertions, 22 deletions
diff --git a/install/ui/test/navigation_tests.js b/install/ui/test/navigation_tests.js
index bc198ac0c..916259b41 100644
--- a/install/ui/test/navigation_tests.js
+++ b/install/ui/test/navigation_tests.js
@@ -23,28 +23,36 @@ module('navigation');
 
 test("Testing nav_create().", function() {
 
-    var mock_tabs_lists =
-        [
-            { name:'identity', label:'IDENTITY', children: [
-                {name:'user', entity:'user'},
-                {name:'group', entity:'group'}
-            ]}];
-
-    var entity = IPA.entity({name: 'user'});
-    entity.setup = function(container){
-        user_mock_called = true;
-        same(container[0].id,'user','user id');
-        same(container[0].nodeName,'DIV','user div');
-    };
-    IPA.add_entity(entity);
-
-    entity = IPA.entity({name: 'group'});
-    entity.setup = function(container){
-        group_mock_called = true;
-        same(container[0].id,'group','group id');
-        same(container[0].nodeName,'DIV','group Div');
-    };
-    IPA.add_entity(entity);
+    var mock_tabs_lists =  [
+        { name:'identity', label:'IDENTITY', children: [
+            {name:'user', entity:'user'},
+            {name:'group', entity:'group'}
+        ]}];
+
+    var entity;
+
+    IPA.register_entity( function() {
+        var that = IPA.entity({name: 'user'});
+        that.setup = function(container){
+            user_mock_called = true;
+            same(container[0].id,'user','user id');
+            same(container[0].nodeName,'DIV','user div');
+        }
+        return that;
+    });
+
+    IPA.register_entity( function(){
+
+        var that  = IPA.entity({name: 'group'});
+        that.setup = function(container){
+            group_mock_called = true;
+            same(container[0].id,'group','group id');
+            same(container[0].nodeName,'DIV','group Div');
+        };
+        return that;
+    });
+
+    IPA.start_entities();
 
     IPA.metadata = {};
     var navigation = $('<div id="navigation"/>').appendTo(document.body);