Added unit tests for performance video endpoints

taboola · Nov 14, 2023 · 0e8903c · 0e8903c
1 parent 3f1acb0
commit 0e8903c
Show file tree

Hide file tree

Showing 2 changed files with 107 additions and 0 deletions.
diff --git a/src/test/java/com/taboola/backstage/BackstageTestBase.java b/src/test/java/com/taboola/backstage/BackstageTestBase.java
@@ -33,6 +33,10 @@
 import com.taboola.backstage.model.media.campaigns.SharedBudgetOperation;
 import com.taboola.backstage.model.media.campaigns.items.CampaignItem;
 import com.taboola.backstage.model.media.campaigns.items.CampaignItemOperation;
+import com.taboola.backstage.model.media.campaigns.items.CampaignPerformanceVideoItem;
+import com.taboola.backstage.model.media.campaigns.items.CampaignPerformanceVideoItemOperation;
+import com.taboola.backstage.model.media.campaigns.items.PerformanceVideoBulkCreateOperation;
+import com.taboola.backstage.model.media.campaigns.items.PerformanceVideoBulkUpdateOperation;
 import com.taboola.backstage.model.media.campaigns.targeting.PostalTargeting;
 
 import uk.co.jemos.podam.api.PodamFactory;
@@ -118,6 +122,22 @@ protected CampaignItem generateDummyCampaignItem() {
         return factory.manufacturePojo(CampaignItem.class);
     }
 
