Merge pull request #274 from apigee-127/load-test-cli

Expose swagger-test-templates load-testing API

Author: theganyo

bin/swagger-project.js

@@ -74,6 +74,7 @@ app
   .option('-f, --test-module <module>', 'one of: ' + testmodules)
   .option('-t, --assertion-format <type>', 'one of: ' + assertiontypes)
   .option('-o, --force', 'allow overwriting of all existing test files matching those generated')
+  .option('-l, --load-test [path]', 'generate load-tests for specified operations')
   .action(execute(project.generateTest));
 
 app.parse(process.argv);

lib/commands/project/project.js

@@ -399,6 +399,7 @@ function testGenerate(directory, options, cb) {
   var testModule = options.testModule || TEST_MODULES[0];
   var assertionFormat = options.assertionFormat || TEST_ASSERTION_TYPES[0];
   var overwriteAll = options.force || false;
+  var loadTesting = options.loadTest || false;
   directory = directory || process.cwd();
 
   findProjectFile(directory, null, function(err, projPath) {
@@ -456,6 +457,17 @@ function testGenerate(directory, options, cb) {
             assertionFormat: assertionFormat
           };
 
+          // pass list of paths targeted for load testing
+          if (loadTesting) {
+            if ((typeof loadTesting) !== 'boolean' && fs.existsSync(path.join(directory, loadTesting))) {
+              config.loadTest = parseJsonFile(directory, loadTesting).loadTargets;
+            } else if (fs.existsSync(path.join(directory, 'load-config.json'))){
+              config.loadTest = parseJsonFile(directory, 'load-config.json').loadTargets;
+            } else {
+              return cb(new Error('Config file not found. Please specify a load test config or add load-config.json file to your project directory.'));
+            }
+          }
+
           var finalResult = template.testGen(result, config);
           var existingFiles = fs.readdirSync(path.join(directory, 'test/api/client'));
           var skipAll = false;
@@ -528,6 +540,10 @@ function testGenerate(directory, options, cb) {
   });
 }
 
+function parseJsonFile(directory, filePath) {
+  return JSON.parse(fs.readFileSync(path.join(directory, filePath)));
+}
+
 function installDependencies(directory, message, cb) {
   spawn('npm', ['install'], directory, function(err) {
     if (err) {

package.json

@@ -31,7 +31,7 @@
     "serve-static": "^1.9.2",
     "swagger-converter": "^0.1.7",
     "swagger-editor": "^2.9.2",
-    "swagger-test-templates": "^1.1.0",
+    "swagger-test-templates": "^1.2.0",
     "swagger-tools": "^0.9.0"
   },
   "devDependencies": {

test/commands/project/project.js

@@ -515,6 +515,62 @@ describe('project', function() {
         });
       });
     });
+
+    it ('should create load tests from myLoadTest.json', function(done) {
+      var loadTargets = {
+        "loadTargets": [
+          {
+            "pathName":"/hello",
+            "operation": "get",
+            "load": {
+              "requests": 1000,
+              "concurrent": 100
+            }
+          }
+        ]
+      };
+
+      fs.writeFileSync(path.join(projPath, 'myLoadTest.json'), JSON.stringify(loadTargets));
+
+      var options = {loadTest: './myLoadTest.json', force: true};
+
+      project.generateTest(projPath, options, function(err) {
+        fs.existsSync(path.resolve(projPath, 'test/api/client/test-test.js')).should.be.ok;
+        fs.existsSync(path.resolve(projPath, 'test/api/client/hello-test.js')).should.be.ok;
+        fs.readFile(path.resolve(projPath, 'test/api/client/hello-test.js'), {encoding: 'utf8'}, function(err, string) {
+          string.search('arete').should.be.ok;
+          done(err);
+        });
+      });
+    });
+
+    it ('should create load tests from load-config.json', function(done) {
+      var loadTargets = {
+        "loadTargets": [
+          {
+            "pathName":"/hello",
+            "operation": "get",
+            "load": {
+              "requests": 1000,
+              "concurrent": 100
+            }
+          }
+        ]
+      };
+
+      fs.writeFileSync(path.join(projPath, 'load-config.json'), JSON.stringify(loadTargets));
+
+      var options = {loadTest: true, force: true};
+
+      project.generateTest(projPath, options, function(err) {
+        fs.existsSync(path.resolve(projPath, 'test/api/client/test-test.js')).should.be.ok;
+        fs.existsSync(path.resolve(projPath, 'test/api/client/hello-test.js')).should.be.ok;
+        fs.readFile(path.resolve(projPath, 'test/api/client/hello-test.js'), {encoding: 'utf8'}, function(err, string) {
+          string.search('arete').should.be.ok;
+          done(err);
+        });
+      });
+    });
   });
 
 });