Fixed bug where comment is not removed on removing embedded frameworks. (#5)

Author: kelvinhokk

lib/pbxProject.js

@@ -484,10 +484,11 @@ pbxProject.prototype.removeFromPbxBuildFileSection = function(file) {
         if (this.pbxBuildFileSection()[uuid].fileRef_comment == file.basename) {
             file.uuid = uuid;
             delete this.pbxBuildFileSection()[uuid];
+
+            var commentKey = f("%s_comment", uuid);
+            delete this.pbxBuildFileSection()[commentKey];
         }
     }
-    var commentKey = f("%s_comment", file.uuid);
-    delete this.pbxBuildFileSection()[commentKey];
 }
 
 pbxProject.prototype.addPbxGroup = function(filePathsArray, name, path, sourceTree) {

test/removeFramework.js

@@ -165,6 +165,33 @@ exports.removeFramework = {
             test.ok(current.indexOf(expectedPath) == -1);
         }
 
+        test.done();
+    },
+    'should remove embedded frameworks': function (test) {
+        var newFile = proj.addFramework('/path/to/Custom.framework', { customFramework: true, embed:true, sign:true }),
+            frameworks = proj.pbxFrameworksBuildPhaseObj(),
+            buildFileSection = proj.pbxBuildFileSection(),
+            bfsLength = Object.keys(buildFileSection).length;
+
+        test.equal(frameworks.files.length, 16);
+        test.equal(62, bfsLength);
+
+        var deletedFile = proj.removeFramework('/path/to/Custom.framework', { customFramework: true, embed:true }),
+            frameworks = proj.pbxFrameworksBuildPhaseObj(),
+            buildFileSection = proj.pbxBuildFileSection(),
+            bfsLength = Object.keys(buildFileSection).length;
+
+        test.equal(frameworks.files.length, 15);
+        test.equal(58, bfsLength);
+
+        var frameworkPaths = frameworkSearchPaths(proj);
+        expectedPath = '"/path/to"';
+
+        for (i = 0; i < frameworkPaths.length; i++) {
+            var current = frameworkPaths[i];
+            test.ok(current.indexOf(expectedPath) == -1);
+        }
+
         test.done();
     }
 }