Extended integration tests with AzureSearch

YuriyIvon · YuriyIvon · commit da22f6f84b6c · 2025-10-26T20:08:39.000Z
diff --git a/integration-tests/ConnectionStrings.ps1 b/integration-tests/ConnectionStrings.ps1
@@ -1,5 +1,6 @@
 #Update the sample connection strings with your specific connection settings
 $connectionStrings = @{
+  AzureSearch = "Endpoint=myendpoint;ApiKey=myapikey"
   SqlServer = "Data Source=.;Initial Catalog=benchmark;Integrated Security=True;"
   Postgres = "Host=localhost;Port=5432;Database=benchmark;Username=postgres;Password=password"
   PostgresJsonb = "Host=localhost;Port=5432;Database=benchmark;Username=postgres;Password=password"
diff --git a/integration-tests/ConnectionStrings.sh b/integration-tests/ConnectionStrings.sh
@@ -1,5 +1,6 @@
 #Update the sample connection strings with your specific connection settings
 declare -A connectionStrings=(
+  ["AzureSearch"]="Endpoint=myendpoint;ApiKey=myapikey"
   ["SqlServer"]="Data Source=.;Initial Catalog=benchmark;Integrated Security=True;"
   ["Postgres"]="Host=localhost;Port=5432;Database=benchmark;Username=postgres;Password=password;"
   ["PostgresJsonb"]="Host=localhost;Port=5432;Database=benchmark;Username=postgres;Password=password;"
diff --git a/integration-tests/Definitions/GeneratedSampleQueryWithArrays.json b/integration-tests/Definitions/GeneratedSampleQueryWithArrays.json
@@ -24,11 +24,6 @@
             "ColumnName": "ReferenceGuids", 
             "Operator": "Contains",
             "RandomizeValue": true 
-          },
-          {
-            "ColumnName": "Tags",
-            "Operator": "Equals",
-            "Value": ["One", "Two"]
           }
         ]
       },
