xuefuz commented on a change in pull request #8212: [FLINK-11519][table] Add 
function related catalog APIs
URL: https://github.com/apache/flink/pull/8212#discussion_r278722515
 
 

 ##########
 File path: 
flink-table/flink-table-api-java/src/test/java/org/apache/flink/table/catalog/GenericInMemoryCatalogTest.java
 ##########
 @@ -562,6 +568,158 @@ public void testRenameView() throws Exception {
                catalog.dropTable(viewPath2, false);
        }
 
+       // ------ functions ------
+
+       @Test
+       public void testCreateFunction() throws Exception {
+               catalog.createDatabase(db1, createDb(), false);
+
+               assertFalse(catalog.functionExists(path1));
+
+               catalog.createFunction(path1, createFunction(), false);
+
+               assertTrue(catalog.functionExists(path1));
+
+               catalog.dropFunction(path1, false);
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testCreateFunction_DatabaseNotExistException() throws 
Exception {
+               assertFalse(catalog.databaseExists(db1));
+
+               exception.expect(DatabaseNotExistException.class);
+               exception.expectMessage("Database db1 does not exist in 
Catalog");
+               catalog.createFunction(path1, createFunction(), false);
+       }
+
+       @Test
+       public void testCreateFunction_FunctionAlreadyExistException() throws 
Exception {
+               catalog.createDatabase(db1, createDb(), false);
+               catalog.createFunction(path1, createFunction(), false);
+
+               exception.expect(FunctionAlreadyExistException.class);
+               exception.expectMessage("Function db1.t1 already exists in 
Catalog");
+               catalog.createFunction(path1, createFunction(), false);
+       }
+
+       @Test
+       public void testCreateFunction_FunctionAlreadyExist_ignored() throws 
Exception {
+               catalog.createDatabase(db1, createDb(), false);
+
+               CatalogFunction func = createFunction();
+               catalog.createFunction(path1, func, false);
+
+               CatalogTestUtil.checkEquals(func, catalog.getFunction(path1));
+
+               catalog.createFunction(path1, createAnotherFunction(), true);
+
+               CatalogTestUtil.checkEquals(func, catalog.getFunction(path1));
+
+               catalog.dropFunction(path1, false);
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testAlterFunction() throws Exception {
+               catalog.createDatabase(db1, createDb(), false);
+
+               CatalogFunction func = createFunction();
+               catalog.createFunction(path1, func, false);
+
+               CatalogTestUtil.checkEquals(func, catalog.getFunction(path1));
+
+               CatalogFunction newFunc = createAnotherFunction();
+               catalog.alterFunction(path1, newFunc, false);
+
+               assertNotEquals(func, catalog.getFunction(path1));
+               CatalogTestUtil.checkEquals(newFunc, 
catalog.getFunction(path1));
+
+               catalog.dropFunction(path1, false);
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testAlterFunction_FunctionNotExistException() throws 
Exception {
+               exception.expect(FunctionNotExistException.class);
+               exception.expectMessage("Function db1.nonexist does not exist 
in Catalog");
+               catalog.alterFunction(nonExistObjectPath, createFunction(), 
false);
+       }
+
+       @Test
+       public void testAlterFunction_FunctionNotExist_ignored() throws 
Exception {
+               catalog.createDatabase(db1, createDb(), false);
+               catalog.alterFunction(nonExistObjectPath, createFunction(), 
true);
+
+               assertFalse(catalog.functionExists(nonExistObjectPath));
+
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testListFunctions() throws Exception {
+               catalog.createDatabase(db1, createDb(), false);
+
+               CatalogFunction func = createFunction();
+               catalog.createFunction(path1, func, false);
+
+               assertEquals(path1.getObjectName(), 
catalog.listFunctions(db1).get(0));
+
+               catalog.dropFunction(path1, false);
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testListFunctions_DatabaseNotExistException() throws 
Exception{
+               exception.expect(DatabaseNotExistException.class);
+               exception.expectMessage("Database db1 does not exist in 
Catalog");
+               catalog.listFunctions(db1);
+       }
+
+       @Test
+       public void testGetFunction_FunctionNotExistException() throws 
Exception {
+               catalog.createDatabase(db1, createDb(), false);
+
+               exception.expect(FunctionNotExistException.class);
+               exception.expectMessage("Function db1.nonexist does not exist 
in Catalog");
+               catalog.getFunction(nonExistObjectPath);
+       }
+
+       @Test
+       public void testGetFunction_FunctionNotExistException_NoDb() throws 
Exception {
+               exception.expect(FunctionNotExistException.class);
+               exception.expectMessage("Function db1.nonexist does not exist 
in Catalog");
+               catalog.getFunction(nonExistObjectPath);
+       }
+
+       @Test
+       public void testDropFunction() throws Exception {
+               catalog.createDatabase(db1, createDb(), false);
+               catalog.createFunction(path1, createFunction(), false);
+
+               assertTrue(catalog.functionExists(path1));
+
+               catalog.dropFunction(path1, false);
+
+               assertFalse(catalog.functionExists(path1));
+
+               catalog.dropDatabase(db1, false);
+       }
+
+       @Test
+       public void testDropFunction_FunctionNotExistException() throws 
Exception {
+               exception.expect(FunctionNotExistException.class);
+               exception.expectMessage("Function non.exist does not exist in 
Catalog");
 
 Review comment:
   Same as above.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


With regards,
Apache Git Services

Reply via email to