From bb6e9f53c23d89fafd9abe4b3596631efbb4b061 Mon Sep 17 00:00:00 2001
From: Maksim Andrianov <maksimandrianov1@gmail.com>
Date: Thu, 8 Aug 2019 16:29:04 +0300
Subject: [PATCH] [generator] Fixes test.

---
 generator/generator_tests/merge_collectors_tests.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/generator/generator_tests/merge_collectors_tests.cpp b/generator/generator_tests/merge_collectors_tests.cpp
index a4e17f8030..b7e4bcc199 100644
--- a/generator/generator_tests/merge_collectors_tests.cpp
+++ b/generator/generator_tests/merge_collectors_tests.cpp
@@ -38,6 +38,8 @@ UNIT_TEST(MergeCollector_Case1)
   collector2->Collect(MakeOsmElement(3 /* id */, {{"admin_level", "3"}}, OsmElement::EntityType::Relation));
   collector2->Collect(MakeOsmElement(4 /* id */, {{"admin_level", "4"}}, OsmElement::EntityType::Relation));
 
+  collector1->Finish();
+  collector2->Finish();
   collector1->Merge(*collector2);
   collector1->Save();
 
@@ -72,6 +74,8 @@ UNIT_TEST(MergeCollector_Case2)
   collection2->Collect(MakeOsmElement(3 /* id */, {{"admin_level", "3"}}, OsmElement::EntityType::Relation));
   collection2->Collect(MakeOsmElement(4 /* id */, {{"admin_level", "4"}}, OsmElement::EntityType::Relation));
 
+  collection1->Finish();
+  collection2->Finish();
   collection1->Merge(*collection2);
   collection1->Save();