Fixed a bug, when ngSrc was undefined at the begninig, image wasn't refreshed after ngSrc changed

Author: Radek Zieliński

src/angular-wurfl-image-tailor.js

@@ -26,21 +26,22 @@ angular.module('angular-wurfl-image-tailor', [])
             replace: false,
             scope:{},
             template: function(element, attributes){
-                if (attributes['ngSrc']) {
+                if ('ngSrc' in attributes) {
                     return '<div class="wit"><img ng-src="{{wit_link}}"/></div>';
                 } else {
                     return '<div class="wit"><img src="{{wit_link}}"/></div>';
                 }
             },
             link: function (scope, element, attributes) {
-                var srcAName = attributes['ngSrc'] ? 'ngSrc' : 'src';
-
-                if(!attributes[srcAName]) {
-                    scope.wit_link = '';
-                    return;
-                }
+                var srcAName = 'ngSrc' in attributes ? 'ngSrc' : 'src';
+                scope.wit_link = '';
 
                 attributes.$observe(srcAName, function (src) {
+                    if(!src) {
+                        scope.wit_link = '';
+                        return;
+                    }
+
                     var wit_link_pieces = [witUrls.get()];
                     angular.forEach(attributes['$attr'], function (attr) {
                         if (attr != 'src' && attr != 'ng-src') {

test/unit/directiveSpec.js

@@ -69,4 +69,17 @@ describe("Unit: Testing angular directive for WURFL Image Tailor", function() {
         expect(elt.html()).to.be.a('string');
         expect(elt.html()).to.be.equal('<div class="wit"><img ng-src="//wit1.wurfl.io/http://test.com/image2.jpg" src="//wit1.wurfl.io/http://test.com/image2.jpg"></div>');
     });
+
+    it('should change when ngSrc directive variable change from undefined', function () {
+        scope.myUrl = undefined;
+        var elt = angular.element('<img-wit ng-src="{{myUrl}}"></img-wit>');
+        compile(elt)(scope);
+        scope.$digest();
+        scope.myUrl = sce.trustAsResourceUrl('http://test.com/image2.jpg');
+        scope.$digest();
+
+        expect(elt.html()).to.be.a('string');
+        expect(elt.html()).to.be.equal('<div class="wit"><img ng-src="//wit.wurfl.io/http://test.com/image2.jpg" src="//wit.wurfl.io/http://test.com/image2.jpg"></div>');
+    });
+
 });