29 lines
688 B
Java
29 lines
688 B
Java
package pl.edu.amu.wmi.bookapi.Integration;
|
|
|
|
import org.junit.jupiter.api.BeforeAll;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.data.mongodb.core.MongoTemplate;
|
|
import pl.edu.amu.wmi.bookapi.models.UserDocument;
|
|
|
|
import java.util.List;
|
|
|
|
public class BaseInt {
|
|
public MongoTemplate mongoTemplate;
|
|
|
|
@Autowired
|
|
public BaseInt(MongoTemplate mongoTemplate) {
|
|
this.mongoTemplate = mongoTemplate;
|
|
}
|
|
|
|
@BeforeAll
|
|
public void cleanUp() {
|
|
List.of(
|
|
UserDocument.class
|
|
).forEach(it -> this.mongoTemplate.remove(it));
|
|
|
|
System.out.println("Removing classes in cleanUp method");
|
|
|
|
}
|
|
|
|
}
|