Add IexecHubServiceTests

Author: jbern0rd

src/test/java/com/iexec/blockchain/tool/IexecHubServiceTests.java

@@ -0,0 +1,98 @@
+/*
+ * Copyright 2023-2023 IEXEC BLOCKCHAIN TECH
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.iexec.blockchain.tool;
+
+import com.iexec.commons.poco.contract.generated.IexecHubContract;
+import lombok.SneakyThrows;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.web3j.crypto.Credentials;
+import org.web3j.crypto.ECKeyPair;
+import org.web3j.crypto.Keys;
+import org.web3j.protocol.core.RemoteFunctionCall;
+import org.web3j.protocol.core.methods.response.TransactionReceipt;
+
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+class IexecHubServiceTests {
+    private final ChainConfig chainConfig = ChainConfig
+            .builder()
+            .hubAddress("0xC129e7917b7c7DeDfAa5Fff1FB18d5D7050fE8ca")
+            .build();
+    @Mock
+    private CredentialsService credentialsService;
+    @Mock
+    private IexecHubContract iexecHubContract;
+    @Mock
+    private Web3jService web3jService;
+    @Mock
+    private RemoteFunctionCall<TransactionReceipt> remoteFunctionCall;
+    private IexecHubService iexecHubService;
+
+    @BeforeEach
+    void init() {
+        MockitoAnnotations.openMocks(this);
+        Credentials credentials = createEthereumCredentials();
+        when(credentialsService.getCredentials()).thenReturn(credentials);
+        iexecHubService = new IexecHubService(credentialsService, web3jService, chainConfig);
+        ReflectionTestUtils.setField(iexecHubService, "iexecHubContract", iexecHubContract);
+    }
+
+    @SneakyThrows
+    private Credentials createEthereumCredentials() {
+        ECKeyPair ecKeyPair = Keys.createEcKeyPair();
+        return Credentials.create(ecKeyPair);
+    }
+
+    @Test
+    void shouldNotInitializeTask() throws Exception {
+        when(iexecHubContract.initialize(any(), any())).thenReturn(remoteFunctionCall);
+        when(remoteFunctionCall.send()).thenThrow(Exception.class);
+        assertThatThrownBy(() -> iexecHubService.initializeTask("chainTaskId", 0))
+                .isInstanceOf(Exception.class);
+    }
+
+    @Test
+    void shouldNotContribute() throws Exception {
+        when(iexecHubContract.contribute(any(), any(), any(), any(), any(), any())).thenReturn(remoteFunctionCall);
+        when(remoteFunctionCall.send()).thenThrow(Exception.class);
+        assertThatThrownBy(() -> iexecHubService.contribute("chainTaskId", "resultDigest",
+                "workerpoolSignature", "enclaveChallenge", "enclaveSignature"))
+                .isInstanceOf(Exception.class);
+    }
+
+    @Test
+    void shouldNotReveal() throws Exception {
+        when(iexecHubContract.reveal(any(), any())).thenReturn(remoteFunctionCall);
+        when(remoteFunctionCall.send()).thenThrow(Exception.class);
+        assertThatThrownBy(() -> iexecHubService.reveal("chainTaskId", "resultDigest"))
+                .isInstanceOf(Exception.class);
+    }
+
+    @Test
+    void shouldNotFinalizeTask() throws Exception {
+        when(iexecHubContract.finalize(any(), any(), any())).thenReturn(remoteFunctionCall);
+        when(remoteFunctionCall.send()).thenThrow(Exception.class);
+        assertThatThrownBy(() -> iexecHubService.finalizeTask("chainTaskId", "resultLink", "callbackData"))
+                .isInstanceOf(Exception.class);
+    }
+}