Merge pull request #8 from mheffner/fix/catch-vol-type-mismatch

Catch error when size (as number) is passed as vol type.
aryounce · Jul 22, 2019 · 374947a · 374947a
2 parents cfb9aff + 4d87336
commit 374947a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/functions/ebs.ts b/src/functions/ebs.ts
@@ -20,7 +20,7 @@ function _ec2_ebs(settings: InvocationSettings, storageType: EBSStorageType, vol
     }
 
     if (volumeType) {
-        volumeType = volumeType.toLowerCase()
+        volumeType = volumeType.toString().toLowerCase()
     }
 
     let ebsPrices = new EBSPrice(settings, storageType, volumeType, volumeUnits.toString())

diff --git a/tests/functions/ebs_test.ts b/tests/functions/ebs_test.ts
@@ -72,6 +72,10 @@ export class EBSFunctionTestSuite extends TestSuite {
             t.willThrow(function() {
                 EC2_EBS_GP2_GB("foo", "us-east-1")
             }, "unable to parse volume units")
+
+            t.willThrow(function() {
+                EC2_EBS_GB(s, 400, "gp2")
+            }, "invalid EBS volume type")
         })
     }
 }