Fix python tests

ildus · ildus · commit 5ab36a6b61ad · 2018-09-19T16:02:00.000+03:00
diff --git a/tests/python/partitioning_test.py b/tests/python/partitioning_test.py
@@ -1022,6 +1022,12 @@ def test_update_node_plan1(self):
                 plan = con.execute('SELECT query_plan(\'%s\')' % test_query)[0][0]
                 plan = plan[0]["Plan"]
 
+                # PartitionOverseer
+                self.assertEqual(plan["Node Type"], "Custom Scan")
+                self.assertEqual(plan["Custom Plan Provider"], 'PartitionOverseer')
+
+                # ModifyTable
+                plan = plan["Plans"][0]
                 self.assertEqual(plan["Node Type"], "ModifyTable")
                 self.assertEqual(plan["Operation"], "Update")
                 self.assertEqual(plan["Relation Name"], "test_range")
