feat: add create method in CatalogSource and PushSource (LENS-874)

src/main/java/com/coveo/pushapiclient/CatalogSource.java

@@ -1,5 +1,6 @@
 package com.coveo.pushapiclient;
 
+import java.io.IOException;
 import java.net.MalformedURLException;
 import java.net.URL;
 
@@ -8,6 +9,23 @@ class CatalogSource implements StreamEnabledSource {
     private final String apiKey;
     private final ApiUrl urlExtractor;
 
+
+    /**
+     * Creates a Catalog Source in Coveo Org
+     *
+     * @param platformUrl
+     * @param organizationId
+     * @param apiKey
+     * @param name
+     * @param sourceVisibility
+     * @return
+     * @throws IOException
+     * @throws InterruptedException
+     */
+    public static void create(PlatformUrl platformUrl, String organizationId, String apiKey, String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
+        new PlatformClient(apiKey,organizationId,platformUrl).createSource(name, SourceType.CATALOG.name(), true, true, sourceVisibility);
+    }
+
     /**
      * Create a Catalog source instance from its
      * <a href="https://docs.coveo.com/en/3295#stream-api-url">Stream API URL</a>

src/main/java/com/coveo/pushapiclient/PlatformClient.java

@@ -82,17 +82,21 @@ public PlatformClient(String apiKey, String organizationId, Environment environm
      * Create a new push source
      *
      * @param name             The name of the source to create
+     * @param sourceType
+     * @param isPushEnabled
+     * @param isStreamEnabled
      * @param sourceVisibility The security option that should be applied to the content of the source. See [Content Security](https://docs.coveo.com/en/1779).
      * @return
      * @throws IOException
      * @throws InterruptedException
      */
-    public HttpResponse<String> createSource(String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
+    public HttpResponse<String> createSource(String name, final String sourceType, final boolean isPushEnabled, final boolean isStreamEnabled, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
         String[] headers = this.getHeaders(this.getAuthorizationHeader(), this.getContentTypeApplicationJSONHeader());
 
         String json = this.toJSON(new HashMap<>() {{
-            put("sourceType", "PUSH");
-            put("pushEnabled", true);
+            put("sourceType", sourceType);
+            put("pushEnabled", isPushEnabled);
+            put("streamEnabled", isStreamEnabled);
             put("name", name);
             put("sourceVisibility", sourceVisibility);
         }});

src/main/java/com/coveo/pushapiclient/PushSource.java

@@ -33,6 +33,22 @@ public String getApiKey() {
         return this.apiKey;
     }
 
+    /**
+     * Creates a push Source in Coveo Org
+     *
+     * @param platformUrl
+     * @param organizationId
+     * @param apiKey
+     * @param name
+     * @param sourceVisibility
+     * @return
+     * @throws IOException
+     * @throws InterruptedException
+     */
+    public static void create(PlatformUrl platformUrl, String organizationId, String apiKey, String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
+        new PlatformClient(apiKey,organizationId,platformUrl).createSource(name, SourceType.PUSH.name(), true, false, sourceVisibility);
+    }
+
     /**
      * Create a Push source instance from its
      * <a href="https://docs.coveo.com/en/1546#push-api-url">Push API URL</a>
@@ -311,4 +327,5 @@ public HttpResponse<String> deleteDocument(String documentId, Boolean deleteChil
         return this.platformClient.deleteDocument(this.getId(), documentId, deleteChildren);
     }
 
+
 }

src/main/java/com/coveo/pushapiclient/Source.java

@@ -53,7 +53,7 @@ public Source(String apiKey, String organizationId, Environment environment) {
      * @throws InterruptedException
      */
     public HttpResponse<String> create(String name, SourceVisibility sourceVisibility) throws IOException, InterruptedException {
-        return this.platformClient.createSource(name, sourceVisibility);
+        return this.platformClient.createSource(name, SourceType.PUSH.name(), true, false, sourceVisibility);
     }
 
     /**

src/main/java/com/coveo/pushapiclient/SourceType.java

@@ -0,0 +1,6 @@
+package com.coveo.pushapiclient;
+
+public enum SourceType {
+    PUSH,
+    CATALOG,
+}

src/test/java/com/coveo/pushapiclient/PlatformClientTest.java

@@ -108,8 +108,8 @@ public void setupClient() {
     }
 
     @Test
-    public void testCreateSource() throws IOException, InterruptedException {
-        client.createSource("the_name", SourceVisibility.SECURED);
+    public void testCreatePushSource() throws IOException, InterruptedException {
+        client.createSource("the_name", SourceType.PUSH.name(), true, false, SourceVisibility.SECURED);
         verify(httpClient).send(argument.capture(), any(HttpResponse.BodyHandlers.ofString().getClass()));
 
         assertEquals("POST", argument.getValue().method());
@@ -124,6 +124,24 @@ public void testCreateSource() throws IOException, InterruptedException {
         assertEquals(true, requestBody.get("pushEnabled"));
     }
 
+    @Test
+    public void testCreateCatalogSource() throws IOException, InterruptedException {
+        client.createSource("the_name", SourceType.CATALOG.name(), true, true, SourceVisibility.SECURED);
+        verify(httpClient).send(argument.capture(), any(HttpResponse.BodyHandlers.ofString().getClass()));
+
+        assertEquals("POST", argument.getValue().method());
+        assertTrue(argument.getValue().uri().getPath().contains("the_org_id/sources"));
+        assertAuthorizationHeader();
+        assertApplicationJsonHeader();
+
+        Map requestBody = StringSubscriber.toMap(argument.getValue().bodyPublisher());
+        assertEquals("the_name", requestBody.get("name"));
+        assertEquals(SourceVisibility.SECURED.toString(), requestBody.get("sourceVisibility"));
+        assertEquals("CATALOG", requestBody.get("sourceType"));
+        assertEquals(true, requestBody.get("pushEnabled"));
+        assertEquals(true, requestBody.get("streamEnabled"));
+    }
+
     @Test
     public void testCreateOrUpdateSecurityIdentity() throws IOException, InterruptedException {
         client.createOrUpdateSecurityIdentity("my_provider", securityIdentityModel());