def test_bulk_put_overwrite(self, es):
sr = SplitterRegulations()
nodes = [{"text": "other", "label": ["111", "3"], "children": [], "node_type": "tyty1"}]
sr.bulk_put(nodes, "verver", "111-3")
regs = Regulation.objects.all()
self.assertEqual(1, len(regs))
self.assertEqual("tyty1", regs[0].node_type)
nodes[0]["node_type"] = "tyty2"
sr.bulk_put(nodes, "verver", "111-3")
regs = Regulation.objects.all()
self.assertEqual(1, len(regs))
self.assertEqual("tyty2", regs[0].node_type)
def test_bulk_put(self, es):
sr = SplitterRegulations()
nodes = [
{"text": "some text", "label": ["111", "2"], "children": [], "node_type": "tyty"},
{"text": "other", "label": ["111", "3"], "children": [], "node_type": "tyty2"},
]
sr.bulk_put(nodes, "verver", "111")
regs = Regulation.objects.all().order_by("text")
self.assertEqual(2, len(regs))
self.assertEqual("verver", regs[0].version)
self.assertEqual("111-3", regs[0].label_string)
self.assertEqual("other", regs[0].text)
self.assertEqual("", regs[0].title)
self.assertEqual("tyty2", regs[0].node_type)
self.assertEqual([], regs[0].children)
self.assertEqual("verver", regs[1].version)
self.assertEqual("111-2", regs[1].label_string)
self.assertEqual("some text", regs[1].text)
self.assertEqual("", regs[1].title)
self.assertEqual("tyty", regs[1].node_type)
self.assertEqual([], regs[1].children)
args = es.return_value.bulk_index.call_args[0]
self.assertEqual(3, len(args))
self.assertEqual("reg_tree", args[1])
nodes[0]["version"] = "verver"
nodes[0]["label_string"] = "111-2"
nodes[0]["regulation"] = "111"
nodes[0]["id"] = "verver/111-2"
nodes[0]["root"] = False
nodes[1]["version"] = "verver"
nodes[1]["label_string"] = "111-3"
nodes[1]["regulation"] = "111"
nodes[1]["id"] = "verver/111-3"
nodes[1]["root"] = False
self.assertEqual(nodes, args[2])
