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_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, [])