Merge pull request #14 from vercel/add-computed-partition-key-test

add a test for pull/13

Author: nkzawa

src/contains-partition-keys.ts

@@ -20,7 +20,8 @@ function conditionKeyNodes(node: { [x: string]: any }): any[] {
 
 function toPartitionKey(node: { [x: string]: any }): string | null {
   if (node.type === "scalar_member_expression") {
-    return `${toPartitionKey(node.object) || ""}/${node.property.name || node.property.value}`;
+    return `${toPartitionKey(node.object) || ""}/${node.property.name ||
+      node.property.value}`;
   }
   return null;
 }

test/partition-keys.ts

@@ -45,3 +45,9 @@ export const notContainsMultipleKeys = testPartitionKeys(
 );
 
 export const empty = testPartitionKeys("SELECT * FROM c", []);
+
+export const computedProperty = testPartitionKeys(
+  'SELECT * FROM c WHERE c["partition"] = @pkValue',
+  ["/partition"],
+  true
+);