5
5
#include < ydb/public/sdk/cpp/client/ydb_proto/accessor.h>
6
6
#include < ydb/public/sdk/cpp/client/ydb_scheme/scheme.h>
7
7
#include < ydb/public/sdk/cpp/client/ydb_topic/topic.h>
8
- #include < ydb/public/sdk/cpp/client/draft/ydb_long_tx.h>
9
8
#include < ydb/core/testlib/cs_helper.h>
10
9
#include < ydb/core/testlib/common_helper.h>
11
10
#include < ydb/core/formats/arrow/serializer/full.h>
@@ -5378,7 +5377,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5378
5377
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5379
5378
tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5380
5379
tableInserter.AddRow ().Add (2 ).Add (" test_res_2" ).Add (123 );
5381
- testHelper.InsertData (testTable, tableInserter);
5380
+ testHelper.BulkUpsert (testTable, tableInserter);
5382
5381
}
5383
5382
5384
5383
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` WHERE id=1" , " [[1;#;[\" test_res_1\" ]]]" );
@@ -5407,7 +5406,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5407
5406
{
5408
5407
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5409
5408
tableInserter.AddRow ().Add (3 ).Add (" test_res_3" ).Add (123 ).Add <uint64_t >(200 );
5410
- testHelper.InsertData (testTable, tableInserter);
5409
+ testHelper.BulkUpsert (testTable, tableInserter);
5411
5410
}
5412
5411
5413
5412
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` WHERE id=3" , " [[3;[123];[200u];[\" test_res_3\" ]]]" );
@@ -5444,7 +5443,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5444
5443
}
5445
5444
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` WHERE id=1" , " [[1;#;#;[\" test_res_1\" ]]]" );
5446
5445
}
5447
-
5446
+ /*
5448
5447
Y_UNIT_TEST(AddColumnOnSchemeChange) {
5449
5448
TKikimrSettings runnerSettings;
5450
5449
runnerSettings.WithSampleTables = false;
@@ -5471,7 +5470,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5471
5470
}
5472
5471
testHelper.ReadData("SELECT * FROM `/Root/ColumnTableTest` WHERE id=1", "[[1;#;#;[\"test_res_1\"]]]");
5473
5472
}
5474
-
5473
+ */
5475
5474
Y_UNIT_TEST (AddColumnWithStore) {
5476
5475
TKikimrSettings runnerSettings;
5477
5476
runnerSettings.WithSampleTables = false ;
@@ -5494,7 +5493,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5494
5493
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5495
5494
tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5496
5495
tableInserter.AddRow ().Add (2 ).Add (" test_res_2" ).Add (123 );
5497
- testHelper.InsertData (testTable, tableInserter);
5496
+ testHelper.BulkUpsert (testTable, tableInserter);
5498
5497
}
5499
5498
5500
5499
testHelper.ReadData (" SELECT * FROM `/Root/TableStoreTest/ColumnTableTest` WHERE id=1" , " [[1;#;[\" test_res_1\" ]]]" );
@@ -5524,7 +5523,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5524
5523
{
5525
5524
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5526
5525
tableInserter.AddRow ().Add (3 ).Add (" test_res_3" ).Add (123 ).Add <uint64_t >(200 );
5527
- testHelper.InsertData (testTable, tableInserter);
5526
+ testHelper.BulkUpsert (testTable, tableInserter);
5528
5527
}
5529
5528
5530
5529
testHelper.ReadData (" SELECT * FROM `/Root/TableStoreTest/ColumnTableTest` WHERE id=3" , " [[3;[123];[200u];[\" test_res_3\" ]]]" );
@@ -5580,7 +5579,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5580
5579
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schemaWithNull));
5581
5580
tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5582
5581
tableInserter.AddRow ().Add (2 ).Add (" test_res_2" ).Add (123 );
5583
- testHelper.InsertData (testTable, tableInserter, {}, EStatus ::GENERIC_ERROR);
5582
+ testHelper.BulkUpsert (testTable, tableInserter, Ydb::StatusIds ::GENERIC_ERROR);
5584
5583
}
5585
5584
{
5586
5585
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schemaWithNull));
@@ -5611,7 +5610,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5611
5610
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5612
5611
tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5613
5612
tableInserter.AddRow ().Add (2 ).Add (" test_res_2" ).Add (123 );
5614
- testHelper.InsertData (testTable, tableInserter);
5613
+ testHelper.BulkUpsert (testTable, tableInserter);
5615
5614
}
5616
5615
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` WHERE id=1" , " [[1;#;[\" test_res_1\" ]]]" );
5617
5616
{
@@ -5628,7 +5627,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5628
5627
}
5629
5628
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` " , " [[1];[2]]" );
5630
5629
}
5631
-
5630
+ /*
5632
5631
Y_UNIT_TEST(DropColumnOnSchemeChange) {
5633
5632
TKikimrSettings runnerSettings;
5634
5633
runnerSettings.WithSampleTables = false;
@@ -5654,34 +5653,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5654
5653
}
5655
5654
testHelper.ReadData("SELECT * FROM `/Root/ColumnTableTest` WHERE id=1", "[[1;#]]");
5656
5655
}
5657
-
5658
- Y_UNIT_TEST (DropColumnOldScheme) {
5659
- TKikimrSettings runnerSettings;
5660
- runnerSettings.WithSampleTables = false ;
5661
- TTestHelper testHelper (runnerSettings);
5662
-
5663
- TVector<TTestHelper::TColumnSchema> schema = {
5664
- TTestHelper::TColumnSchema ().SetName (" id" ).SetType (NScheme::NTypeIds::Int32).SetNullable (false ),
5665
- TTestHelper::TColumnSchema ().SetName (" resource_id" ).SetType (NScheme::NTypeIds::Utf8),
5666
- TTestHelper::TColumnSchema ().SetName (" level" ).SetType (NScheme::NTypeIds::Int32)
5667
- };
5668
-
5669
- TTestHelper::TColumnTable testTable;
5670
-
5671
- testTable.SetName (" /Root/ColumnTableTest" ).SetPrimaryKey ({" id" }).SetSharding ({" id" }).SetSchema (schema);
5672
- testHelper.CreateTable (testTable);
5673
- {
5674
- auto alterQuery = TStringBuilder () << " ALTER TABLE `" << testTable.GetName () << " `DROP COLUMN resource_id;" ;
5675
- auto alterResult = testHelper.GetSession ().ExecuteSchemeQuery (alterQuery).GetValueSync ();
5676
- UNIT_ASSERT_VALUES_EQUAL_C (alterResult.GetStatus (), EStatus::SUCCESS, alterResult.GetIssues ().ToString ());
5677
- }
5678
- {
5679
- TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5680
- tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5681
- testHelper.InsertData (testTable, tableInserter, {}, EStatus::SUCCESS);
5682
- }
5683
- // testHelper.ReadData("SELECT * FROM `/Root/ColumnTableTest` WHERE id=1", "[[1;#]]");
5684
- }
5656
+ */
5685
5657
5686
5658
Y_UNIT_TEST (DropColumnOldSchemeBulkUpsert) {
5687
5659
TKikimrSettings runnerSettings;
@@ -5729,7 +5701,7 @@ Y_UNIT_TEST_SUITE(KqpOlapScheme) {
5729
5701
TTestHelper::TUpdatesBuilder tableInserter (testTable.GetArrowSchema (schema));
5730
5702
tableInserter.AddRow ().Add (1 ).Add (" test_res_1" ).AddNull ();
5731
5703
tableInserter.AddRow ().Add (2 ).Add (" test_res_2" ).Add (123 );
5732
- testHelper.InsertData (testTable, tableInserter);
5704
+ testHelper.BulkUpsert (testTable, tableInserter);
5733
5705
}
5734
5706
testHelper.ReadData (" SELECT * FROM `/Root/ColumnTableTest` WHERE id=1" , " [[1;#;[\" test_res_1\" ]]]" );
5735
5707
{
0 commit comments