test coverage

SC5 · May 10, 2016 · ee28c35 · ee28c35
1 parent 543b290
commit ee28c35
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/app/js/controllers/sections.js b/lib/app/js/controllers/sections.js
@@ -38,7 +38,7 @@ angular.module('sgApp')
     }
 
     $scope.isMainSection = function(section) {
-      return $scope.currentSection.indexOf('.') === -1;
+      return section.reference.indexOf('.') === -1;
     };
 
     $scope.isEmptyMainSection = function(section) {

diff --git a/test/angular/unit/controllers/sections.test.js b/test/angular/unit/controllers/sections.test.js
@@ -87,6 +87,16 @@ describe('SectionsCtrl', function() {
   });
 
   describe('empty main section detection', function() {
+    it('should return true if main section', function() {
+      var section = {
+        header: 'Section header text',
+        reference: '1',
+        renderMarkup: '',
+        markup: ''
+      };
+      expect(scope.isMainSection(section)).to.eql(true);
+    });
+
     it('should return true for empty main sections', function() {
       var section = {
         header: 'Section header text',