fix(data-masking): add unit test case for select sql syntax (oceanbas…

…e#398)

* add: unit test

* fix: unit test

server/odc-service/src/test/java/com/oceanbase/odc/service/datasecurity/extractor/OBColumnExtractorTest.java

@@ -329,6 +329,36 @@ public void test_extract_OBMySQL_fromMultiNestingSubquery() {
  Assert.assertEquals(expectLabels, actualLabels);
  }
 
+ @Test
+ public void test_extract_OBMySQL_whereSingleCondition() {
+ String sql = TestColumnExtractorUtil.getTestSql(DialectType.OB_MYSQL, "where-single-condition");
+ LogicalTable result = obMySQLColumnExtractor.extract(obMySQLParser.parse(new StringReader(sql)));
+ List<String> actualLabels = getResultColumnLabels(result);
+ List<String> expectLabels = Arrays.asList("id", "name", "birthday", "description");
+ Assert.assertEquals(4, result.getColumnList().size());
+ Assert.assertEquals(expectLabels, actualLabels);
+ }
+
+ @Test
+ public void test_extract_OBMySQL_whereMultiCondition() {
+ String sql = TestColumnExtractorUtil.getTestSql(DialectType.OB_MYSQL, "where-multi-condition");
+ LogicalTable result = obMySQLColumnExtractor.extract(obMySQLParser.parse(new StringReader(sql)));
+ List<String> actualLabels = getResultColumnLabels(result);
+ List<String> expectLabels = Arrays.asList("id", "name", "birthday", "description");
+ Assert.assertEquals(4, result.getColumnList().size());
+ Assert.assertEquals(expectLabels, actualLabels);
+ }
+
+ @Test
+ public void test_extract_OBMySQL_whereNestingSelect() {
+ String sql = TestColumnExtractorUtil.getTestSql(DialectType.OB_MYSQL, "where-nesting-select");
+ LogicalTable result = obMySQLColumnExtractor.extract(obMySQLParser.parse(new StringReader(sql)));
+ List<String> actualLabels = getResultColumnLabels(result);
+ List<String> expectLabels = Arrays.asList("id", "name", "birthday", "description");
+ Assert.assertEquals(4, result.getColumnList().size());
+ Assert.assertEquals(expectLabels, actualLabels);
+ }
+
  @Test
  public void test_extract_OBMySQL_unionTwoTables() {
  String sql = TestColumnExtractorUtil.getTestSql(DialectType.OB_MYSQL, "union-two-tables");

server/odc-service/src/test/resources/datasecurity/test_sensitive_column_extractor_sql.yaml

@@ -237,15 +237,15 @@ mysql:
  FROM
  test_data_masking_1 t1
  WHERE
- t.id = (
+ t1.id = (
  SELECT
  id
  FROM
  test_data_masking_2
  LIMIT
  1
  )
- OR t.name IN (
+ OR t1.name IN (
  SELECT
  name
  FROM