forked from apache/arrow-adbc
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
31 changed files
with
1,977 additions
and
1,621 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
154 changes: 154 additions & 0 deletions
154
csharp/test/Drivers/Apache/Common/BinaryBooleanValueTests.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,154 @@ | ||
| /* | ||
| * Licensed to the Apache Software Foundation (ASF) under one or more | ||
| * contributor license agreements. See the NOTICE file distributed with | ||
| * this work for additional information regarding copyright ownership. | ||
| * The ASF licenses this file to You under the Apache License, Version 2.0 | ||
| * (the "License"); you may not use this file except in compliance with | ||
| * the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
|
|
||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.Globalization; | ||
| using System.Threading.Tasks; | ||
| using Apache.Arrow.Adbc.Tests.Drivers.Apache.Hive2; | ||
| using Xunit; | ||
| using Xunit.Abstractions; | ||
|
|
||
| namespace Apache.Arrow.Adbc.Tests.Drivers.Apache.Common | ||
| { | ||
| // TODO: When supported, use prepared statements instead of SQL string literals | ||
| // Which will better test how the driver handles values sent/received | ||
|
|
||
| /// <summary> | ||
| /// Validates that specific binary and boolean values can be inserted, retrieved and targeted correctly | ||
| /// </summary> | ||
| public abstract class BinaryBooleanValueTests<TConfig, TEnv> : TestBase<TConfig, TEnv> | ||
| where TConfig : TestConfiguration | ||
| where TEnv : HiveServer2TestEnvironment<TConfig> | ||
| { | ||
| public BinaryBooleanValueTests(ITestOutputHelper output, TestEnvironment<TConfig>.Factory<TEnv> testEnvFactory) | ||
| : base(output, testEnvFactory) { } | ||
|
|
||
| public static IEnumerable<object[]> ByteArrayData(int size) | ||
| { | ||
| var rnd = new Random(); | ||
| byte[] bytes = new byte[size]; | ||
| rnd.NextBytes(bytes); | ||
| yield return new object[] { bytes }; | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Validates if driver can send and receive specific Binary values correctly. | ||
| /// </summary> | ||
| [SkippableTheory] | ||
| [InlineData(null)] | ||
| [MemberData(nameof(ByteArrayData), 0)] | ||
| [MemberData(nameof(ByteArrayData), 2)] | ||
| [MemberData(nameof(ByteArrayData), 1024)] | ||
| public async Task TestBinaryData(byte[]? value) | ||
| { | ||
| string columnName = "BINARYTYPE"; | ||
| using TemporaryTable table = await NewTemporaryTableAsync(Statement, string.Format("{0} {1}", columnName, "BINARY")); | ||
| string? formattedValue = value != null ? $"X'{BitConverter.ToString(value).Replace("-", "")}'" : null; | ||
| await ValidateInsertSelectDeleteSingleValueAsync( | ||
| table.TableName, | ||
| columnName, | ||
| value, | ||
| formattedValue); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Validates if driver can send and receive specific Boolean values correctly. | ||
| /// </summary> | ||
| [SkippableTheory] | ||
| [InlineData(null)] | ||
| [InlineData(true)] | ||
| [InlineData(false)] | ||
| public async Task TestBooleanData(bool? value) | ||
| { | ||
| string columnName = "BOOLEANTYPE"; | ||
| using TemporaryTable table = await NewTemporaryTableAsync(Statement, string.Format("{0} {1}", columnName, "BOOLEAN")); | ||
| string? formattedValue = value == null ? null : $"{value?.ToString(CultureInfo.InvariantCulture)}"; | ||
| await ValidateInsertSelectDeleteTwoValuesAsync( | ||
| table.TableName, | ||
| columnName, | ||
| value, | ||
| formattedValue); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Validates if driver can receive specific NULL values correctly. | ||
| /// </summary> | ||
| [SkippableTheory] | ||
| [InlineData("NULL")] | ||
| [InlineData("CAST(NULL AS INT)")] | ||
| [InlineData("CAST(NULL AS BIGINT)")] | ||
| [InlineData("CAST(NULL AS SMALLINT)")] | ||
| [InlineData("CAST(NULL AS TINYINT)")] | ||
| [InlineData("CAST(NULL AS FLOAT)")] | ||
| [InlineData("CAST(NULL AS DOUBLE)")] | ||
| [InlineData("CAST(NULL AS DECIMAL(38,0))")] | ||
| [InlineData("CAST(NULL AS STRING)")] | ||
| [InlineData("CAST(NULL AS VARCHAR(10))")] | ||
| [InlineData("CAST(NULL AS CHAR(10))")] | ||
| [InlineData("CAST(NULL AS BOOLEAN)")] | ||
| [InlineData("CAST(NULL AS BINARY)")] | ||
| [InlineData("CAST(NULL AS MAP<STRING, INT>)")] | ||
| [InlineData("CAST(NULL AS STRUCT<NAME: STRING>)")] | ||
| [InlineData("CAST(NULL AS ARRAY<INT>)")] | ||
| public async Task TestNullData(string projectionClause) | ||
| { | ||
| string selectStatement = $"SELECT {projectionClause};"; | ||
| // Note: by default, this returns as String type, not NULL type. | ||
| await SelectAndValidateValuesAsync(selectStatement, (object?)null, 1); | ||
| } | ||
|
|
||
| [SkippableTheory] | ||
| [InlineData(1)] | ||
| [InlineData(7)] | ||
| [InlineData(8)] | ||
| [InlineData(9)] | ||
| [InlineData(15)] | ||
| [InlineData(16)] | ||
| [InlineData(17)] | ||
| [InlineData(23)] | ||
| [InlineData(24)] | ||
| [InlineData(25)] | ||
| [InlineData(31)] | ||
| [InlineData(32)] // Full integer | ||
| [InlineData(33)] | ||
| [InlineData(39)] | ||
| [InlineData(40)] | ||
| [InlineData(41)] | ||
| [InlineData(47)] | ||
| [InlineData(48)] | ||
| [InlineData(49)] | ||
| [InlineData(63)] | ||
| [InlineData(64)] // Full 2 integers | ||
| [InlineData(65)] | ||
| public async Task TestMultilineNullData(int numberOfValues) | ||
| { | ||
| Random rnd = new(); | ||
| int percentIsNull = 50; | ||
|
|
||
| object?[] values = new object?[numberOfValues]; | ||
| for (int i = 0; i < numberOfValues; i++) | ||
| { | ||
| values[i] = rnd.Next(0, 100) < percentIsNull ? null : rnd.Next(0, 2) != 0; | ||
| } | ||
| string columnName = "BOOLEANTYPE"; | ||
| string indexColumnName = "INDEXCOL"; | ||
| using TemporaryTable table = await NewTemporaryTableAsync(Statement, string.Format("{0} {1}, {2} {3}", indexColumnName, "INT", columnName, "BOOLEAN")); | ||
| await ValidateInsertSelectDeleteMultipleValuesAsync(table.TableName, columnName, indexColumnName, values); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.