1 files changed, 156 insertions, 0 deletions
diff --git a/test/mock/syscall_io_mock.hpp b/test/mock/syscall_io_mock.hpp
new file mode 100644
index 0000000..75efdb1
--- /dev/null
+++ b/test/mock/syscall_io_mock.hpp
@@ -0,0 +1,156 @@
+#include <gmock/gmock.h>
+#include <functional>
+
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+/*
+ssize_t read(int fd, void *buf, size_t count);
+ssize_t write(int fd, const void *buf, size_t count);
+int close(int fd);
+*/
+static std::function<ssize_t(int fd, void *buf, size_t count)> _read;
+static std::function<ssize_t(int fd, const void *buf, size_t count)> _write;
+static std::function<int(int)> _close;
+
+/*
+int unlink(const char *pathname);
+int stat(const char *pathname, struct stat *buf);
+*/
+static std::function<int(const char *)> _unlink;
+static std::function<int(const char *pathname, struct stat *buf)> _stat;
+
+
+/*
+int socket(int socket_family, int socket_type, int protocol);
+int bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen);
+int listen(int sockfd, int backlog);
+int accept4(int sockfd, struct sockaddr *addr,
+			socklen_t *addrlen, int flags);
+*/
+static std::function<int(int socket_family, int socket_type, int protocol)> _socket;
+static std::function<int(int sockfd, const struct sockaddr *addr,socklen_t addrlen)> _bind;
+static std::function<int(int sockfd, int backlog)> _listen;
+static std::function<int(int sockfd, struct sockaddr *addr,
+			socklen_t *addrlen, int flags)> _accept4;
+
+class SyscallIOMocker {
+public:
+	SyscallIOMocker() {
+		_read = [this](int fd, void *buf, size_t count) {
+			return read(fd, buf, count);
+		};
+		_write = [this](int fd, const void *buf, size_t count) {
+			return write(fd, buf, count);
+		};
+		_close = [this](int fd){
+			return close(fd);
+		};
+
+		_unlink = [this](const char *pathname){
+			return unlink(pathname);
+		};
+		_stat = [this](const char *pathname, struct stat *buf) {
+			return stat(pathname, buf);
+		};
+
+		_socket = [this](int socket_family, int socket_type, int protocol){
+			return socket(socket_family, socket_type, protocol);
+		};
+		_bind = [this](int sockfd, const struct sockaddr *addr,socklen_t addrlen){
+			return bind(sockfd, addr,addrlen);
+		};
+		_listen = [this](int sockfd, int backlog){
+			return listen(sockfd, backlog);
+		};
+		_accept4
+			= [this](int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags) {
+			return accept4(sockfd, addr, addrlen, flags);
+		};
+	}
+
+	~SyscallIOMocker() {
+		_read = {};
+		_write = {};
+		_close = {};
+
+		_unlink = {};
+		_stat = {};
+
+		_socket = {};
+		_bind = {};
+		_listen = {};
+		_accept4 = {};
+	}
+
+	MOCK_CONST_METHOD3(read, ssize_t(int fd, void *buf, size_t count));
+	MOCK_CONST_METHOD3(write, ssize_t(int fd, const void *buf, size_t count));
+	MOCK_CONST_METHOD1(close, int(int));
+
+	MOCK_CONST_METHOD1(unlink, int(const char *));
+	MOCK_CONST_METHOD2(stat, int(const char *pathname, struct stat *buf));
+
+	MOCK_CONST_METHOD3(socket, int(int socket_family, int socket_type, int protocol));
+	MOCK_CONST_METHOD3(bind, int(int sockfd, const struct sockaddr *addr,socklen_t addrlen));
+	MOCK_CONST_METHOD2(listen, int(int sockfd, int backlog));
+	MOCK_CONST_METHOD4(accept4, int(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags));
+};
+
+class SyscallIOMockBase {
+protected:
+	SyscallIOMocker sysiom;
+};
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+ssize_t read(int fd, void *buf, size_t count)
+{
+    return _read(fd, buf, count);
+}
+
+ssize_t write(int fd, const void *buf, size_t count)
+{
+    return _write(fd, buf, count);
+}
+
+int close(int fd)
+{
+    return _close(fd);
+}
+
+int unlink(const char *pathname)
+{
+	return _unlink(pathname);
+}
+
+int stat(const char *pathname, struct stat *buf)
+{
+	return _stat(pathname, buf);
+}
+
+int socket(int socket_family, int socket_type, int protocol)
+{
+	return _socket(socket_family, socket_type, protocol);
+}
+
+int bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen)
+{
+	return _bind(sockfd, addr, addrlen);
+}
+
+int listen(int sockfd, int backlog)
+{
+	return _listen(sockfd, backlog);
+}
+
+int accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags)
+{
+	return _accept4(sockfd, addr, addrlen, flags);
+}
+
+#ifdef __cplusplus
+}
+#endif