def test_should_delete_node_from_elasticsearch_when_deleted(self):
node = DeliveryNodeFactory()
SyncInfo.objects.create(status=SyncInfo.STATUS.SUCCESSFUL)
node.delete()
nodes_to_delete = list_nodes_to_delete()
self.assertIn(node.id, nodes_to_delete)
def test_should_delete_node_from_elasticsearch_when_deleted(self):
node = DeliveryNodeFactory()
SyncInfo.objects.create(status=SyncInfo.STATUS.SUCCESSFUL)
node.delete()
nodes_to_delete = list_nodes_to_delete()
self.assertIn(node.id, nodes_to_delete)
def test_should_update_balance_when_node_quantities_change(self):
node = DeliveryNodeFactory(quantity=100, acknowledged=100)
child = DeliveryNodeFactory(parents=[(node, 50)])
self.assertEqual(child.balance, 50)
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 50)
child_two = DeliveryNodeFactory(parents=[(node, 40)])
self.assertEqual(child_two.balance, 40)
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 10)
child_two.delete()
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 50)
def test_should_update_balance_when_node_quantities_change(self):
node = DeliveryNodeFactory(quantity=100, acknowledged=100)
child = DeliveryNodeFactory(parents=[(node, 50)])
self.assertEqual(child.balance, 50)
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 50)
child_two = DeliveryNodeFactory(parents=[(node, 40)])
self.assertEqual(child_two.balance, 40)
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 10)
child_two.delete()
self.assertEqual(DeliveryNode.objects.get(id=node.id).balance, 50)
def test_should_push_deleted_nodes_to_elasticsearch(self, mock_serialise_nodes,
mock_post, mock_get, mock_converter, *_):
mock_get.return_value = FakeResponse({}, status_code=HTTP_200_OK)
node = DeliveryNodeFactory()
run()
node.delete()
api_data = '{}'
stub_nodes_to_update = {'id': 10}
mock_serialise_nodes.return_value = stub_nodes_to_update
mock_converter.return_value = api_data
mock_post.return_value = FakeResponse({}, status_code=HTTP_200_OK)
run()
mock_converter.assert_called_with(stub_nodes_to_update, [node.id])
mock_post.assert_called_with(settings.ELASTIC_SEARCH.BULK, data=api_data)
self.assertListEqual(DeleteRecords.objects.first().nodes_to_delete, [])