+    protected CampaignPerformanceVideoItem generateDummyCampaignPerformanceVideoItem() {
+        return factory.manufacturePojo(CampaignPerformanceVideoItem.class);
+    }
+
+    protected CampaignPerformanceVideoItemOperation generateDummyCampaignPerformanceVideoItemOperation() {
+        return factory.manufacturePojo(CampaignPerformanceVideoItemOperation.class);
+    }
+
+    protected PerformanceVideoBulkCreateOperation generateDummyPerformanceVideoBulkCreateOperation() {
+        return factory.manufacturePojo(PerformanceVideoBulkCreateOperation.class);
+    }
+
+    protected PerformanceVideoBulkUpdateOperation generateDummyPerformanceVideoBulkUpdateOperation() {
+        return factory.manufacturePojo(PerformanceVideoBulkUpdateOperation.class);
+    }
+
     protected CampaignItemOperation generateDummyCampaignItemOperation() {
         CampaignItemOperation campaignItemOperation = factory.manufacturePojo(CampaignItemOperation.class);
         campaignItemOperation.setUrl(null);

diff --git a/src/test/java/com/taboola/backstage/services/CampaignItemsServiceImplTest.java b/src/test/java/com/taboola/backstage/services/CampaignItemsServiceImplTest.java
@@ -11,6 +11,10 @@
 import org.junit.Before;
 import org.junit.Test;
 import com.taboola.backstage.BackstageTestBase;
+import com.taboola.backstage.model.media.campaigns.items.CampaignPerformanceVideoItem;
+import com.taboola.backstage.model.media.campaigns.items.CampaignPerformanceVideoItemOperation;
+import com.taboola.backstage.model.media.campaigns.items.PerformanceVideoBulkCreateOperation;
+import com.taboola.backstage.model.media.campaigns.items.PerformanceVideoBulkUpdateOperation;
 
 import java.util.Collections;
 
@@ -251,4 +255,87 @@ public void testDeleteItem() {
         assertEquals("Invalid campaign item", campaignItem, actual);
         verify(endpointMock, times(1)).deleteItem(any(), any(), any(), any());
     }
+
+    @Test
+    public void testReadPerformanceVideoItems() {
+        CampaignPerformanceVideoItem campaignItem = generateDummyCampaignPerformanceVideoItem();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        Results<CampaignPerformanceVideoItem> expectedResult = new Results<>(Collections.singletonList(campaignItem));
+        when(endpointMock.getVideoCreatives(auth.getToken().getAccessTokenForHeader(),"accountId", "1")).thenReturn(expectedResult);
+
+        Results<CampaignPerformanceVideoItem> actual = testInstance.readPerformanceVideoItems(auth, "accountId", "1");
+        assertEquals("Invalid return value", expectedResult, actual);
+        assertEquals("Invalid results", expectedResult.getResults(), actual.getResults());
+        verify(endpointMock, times(1)).getVideoCreatives(any(), any(), any());
+    }
+
+    @Test
+    public void testReadPerformanceVideoItem() {
+        CampaignPerformanceVideoItem campaignItem = generateDummyCampaignPerformanceVideoItem();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        when(endpointMock.getVideoCreative(auth.getToken().getAccessTokenForHeader(),"accountId", "1", "2")).thenReturn(campaignItem);
+
+        CampaignPerformanceVideoItem actual = testInstance.readPerformanceVideoItem(auth, "accountId", "1", "2");
+        assertEquals("Invalid return value", campaignItem, actual);
+        verify(endpointMock, times(1)).getVideoCreative(any(), any(), any(), any());
+    }
+
+    @Test
+    public void testCreatePerformanceVideoItem() {
+        CampaignPerformanceVideoItemOperation campaignItem = generateDummyCampaignPerformanceVideoItemOperation();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        when(endpointMock.insertVideoCreative(auth.getToken().getAccessTokenForHeader(),"accountId", "1", campaignItem)).thenReturn(campaignItem);
+
+        CampaignPerformanceVideoItem actual = testInstance.createPerformanceVideoItem(auth, "accountId", "1", campaignItem);
+        assertEquals("Invalid return value", campaignItem, actual);
+        verify(endpointMock, times(1)).insertVideoCreative(any(), any(), any(), any());
+    }
+
+    @Test
+    public void testUpdatePerformanceVideoItem() {
+        CampaignPerformanceVideoItemOperation campaignItem = generateDummyCampaignPerformanceVideoItemOperation();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        when(endpointMock.updateVideoCreative(auth.getToken().getAccessTokenForHeader(),"accountId", "1", "2", false, campaignItem)).thenReturn(campaignItem);
+
+        CampaignPerformanceVideoItem actual = testInstance.updatePerformanceVideoItem(auth, "accountId", "1", "2", campaignItem);
+        assertEquals("Invalid return value", campaignItem, actual);
+        verify(endpointMock, times(1)).updateVideoCreative(any(), any(), any(), any(), anyBoolean(), any());
+    }
+
+    @Test
+    public void testDeletePerformanceVideoItem() {
+        CampaignPerformanceVideoItemOperation campaignItem = generateDummyCampaignPerformanceVideoItemOperation();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        when(endpointMock.deleteVideoCreative(auth.getToken().getAccessTokenForHeader(),"accountId", "1", "2")).thenReturn(campaignItem);
+
+        CampaignPerformanceVideoItem actual = testInstance.deletePerformanceVideoItem(auth, "accountId", "1", "2");
+        assertEquals("Invalid return value", campaignItem, actual);
+        verify(endpointMock, times(1)).deleteVideoCreative(any(), any(), any(), any());
+    }
+
+    @Test
+    public void testBulkCreatePerformanceVideoItem() {
+        CampaignPerformanceVideoItemOperation campaignItem = generateDummyCampaignPerformanceVideoItemOperation();
+        PerformanceVideoBulkCreateOperation bulkCreateOperation = generateDummyPerformanceVideoBulkCreateOperation();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        Results<CampaignPerformanceVideoItem> expectedResult = new Results<>(Collections.singletonList(campaignItem));
+        when(endpointMock.bulkCreateVideo(auth.getToken().getAccessTokenForHeader(),"accountId", bulkCreateOperation)).thenReturn(expectedResult);
+
+        Results<CampaignPerformanceVideoItem> actual = testInstance.bulkCreatePerformanceVideoItem(auth, "accountId", bulkCreateOperation);
+        assertEquals("Invalid return value", campaignItem, actual.getResults().iterator().next());
+        verify(endpointMock, times(1)).bulkCreateVideo(any(), any(), any());
+    }
+
+    @Test
+    public void testBulkUpdatePerformanceVideoItem() {
+        CampaignPerformanceVideoItemOperation campaignItem = generateDummyCampaignPerformanceVideoItemOperation();
+        PerformanceVideoBulkUpdateOperation bulkUpdateOperation = generateDummyPerformanceVideoBulkUpdateOperation();
+        BackstageAuthentication auth = generateDummyClientCredentialsBackstageAuth();
+        Results<CampaignPerformanceVideoItem> expectedResult = new Results<>(Collections.singletonList(campaignItem));
+        when(endpointMock.bulkUpdateVideo(auth.getToken().getAccessTokenForHeader(),"accountId", false, bulkUpdateOperation)).thenReturn(expectedResult);
+
+        Results<CampaignPerformanceVideoItem> actual = testInstance.bulkUpdatePerformanceVideoItem(auth, "accountId", bulkUpdateOperation);
+        assertEquals("Invalid return value", campaignItem, actual.getResults().iterator().next());
+        verify(endpointMock, times(1)).bulkUpdateVideo(any(), any(), anyBoolean(), any());
+    }
 }