libnetdutils/include/netdutils/MockSyscalls.h

/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef NETUTILS_MOCK_SYSCALLS_H
#define NETUTILS_MOCK_SYSCALLS_H

#include <atomic>
#include <cassert>
#include <memory>

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include "netdutils/Syscalls.h"

namespace android {
namespace netdutils {

class MockSyscalls : public Syscalls {
  public:
    virtual ~MockSyscalls() = default;
    // Use Return(ByMove(...)) to deal with movable return types.
    MOCK_CONST_METHOD3(socket, StatusOr<UniqueFd>(int domain, int type, int protocol));
    MOCK_CONST_METHOD3(getsockname, Status(Fd sock, sockaddr* addr, socklen_t* addrlen));
    MOCK_CONST_METHOD5(setsockopt, Status(Fd sock, int level, int optname, const void* optval,
                                          socklen_t optlen));

    MOCK_CONST_METHOD3(bind, Status(Fd sock, const sockaddr* addr, socklen_t addrlen));
    MOCK_CONST_METHOD3(connect, Status(Fd sock, const sockaddr* addr, socklen_t addrlen));

    // Use Return(ByMove(...)) to deal with movable return types.
    MOCK_CONST_METHOD2(eventfd, StatusOr<UniqueFd>(unsigned int initval, int flags));
    MOCK_CONST_METHOD3(ppoll, StatusOr<int>(pollfd* fds, nfds_t nfds, double timeout));
    MOCK_CONST_METHOD2(write, StatusOr<size_t>(Fd fd, const Slice buf));
    MOCK_CONST_METHOD2(read, StatusOr<Slice>(Fd fd, const Slice buf));
    MOCK_CONST_METHOD5(sendto, StatusOr<size_t>(Fd sock, const Slice buf, int flags,
                                                const sockaddr* dst, socklen_t dstlen));
    MOCK_CONST_METHOD5(recvfrom, StatusOr<Slice>(Fd sock, const Slice dst, int flags, sockaddr* src,
                                                 socklen_t* srclen));
    MOCK_CONST_METHOD2(shutdown, Status(Fd fd, int how));
    MOCK_CONST_METHOD1(close, Status(Fd fd));
};

// For the lifetime of this mock, replace the contents of sSyscalls
// with a pointer to this mock. Behavior is undefined if multiple
// ScopedMockSyscalls instances exist concurrently.
class ScopedMockSyscalls : public MockSyscalls {
  public:
    ScopedMockSyscalls() : mOld(sSyscalls.swap(*this)) { assert((mRefcount++) == 1); }
    virtual ~ScopedMockSyscalls() {
        sSyscalls.swap(mOld);
        assert((mRefcount--) == 0);
    }

  private:
    std::atomic<int> mRefcount{0};
    Syscalls& mOld;
};

}  // namespace netdutils
}  // namespace android

#endif /* NETUTILS_MOCK_SYSCALLS_H */