handle sonnar

axonivy-market · Sep 6, 2024 · ecd4814 · ecd4814
1 parent 93dc8b8
commit ecd4814
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/...service/src/main/java/com/axonivy/market/repository/impl/CustomProductRepositoryImpl.java b/...service/src/main/java/com/axonivy/market/repository/impl/CustomProductRepositoryImpl.java
@@ -6,10 +6,11 @@
 import com.axonivy.market.repository.CustomProductRepository;
 import com.axonivy.market.repository.ProductModuleContentRepository;
 import lombok.Builder;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.mongodb.core.FindAndModifyOptions;
 import org.springframework.data.mongodb.core.MongoTemplate;
-import org.springframework.data.mongodb.core.aggregation.*;
+import org.springframework.data.mongodb.core.aggregation.Aggregation;
+import org.springframework.data.mongodb.core.aggregation.AggregationOperation;
+import org.springframework.data.mongodb.core.aggregation.AggregationResults;
 import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.data.mongodb.core.query.Update;

diff --git a/...ice/src/test/java/com/axonivy/market/repository/impl/CustomProductRepositoryImplTest.java b/...ice/src/test/java/com/axonivy/market/repository/impl/CustomProductRepositoryImplTest.java
@@ -124,4 +124,12 @@ void testIncreaseInstallationCount_NullProduct() {
     int updatedCount = repo.increaseInstallationCount(ID);
     assertEquals(0, updatedCount);
   }
+
+  @Test
+  void testUpdateInitialCount() {
+    setUpMockAggregateResult();
+    int initialCount = 10;
+    repo.updateInitialCount(ID, initialCount);
+    verify(mongoTemplate).updateFirst(any(Query.class), eq(new Update().inc("InstallationCount", initialCount).set("SynchronizedInstallationCount", true)), eq(Product.class));
+  }
 }