[ignite-teamcity-bot] branch master updated: Tests fix after PR #76

[dp...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

dpavlov pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/ignite-teamcity-bot.git


The following commit(s) were added to refs/heads/master by this push:
     new e7d4319  Tests fix after PR #76
e7d4319 is described below

commit e7d43196ca14e3b01b1e5df3fd33708641fe8100
Author: Dmitriy Pavlov <dp...@apache.org>
AuthorDate: Wed Nov 21 20:26:01 2018 +0300

    Tests fix after PR #76
---
 .../test/java/org/apache/ignite/ci/di/DiContextTest.java    | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/ignite-tc-helper-web/src/test/java/org/apache/ignite/ci/di/DiContextTest.java b/ignite-tc-helper-web/src/test/java/org/apache/ignite/ci/di/DiContextTest.java
index 770b6ce..eb9a92a 100644
--- a/ignite-tc-helper-web/src/test/java/org/apache/ignite/ci/di/DiContextTest.java
+++ b/ignite-tc-helper-web/src/test/java/org/apache/ignite/ci/di/DiContextTest.java
@@ -22,16 +22,23 @@ import com.google.inject.Injector;
 import java.util.Collection;
 import java.util.concurrent.CompletableFuture;
 import org.apache.ignite.Ignite;
+import org.apache.ignite.IgniteCache;
 import org.apache.ignite.ci.ITcHelper;
 import org.apache.ignite.ci.observer.BuildObserver;
 import org.apache.ignite.ci.observer.ObserverTask;
 import org.apache.ignite.ci.teamcity.restcached.ITcServerProvider;
 import org.apache.ignite.ci.teamcity.restcached.ITcServerFactory;
 import org.apache.ignite.ci.web.TcUpdatePool;
+import org.apache.ignite.configuration.CacheConfiguration;
 import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
 
 import static org.junit.Assert.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 public class DiContextTest {
     /**
@@ -48,7 +55,11 @@ public class DiContextTest {
 
     public Injector getInjector() {
         IgniteTcBotModule igniteTcBotModule = new IgniteTcBotModule();
-        igniteTcBotModule.setIgniteFut(CompletableFuture.completedFuture(mock(Ignite.class)));
+        Ignite ignite = mock(Ignite.class);
+        when(ignite.getOrCreateCache(anyString())).thenReturn(Mockito.mock(IgniteCache.class));
+        when(ignite.getOrCreateCache(any(CacheConfiguration.class))).thenReturn(Mockito.mock(IgniteCache.class));
+
+        igniteTcBotModule.setIgniteFut(CompletableFuture.completedFuture(ignite));
 
         return Guice.createInjector(igniteTcBotModule);
     }