1 package org.onap.vid.job.command;
3 import com.google.common.collect.ImmutableMap;
4 import org.mockito.Mock;
5 import org.mockito.MockitoAnnotations;
6 import org.onap.vid.job.Job;
7 import org.onap.vid.job.JobCommand;
8 import org.onap.vid.job.JobType;
9 import org.testng.annotations.BeforeMethod;
10 import org.testng.annotations.DataProvider;
11 import org.testng.annotations.Test;
13 import java.util.Arrays;
15 import java.util.UUID;
16 import java.util.stream.Collectors;
18 import static org.hamcrest.CoreMatchers.equalTo;
19 import static org.hamcrest.MatcherAssert.assertThat;
20 import static org.mockito.Mockito.verify;
21 import static org.mockito.Mockito.when;
23 public class JobCommandFactoryTest {
25 private JobCommandFactory jobCommandFactory;
31 private JobCommand mockCommand;
34 public void initMocks() {
35 MockitoAnnotations.initMocks(this);
40 jobCommandFactory = new JobCommandFactory(any -> mockCommand);
44 public Object[][] jobTypes() {
47 ).map(v -> new Object[]{v}).collect(Collectors.toList()).toArray(new Object[][]{});
51 @Test(dataProvider = "jobTypes")
52 public void givenJob_createCommandCallsTheInitAndReturnsTheInstance(JobType jobType) {
54 final UUID uuid = UUID.randomUUID();
55 final Map<String, Object> data = ImmutableMap.of("foo", "bar");
57 when(job.getType()).thenReturn(jobType);
58 when(job.getUuid()).thenReturn(uuid);
59 when(job.getData()).thenReturn(data);
61 final JobCommand command = jobCommandFactory.toCommand(job);
63 verify(mockCommand).init(uuid, data);
65 assertThat(command, equalTo(mockCommand));