server/AyaNova/Controllers/TrialController.cs

@@ -52,7 +52,7 @@ namespace AyaNova.Api.Controllers
         /// <param name="timeZoneOffset">Value in hours of local time zone offset from UTC / GMT. This ensures that data is generated relative to the desired time zone</param>
         /// <returns></returns>
         [HttpPost("seed/{size}/{timeZoneOffset}")]
-        public async Task<IActionResult> SeedTrialDatabase([FromRoute] string size,[FromRoute] decimal timeZoneOffset )
+        public async Task<IActionResult> SeedTrialDatabase([FromRoute] string size, [FromRoute] decimal timeZoneOffset)
         {
             if (!serverState.IsOpen)
             {
@@ -69,32 +69,16 @@ namespace AyaNova.Api.Controllers
                 return BadRequest(new ApiErrorResponse(ApiErrorCode.INVALID_OPERATION, null, "Current license is not a trial license key.  Only a trial can be seeded."));
             }
 
-            Seeder.SeedLevel seedLevel = Seeder.SeedLevel.SmallOneManShopTrialDataSet;
-            switch (size.ToLowerInvariant())
-            {
-                case "small":
-                    seedLevel = Seeder.SeedLevel.SmallOneManShopTrialDataSet;
-                    break;
-                case "medium":
-                    seedLevel = Seeder.SeedLevel.MediumLocalServiceCompanyTrialDataSet;
-                    break;
-                case "large":
-                    seedLevel = Seeder.SeedLevel.LargeCorporateMultiRegionalTrialDataSet;
-                    break;
-                case "huge":
-                    seedLevel = Seeder.SeedLevel.HugeForLoadTest;
-                    break;
-                default:
-                    return BadRequest(new ApiErrorResponse(ApiErrorCode.NOT_FOUND, "size", "Valid values are \"small\", \"medium\", \"large\", \"huge\""));
-            }
-
+            Seeder.SeedLevel seedLevel = Seeder.StringToSeedLevel(size);
+            if (seedLevel == Seeder.SeedLevel.NotValid)
+                return BadRequest(new ApiErrorResponse(ApiErrorCode.NOT_FOUND, "size", "Valid values are \"small\", \"medium\", \"large\", \"huge\""));
 
             //Create the job here
 
             JObject o = JObject.FromObject(new
             {
                 seedLevel = seedLevel,
-                timeZoneOffset=timeZoneOffset
+                timeZoneOffset = timeZoneOffset
             });
 
             OpsJob j = new OpsJob();