diff --git a/src/angular-pickadate.js b/src/angular-pickadate.js index bff80cc..efdb4e2 100755 --- a/src/angular-pickadate.js +++ b/src/angular-pickadate.js @@ -255,6 +255,7 @@ link: function(scope, element, attrs, ngModel) { var allowMultiple = attrs.hasOwnProperty('multiple'), + customClass = attrs.hasOwnProperty('customClass') ? attrs.customClass : '', selectedDates = [], wantsModal = element[0] instanceof HTMLInputElement, compiledHtml = $compile(TEMPLATE)(scope), @@ -341,7 +342,15 @@ }); element.after(compiledHtml.addClass('pickadate-modal')); + + if (customClass) { + element.after(compiledHtml.addClass(customClass)); + } } else { + if (customClass) { + element.after(compiledHtml.addClass(customClass)); + } + element.append(compiledHtml); } diff --git a/test/angular-pickadate.spec.js b/test/angular-pickadate.spec.js index 6220499..44a2dba 100644 --- a/test/angular-pickadate.spec.js +++ b/test/angular-pickadate.spec.js @@ -583,7 +583,7 @@ describe('pickadate', function () { describe('When used as a modal', function() { var inputHtml = '
' + - '' + + '' + '
', input, form; @@ -596,10 +596,15 @@ describe('pickadate', function () { element = $('.pickadate'); }); + it('adds the pickadate-modal class', function() { expect(element).to.have.class('pickadate-modal'); }); + it('adds the pickadate-custom class', function() { + expect(element).to.have.class('pickadate-custom'); + }); + it('renders the datepicker already hidden', function() { expect(element).to.have.class('ng-hide'); });