fix test

tests/integration/test_force_deduplication/test.py

@@ -19,8 +19,15 @@ def start_cluster():
     finally:
         cluster.shutdown()
 
+def get_counts():
+    src = int(node.query("SELECT count() FROM test"))
+    a = int(node.query("SELECT count() FROM test_mv_a"))
+    b = int(node.query("SELECT count() FROM test_mv_b"))
+    c = int(node.query("SELECT count() FROM test_mv_c"))
+    return src, a, b, c
+
+
 def test_basic(start_cluster):
-    with pytest.raises(QueryRuntimeException):
     node.query(
         '''
         CREATE TABLE test (A Int64) ENGINE = ReplicatedMergeTree ('/clickhouse/test/tables/test','1') ORDER BY tuple();
@@ -29,21 +36,30 @@ def test_basic(start_cluster):
         CREATE MATERIALIZED VIEW test_mv_c Engine=ReplicatedMergeTree ('/clickhouse/test/tables/test_mv_c','1') order by tuple() AS SELECT A FROM test;
         INSERT INTO test values(999);
         INSERT INTO test values(999);
+        '''
+    )
+    with pytest.raises(QueryRuntimeException):
+        node.query(
+            '''
             SET max_partitions_per_insert_block = 3;
             INSERT INTO test SELECT number FROM numbers(10);
             '''
         )
 
-    assert int(node.query("SELECT count() FROM test")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_a")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_b")) == 1
-    assert int(node.query("SELECT count() FROM test_mv_c")) == 1
+    old_src, old_a, old_b, old_c = get_counts()
+    # number of rows in test_mv_a and test_mv_c depends on order of inserts into views
+    assert old_src == 11
+    assert old_a in (1, 11)
+    assert old_b == 1
+    assert old_c in (1, 11)
 
     node.query("INSERT INTO test SELECT number FROM numbers(10)")
-    assert int(node.query("SELECT count() FROM test")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_a")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_b")) == 1
-    assert int(node.query("SELECT count() FROM test_mv_c")) == 1
+    src, a, b, c = get_counts()
+    # no changes because of deduplication in source table
+    assert src == old_src
+    assert a == old_a
+    assert b == old_b
+    assert c == old_c
 
     node.query(
         '''
@@ -51,10 +67,11 @@ def test_basic(start_cluster):
         INSERT INTO test SELECT number FROM numbers(10);
         '''
     )
-    assert int(node.query("SELECT count() FROM test")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_a")) == 21  # first insert was succesfull with disabled dedup..
-    assert int(node.query("SELECT count() FROM test_mv_b")) == 11
-    assert int(node.query("SELECT count() FROM test_mv_c")) == 11
+    src, a, b, c = get_counts()
+    assert src == 11
+    assert a == old_a + 10    # first insert could be succesfull with disabled dedup
+    assert b == 11
+    assert c == old_c + 10
 
     with pytest.raises(QueryRuntimeException):
         node.query(
@@ -72,7 +89,10 @@ def test_basic(start_cluster):
         '''
     )
 
-    assert int(node.query("SELECT count() FROM test")) == 21
-    assert int(node.query("SELECT count() FROM test_mv_a")) == 31
-    assert int(node.query("SELECT count() FROM test_mv_b")) == 21
-    assert int(node.query("SELECT count() FROM test_mv_c")) == 21
+    src, a, b, c = get_counts()
+    assert src == 21
+    assert a == old_a + 20
+    assert b == 21
+    assert c == old_c + 20
+
+