diff --git a/integration-tests/Definitions/GeneratedSampleRawAzureSearchQuery.json b/integration-tests/Definitions/GeneratedSampleRawAzureSearchQuery.json
@@ -0,0 +1,3 @@
+{
+  "Filter": "UserId eq ${UserId}"
+} 
diff --git a/integration-tests/Definitions/GeneratedSampleTable.json b/integration-tests/Definitions/GeneratedSampleTable.json
@@ -1,15 +1,15 @@
 {
   "Name": "GeneratedSample",
   "Columns": [
-    { "Name": "Id", "Type": "Integer", "Nullable": false, "Queryable": false, "PartitionKey": true },
+    { "Name": "Id", "Type": "String", "Nullable": false, "Queryable": false, "PrimaryKey": true, "PartitionKey": true },
     { "Name": "CountryCode", "Type": "String", "Queryable": true },
     { "Name": "IsActive", "Type": "Boolean", "Queryable": true },
     { "Name": "CompanyName", "Type": "String", "Queryable": true },
     { "Name": "CreatedAt", "Type": "DateTime", "Queryable": true },
     { "Name": "Iban", "Type": "String", "Queryable": true },
     { "Name": "Rating", "Type": "Double", "Queryable": true },
     { "Name": "ObjectGuid", "Type": "Guid", "Queryable": true },
-    { "Name": "UserId", "Type": "Integer", "Queryable": true },
+    { "Name": "UserId", "Type": "String", "Queryable": true },
     { "Name": "Count", "Type": "Integer", "Queryable": true },
     { "Name": "IpAddress", "Type": "String", "Queryable": true },
     { "Name": "Color", "Type": "String", "Queryable": true },
diff --git a/integration-tests/Definitions/GeneratedSampleTableWithArrays.json b/integration-tests/Definitions/GeneratedSampleTableWithArrays.json
@@ -1,15 +1,15 @@
 {
   "Name": "GeneratedSample",
   "Columns": [
-    { "Name": "Id", "Type": "Integer", "Nullable": false, "Queryable": false, "PartitionKey": true },
+    { "Name": "Id", "Type": "String", "Nullable": false, "Queryable": false, "PrimaryKey": true, "PartitionKey": true },
     { "Name": "CountryCode", "Type": "String", "Queryable": true },
     { "Name": "IsActive", "Type": "Boolean", "Queryable": true },
     { "Name": "CompanyName", "Type": "String", "Queryable": true },
     { "Name": "CreatedAt", "Type": "DateTime", "Queryable": true },
     { "Name": "Iban", "Type": "String", "Queryable": true },
     { "Name": "Rating", "Type": "Double", "Queryable": true },
     { "Name": "ObjectGuid", "Type": "Guid", "Queryable": true },
-    { "Name": "UserId", "Type": "Integer", "Queryable": true },
+    { "Name": "UserId", "Type": "String", "Queryable": true },
     { "Name": "Count", "Type": "Integer", "Queryable": true },
     { "Name": "IpAddress", "Type": "String", "Queryable": true },
     { "Name": "Color", "Type": "String", "Queryable": true },
diff --git a/integration-tests/Definitions/GeneratedUsersTable.json b/integration-tests/Definitions/GeneratedUsersTable.json
@@ -1,7 +1,7 @@
 {
   "Name": "GeneratedUsers",
   "Columns": [
-    { "Name": "Id", "Type": "Integer", "Nullable": false, "Queryable": false, "PartitionKey": true },
+    { "Name": "Id", "Type": "String", "Nullable": false, "Queryable": false, "PrimaryKey": true, "PartitionKey": true },
     { "Name": "CreatedAt", "Type": "DateTime", "Queryable": true },
     { "Name": "FirstName", "Type": "String", "Queryable": true },
     { "Name": "LastName", "Type": "String", "Queryable": true }
diff --git a/integration-tests/InputFiles.ps1 b/integration-tests/InputFiles.ps1
@@ -93,4 +93,12 @@ $inputFiles = @{
     QueryFile = "GeneratedSampleDynamoDbQuery.json"
     RawQueryFile = $null
   }
+  AzureSearch = @{
+    TableFile = "GeneratedSampleTableWithArrays.json"
+    TableName = "GeneratedSample"
+    DataSourceFile = "GeneratedSampleDataSourceWithArrays.json"
+    QueryFile = "GeneratedSampleQueryWithArrays.json"
+    RawQueryFile = "GeneratedSampleRawAzureSearchQuery.json"
+    RawQueryParametersFile = "GeneratedSampleRawQueryParameters.json"
+  }
 }
diff --git a/integration-tests/InputFiles.sh b/integration-tests/InputFiles.sh
@@ -94,4 +94,12 @@ inputFiles[DynamoDb_TableName]=""
 inputFiles[DynamoDb_DataSourceFile]="GeneratedSampleDataSourceWithArrays.json"
 inputFiles[DynamoDb_QueryFile]="GeneratedSampleDynamoDbQuery.json"
 inputFiles[DynamoDb_RawQueryFile]=""
-inputFiles[DynamoDb_RawQueryParametersFile]=""
+inputFiles[DynamoDb_RawQueryParametersFile]=""
+
+# AzureSearch
+inputFiles[AzureSearch_TableFile]="GeneratedSampleTableWithArrays.json"
+inputFiles[AzureSearch_TableName]="GeneratedSample"
+inputFiles[AzureSearch_DataSourceFile]="GeneratedSampleDataSourceWithArrays.json"
+inputFiles[AzureSearch_QueryFile]="GeneratedSampleQueryWithArrays.json"
+inputFiles[AzureSearch_RawQueryFile]="GeneratedSampleRawAzureSearchQuery.json"
+inputFiles[AzureSearch_RawQueryParametersFile]="GeneratedSampleRawQueryParameters.json"

Original file line number	Diff line number	Diff line change
`@@ -24,11 +24,6 @@`
`24`	`24`	`"ColumnName": "ReferenceGuids",`
`25`	`25`	`"Operator": "Contains",`
`26`	`26`	`"RandomizeValue": true`
`27`		`- },`
`28`		`- {`
`29`		`- "ColumnName": "Tags",`
`30`		`- "Operator": "Equals",`
`31`		`- "Value": ["One", "Two"]`
`32`	`27`	`}`
`33`	`28`	`]`
`34`	`29`	`},`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+{`
	`2`	`+ "Filter": "UserId eq ${UserId}"`
	`3`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"Name": "GeneratedUsers",`
`3`	`3`	`"Columns": [`
`4`		`- { "Name": "Id", "Type": "Integer", "Nullable": false, "Queryable": false, "PartitionKey": true },`
	`4`	`+ { "Name": "Id", "Type": "String", "Nullable": false, "Queryable": false, "PrimaryKey": true, "PartitionKey": true },`
`5`	`5`	`{ "Name": "CreatedAt", "Type": "DateTime", "Queryable": true },`
`6`	`6`	`{ "Name": "FirstName", "Type": "String", "Queryable": true },`
`7`	`7`	`{ "Name": "LastName", "Type": "String", "Queryable": true }`
Original file line number	Diff line number	Diff line change
`@@ -93,4 +93,12 @@ $inputFiles = @{`
`93`	`93`	`QueryFile = "GeneratedSampleDynamoDbQuery.json"`
`94`	`94`	`RawQueryFile = $null`
`95`	`95`	`}`
	`96`	`+ AzureSearch = @{`
	`97`	`+ TableFile = "GeneratedSampleTableWithArrays.json"`
	`98`	`+ TableName = "GeneratedSample"`
	`99`	`+ DataSourceFile = "GeneratedSampleDataSourceWithArrays.json"`
	`100`	`+ QueryFile = "GeneratedSampleQueryWithArrays.json"`
	`101`	`+ RawQueryFile = "GeneratedSampleRawAzureSearchQuery.json"`
	`102`	`+ RawQueryParametersFile = "GeneratedSampleRawQueryParameters.json"`
	`103`	`+ }`
`96`	`104`	`}`