25 lines
655 B
Java
25 lines
655 B
Java
package pl.edu.amu.wmi.bookapi.fixtures;
|
|
|
|
import org.springframework.beans.factory.annotation.*;
|
|
import org.springframework.data.mongodb.core.*;
|
|
import org.springframework.data.mongodb.core.query.*;
|
|
import org.springframework.stereotype.*;
|
|
import pl.edu.amu.wmi.bookapi.models.*;
|
|
|
|
import java.util.*;
|
|
|
|
@Component
|
|
public class IntegrationTestUtil {
|
|
@Autowired
|
|
MongoTemplate mongoTemplate;
|
|
|
|
public void cleanCollections() {
|
|
List.of(
|
|
UserDocument.class,
|
|
BookDocument.class,
|
|
MessageDocument.class,
|
|
ThreadDocument.class
|
|
).forEach(it -> mongoTemplate.remove(new Query(), it));
|
|
}
|
|
}
|