Skip to content

Commit 3b31262

Browse files
committed
Fix based on feedback
1 parent 6071b6f commit 3b31262

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

core/src/test/java/com/scalar/db/storage/cassandra/BatchHandlerTest.java

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@
2222
import com.scalar.db.exception.storage.ExecutionException;
2323
import com.scalar.db.exception.storage.NoMutationException;
2424
import com.scalar.db.io.Key;
25-
import java.util.ArrayList;
2625
import java.util.Arrays;
2726
import java.util.List;
27+
import java.util.stream.Collectors;
2828
import org.junit.jupiter.api.BeforeEach;
2929
import org.junit.jupiter.api.Test;
3030
import org.mockito.Mock;
@@ -102,17 +102,15 @@ private List<Mutation> prepareNonConditionalPuts() {
102102
}
103103

104104
private List<Mutation> prepareConditionalPuts() {
105-
List<Mutation> mutations = prepareNonConditionalPuts();
106-
List<Mutation> conditionalMutations = new ArrayList<>();
107-
for (Mutation m : mutations) {
108-
if (m instanceof Put) {
109-
conditionalMutations.add(
110-
Put.newBuilder((Put) m).condition(ConditionBuilder.putIfNotExists()).build());
111-
} else {
112-
conditionalMutations.add(m);
113-
}
114-
}
115-
return conditionalMutations;
105+
return prepareNonConditionalPuts().stream()
106+
.map(
107+
m -> {
108+
if (m instanceof Put) {
109+
return Put.newBuilder((Put) m).condition(ConditionBuilder.putIfNotExists()).build();
110+
}
111+
return m;
112+
})
113+
.collect(Collectors.toList());
116114
}
117115

118116
private BatchHandler prepareSpiedBatchHandler() {

0 commit comments

Comments
 (0)