delete_node_from_list.cc

#include "list_node.h"
#include "test_framework/generic_test.h"
#include "test_framework/timed_executor.h"

// Assumes node_to_delete is not tail.
void DeletionFromList(const shared_ptr<ListNode<int>>& node_to_delete) {
  node_to_delete->data = node_to_delete->next->data;
  node_to_delete->next = node_to_delete->next->next;
}

shared_ptr<ListNode<int>> DeletionFromListWrapper(
    TimedExecutor& executor, const shared_ptr<ListNode<int>>& head,
    int node_to_delete_idx) {
  shared_ptr<ListNode<int>> selected_node = head;
  while (node_to_delete_idx-- > 0) {
    if (!selected_node || !selected_node->next)
      throw std::runtime_error("Node index is out of range");
    selected_node = selected_node->next;
  }

  executor.Run([&] { DeletionFromList(selected_node); });
  return head;
}

int main(int argc, char* argv[]) {
  std::vector<std::string> args{argv + 1, argv + argc};
  std::vector<std::string> param_names{"executor", "head",
                                       "node_to_delete_idx"};
  return GenericTestMain(args, "delete_node_from_list.cc",
                         "delete_node_from_list.tsv", &DeletionFromListWrapper,
                         DefaultComparator{}, param_names);
}