Home | History | Annotate | Download | only in netdutils 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #ifndef NETUTILS_MOCK_SYSCALLS_H
     18 #define NETUTILS_MOCK_SYSCALLS_H
     19 
     20 #include <atomic>
     21 #include <cassert>
     22 #include <memory>
     23 
     24 #include <gmock/gmock.h>
     25 #include <gtest/gtest.h>
     26 
     27 #include "netdutils/Syscalls.h"
     28 
     29 namespace android {
     30 namespace netdutils {
     31 
     32 class MockSyscalls : public Syscalls {
     33   public:
     34     virtual ~MockSyscalls() = default;
     35     // Use Return(ByMove(...)) to deal with movable return types.
     36     MOCK_CONST_METHOD3(open,
     37                        StatusOr<UniqueFd>(const std::string& pathname, int flags, mode_t mode));
     38     MOCK_CONST_METHOD3(socket, StatusOr<UniqueFd>(int domain, int type, int protocol));
     39     MOCK_CONST_METHOD3(getsockname, Status(Fd sock, sockaddr* addr, socklen_t* addrlen));
     40     MOCK_CONST_METHOD5(getsockopt, Status(Fd sock, int level, int optname, void* optval,
     41                                           socklen_t *optlen));
     42     MOCK_CONST_METHOD5(setsockopt, Status(Fd sock, int level, int optname, const void* optval,
     43                                           socklen_t optlen));
     44 
     45     MOCK_CONST_METHOD3(bind, Status(Fd sock, const sockaddr* addr, socklen_t addrlen));
     46     MOCK_CONST_METHOD3(connect, Status(Fd sock, const sockaddr* addr, socklen_t addrlen));
     47 
     48     // Use Return(ByMove(...)) to deal with movable return types.
     49     MOCK_CONST_METHOD2(eventfd, StatusOr<UniqueFd>(unsigned int initval, int flags));
     50     MOCK_CONST_METHOD3(ppoll, StatusOr<int>(pollfd* fds, nfds_t nfds, double timeout));
     51 
     52     MOCK_CONST_METHOD2(writev, StatusOr<size_t>(Fd fd, const std::vector<iovec>& iov));
     53     MOCK_CONST_METHOD2(write, StatusOr<size_t>(Fd fd, const Slice buf));
     54     MOCK_CONST_METHOD2(read, StatusOr<Slice>(Fd fd, const Slice buf));
     55     MOCK_CONST_METHOD5(sendto, StatusOr<size_t>(Fd sock, const Slice buf, int flags,
     56                                                 const sockaddr* dst, socklen_t dstlen));
     57     MOCK_CONST_METHOD5(recvfrom, StatusOr<Slice>(Fd sock, const Slice dst, int flags, sockaddr* src,
     58                                                  socklen_t* srclen));
     59     MOCK_CONST_METHOD2(shutdown, Status(Fd fd, int how));
     60     MOCK_CONST_METHOD1(close, Status(Fd fd));
     61 
     62     MOCK_CONST_METHOD2(fopen,
     63                        StatusOr<UniqueFile>(const std::string& path, const std::string& mode));
     64     MOCK_CONST_METHOD3(vfprintf, StatusOr<int>(FILE* file, const char* format, va_list ap));
     65     MOCK_CONST_METHOD3(vfscanf, StatusOr<int>(FILE* file, const char* format, va_list ap));
     66     MOCK_CONST_METHOD1(fclose, Status(FILE* file));
     67     MOCK_CONST_METHOD0(fork, StatusOr<pid_t>());
     68 };
     69 
     70 // For the lifetime of this mock, replace the contents of sSyscalls
     71 // with a pointer to this mock. Behavior is undefined if multiple
     72 // ScopedMockSyscalls instances exist concurrently.
     73 class ScopedMockSyscalls : public MockSyscalls {
     74   public:
     75     ScopedMockSyscalls() : mOld(sSyscalls.swap(*this)) { assert((mRefcount++) == 1); }
     76     virtual ~ScopedMockSyscalls() {
     77         sSyscalls.swap(mOld);
     78         assert((mRefcount--) == 0);
     79     }
     80 
     81   private:
     82     std::atomic<int> mRefcount{0};
     83     Syscalls& mOld;
     84 };
     85 
     86 }  // namespace netdutils
     87 }  // namespace android
     88 
     89 #endif /* NETUTILS_MOCK_SYSCALLS_H */
     90