chromium/tools/clang/spanify/tests/fct-decl-tests-original.cc

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <vector>

#include "base/memory/raw_ptr.h"

struct S {};

// Expected rewrite:
// S fct_declaration(base::span<const S> arg1,int arg2,S* arg3);
S fct_declaration(const S* arg1, int arg2, S* arg3);

// Expected rewrite:
// int fct_declaration2(base::raw_span<int> arg);
int fct_declaration2(raw_ptr<int> arg);

class Parent {
 public:
  Parent() = default;
  // Expected rewrite:
  // Parent(base::span<S> ptr): member(ptr){}
  Parent(S* ptr) : member(ptr) {}

  // Expected rewrite:
  // virtual base::span<S> get();
  virtual S* get();

 protected:
  // Expected rewrite:
  // base::span<S> member;
  S* member;
};

// Expected rewrite:
// virtual base::span<S> Parent::get()
S* Parent::get() {
  // Expected rewrite:
  // fct_declaration(member, 0, member.data());
  fct_declaration(member, 0, member);
  return member;
}

class Child : public Parent {
 public:
  Child() = default;

  // Expected rewrite:
  // base::span<S> get() override;
  S* get() override;
};

// Expected rewrite:
// base::span<S> Child::get()
S* Child::get() {
  // Expected rewrite:
  // return {};
  return nullptr;
}

// Expected rewrite:
// S fct_declaration(base::span<const S> arg1, int arg2, S* arg3)
S fct_declaration(const S* arg1, int arg2, S* arg3) {
  // leads arg1 to be rewritten.
  return arg1[1];
}

// Expected rewrite:
// int fct_declaration2(base::raw_span<int> arg)
int fct_declaration2(raw_ptr<int> arg) {
  return arg[1];
}

void fct() {
  // Expected rewrite:
  // Parent p({});
  Parent p(nullptr);
  // Leads Parent::get() return value to be rewritten.
  p.get()[0] = {};

  // Expected rewrite:
  // fct_declaration2({});
  fct_declaration2(nullptr);
}