From 9b02a49bc9d25e8972932f6cde9a3e668477f02c Mon Sep 17 00:00:00 2001
From: M1ha <work_shvein_mihail@mail.ru>
Date: Thu, 22 Nov 2018 11:09:38 +0500
Subject: [PATCH] Added all MergeTree engines supported in infi.clickhouse_orm

---
 src/django_clickhouse/engines.py | 8 ++++++++
 tests/clickhouse_models.py       | 4 ++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/django_clickhouse/engines.py b/src/django_clickhouse/engines.py
index d98e380..f81f33e 100644
--- a/src/django_clickhouse/engines.py
+++ b/src/django_clickhouse/engines.py
@@ -34,6 +34,14 @@ class MergeTree(InsertOnlyEngineMixin, infi_engines.MergeTree):
     pass
 
 
+class ReplacingMergeTree(InsertOnlyEngineMixin, infi_engines.ReplacingMergeTree):
+    pass
+
+
+class SummingMergeTree(InsertOnlyEngineMixin, infi_engines.SummingMergeTree):
+    pass
+
+
 class CollapsingMergeTree(InsertOnlyEngineMixin, infi_engines.CollapsingMergeTree):
     def get_final_versions(self, model_cls, objects):
         """
diff --git a/tests/clickhouse_models.py b/tests/clickhouse_models.py
index a773a5b..df8774e 100644
--- a/tests/clickhouse_models.py
+++ b/tests/clickhouse_models.py
@@ -1,7 +1,7 @@
 from infi.clickhouse_orm import fields
 
 from django_clickhouse.clickhouse_models import ClickHouseModel
-from django_clickhouse.engines import MergeTree, CollapsingMergeTree
+from django_clickhouse.engines import ReplacingMergeTree, CollapsingMergeTree
 from tests.models import TestModel
 
 
@@ -13,7 +13,7 @@ class ClickHouseTestModel(ClickHouseModel):
     created_date = fields.DateField()
     value = fields.Int32Field()
 
-    engine = MergeTree('created_date', ('id',))
+    engine = ReplacingMergeTree('created_date', ('id',))
 
 
 class ClickHouseCollapseTestModel(ClickHouseModel):