WIP #939 improved the validate arch plugin

deepforge-dev · Dec 20, 2016 · c0a44af · c0a44af
1 parent 7ccf8da
commit c0a44af
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/plugins/ValidateArchitecture/ValidateArchitecture.spec.js b/test/plugins/ValidateArchitecture/ValidateArchitecture.spec.js
@@ -131,21 +131,20 @@ describe('ValidateArchitecture', function () {
             plugin.main((err, result) => {
                 var invalidLayers = result.messages[0].message.errors.map(msg => msg.id);
                 expect(result.messages[0]).to.not.equal(undefined);
+                expect(invalidLayers.length).to.equal(2);
                 done();
             });
         });
 
         describe('w/o torch support', function() {
             before(function(done) {
                 preparePlugin(() => {
-                    plugin.setTorchInstalled(false);
                     done();
                 });
             });
 
-            after(() => plugin.setTorchInstalled(true));
-
             it('should return "null" for messages', function(done) {
+                plugin.setTorchInstalled(false);
                 plugin.main((err, result) => {
                     var errors = result.messages[0].message.errors;
                     expect(errors).to.equal(null);