Integration tests with mongodb - mongodb

Question

I'm using mongodb as my main database, so on my integration tests I try to test if my system handles the data as expected. The process is as follow:
Insert data on needed collections to prepare scenario
Perform call to the system
Validate data change or exist on database, depending of the scenario
I'm using behat as my BDD testing tool and I implement on my FeatureContext to insert and read data from collections, something like this:
Given I have the following data on collection Brand
[
{
"_id": "575df0b00419c905492d0461",
"name": "Adidas",
"image": "adidas.png"
}
]
Then I validate the following data exists on collection Brand
[
{
"_id": "575df0b00419c905492d0461",
"name": "Adidas",
"image": "adidas.png"
}
]
EDIT
/**
* #Given I have the following data on collection :collectionName
* #param string $collectionName
* #param PyStringNode $collectionData
* #throws Exception
*/
public function iHaveTheFollowingDataOnCollection($collectionName, PyStringNode $collectionData)
{
$data = json_decode($collectionData->getRaw(), true);
foreach ($data as $index => $element) {
foreach ($element as $key => $value) {
if ($key === '_id') {
$data[$index][$key] = new \MongoDB\BSON\ObjectID($value);
}
if (strpos($key, "#") !== false) {
$key = explode("#", $key)[1];
$data[$index][$key] = array(
"\$ref" => $value["\$ref"],
"\$id" => new \MongoDB\BSON\ObjectID($value["\$id"]),
"\$db" => $value["\$db"],
);
}
if ($value === "UNIX_TIME()") {
$data[$index][$key] = time();
}
}
}
$collection = $this->db->$collectionName;
foreach ($data as $document) {
$collection->insertOne($document);
}
}
END EDIT
The issue is that I have at least 1k different scenarios on different features and with that heavy usage test randomly fails becase it don't find expected data on collection.
My first thought is that exist some delay on insertion and availability of the data.
Any thoughts and how to fix it?