diff --git a/platform/features/timeline/test/actions/TimelineTraverserSpec.js b/platform/features/timeline/test/actions/TimelineTraverserSpec.js
index 3d9dcfdb4b..1ccb7625c3 100644
--- a/platform/features/timeline/test/actions/TimelineTraverserSpec.js
+++ b/platform/features/timeline/test/actions/TimelineTraverserSpec.js
@@ -42,6 +42,7 @@ define([
                         'getModel'
                     ]
                 ),
+                mockRelationships,
                 model = testModels[id];
 
             mockDomainObject.getId.andReturn(id);
@@ -53,6 +54,31 @@ define([
                         false;
             });
 
+            if (!!model.composition) {
+                mockDomainObject.useCapability.andCallFake(function (c) {
+                    return c === 'composition' &&
+                        Promise.resolve(model.composition.map(function (id) {
+                            return mockDomainObjects[id]
+                        }));
+                });
+            }
+
+            if (!!model.relationships) {
+                mockRelationships = jasmine.createSpyObj(
+                    'relationship',
+                    ['getRelatedObjects']
+                );
+                mockRelationships.getRelatedObjects.andCallFake(function (k) {
+                    var ids = model.relationships[k] || [];
+                    return Promise.resolve(ids.map(function (id) {
+                        return mockDomainObjects[id];
+                    }));
+                });
+                mockDomainObject.getCapability.andCallFake(function (c) {
+                    return c === 'relationship' && mockRelationships;
+                });
+            }
+
             mockDomainObjects[id] = mockDomainObject;
         }