Fixed array casts

Author: Damtev

Diff for: utbot-framework/src/main/java/org/utbot/engine/overrides/collections/Collection.java

@@ -1,6 +1,7 @@
 package org.utbot.engine.overrides.collections;
 
 import org.utbot.api.annotation.UtClassMock;
+import org.utbot.api.mock.UtMock;
 import org.utbot.engine.overrides.stream.UtStream;
 
 import java.util.stream.Stream;
@@ -11,6 +12,8 @@ public interface Collection<E> extends java.util.Collection<E> {
     @Override
     default Stream<E> stream() {
         Object[] data = toArray();
+        UtMock.disableClassCastExceptionCheck(data);
+
         int size = data.length;
 
         return new UtStream<>((E[]) data, size);

Diff for: utbot-framework/src/main/java/org/utbot/engine/overrides/stream/Stream.java

@@ -13,10 +13,10 @@
 public interface Stream<E> extends BaseStream<E, Stream<E>> {
     @SuppressWarnings("unchecked")
     static <E> java.util.stream.Stream<E> of(E element) {
-        Object[] data = new Object[1];
+        E[] data = (E[]) new Object[1];
         data[0] = element;
 
-        return new UtStream<>((E[]) data, 1);
+        return new UtStream<>(data, 1);
     }
 
     @SuppressWarnings("unchecked")

Diff for: utbot-framework/src/main/java/org/utbot/engine/overrides/stream/UtStream.java

@@ -117,7 +117,7 @@ public Stream<E> filter(Predicate<? super E> predicate) {
         preconditionCheckWithClosingStream();
 
         int size = elementData.end;
-        Object[] filtered = new Object[size];
+        E[] filtered = (E[]) new Object[size];
         int j = 0;
         for (int i = 0; i < size; i++) {
             E element = elementData.get(i);
@@ -126,7 +126,7 @@ public Stream<E> filter(Predicate<? super E> predicate) {
             }
         }
 
-        return new UtStream<>((E[]) filtered, j);
+        return new UtStream<>(filtered, j);
     }
 
     @SuppressWarnings("unchecked")
@@ -230,7 +230,7 @@ public Stream<E> distinct() {
         preconditionCheckWithClosingStream();
 
         int size = elementData.end;
-        Object[] distinctElements = new Object[size];
+        E[] distinctElements = (E[]) new Object[size];
         int distinctSize = 0;
         for (int i = 0; i < size; i++) {
             E element = elementData.get(i);
@@ -259,7 +259,7 @@ public Stream<E> distinct() {
             }
         }
 
-        return new UtStream<>((E[]) distinctElements, distinctSize);
+        return new UtStream<>(distinctElements, distinctSize);
     }
 
     // TODO choose the best sorting https://github.com/UnitTestBot/UTBotJava/issues/188