Add more coverage to impl

Signed-off-by: Derek Ho <dxho@amazon.com>
derek-ho · Oct 13, 2023 · 970b5bf · 970b5bf
1 parent af4c924
commit 970b5bf
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...urces/src/test/java/org/opensearch/sql/datasources/service/DataSourceServiceImplTest.java b/...urces/src/test/java/org/opensearch/sql/datasources/service/DataSourceServiceImplTest.java
@@ -290,6 +290,19 @@ void testUpdateDefaultDataSource() {
         unsupportedOperationException.getMessage());
   }
 
+  @Test
+  void testPatchDefaultDataSource() {
+    Map<String, Object> dataSourceData =
+        Map.of(NAME_FIELD, DEFAULT_DATASOURCE_NAME, DESCRIPTION_FIELD, "test");
+    UnsupportedOperationException unsupportedOperationException =
+        assertThrows(
+            UnsupportedOperationException.class,
+            () -> dataSourceService.patchDataSource(dataSourceData));
+    assertEquals(
+        "Not allowed to update default datasource :" + DEFAULT_DATASOURCE_NAME,
+        unsupportedOperationException.getMessage());
+  }
+
   @Test
   void testPatchDataSourceSuccessCase() {
     // Tests that patch underlying implementation is to call update