You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-06-13 16:01:32 +03:00
Wrote a component test, which has a stand-in server thread
in place of an actual StorageManager. The server thread returns the same error over and over again. Used that test to debug some basic comm stuff & find errors in error handling.
This commit is contained in:
184
utils/cloudio/component_test.cpp
Normal file
184
utils/cloudio/component_test.cpp
Normal file
@ -0,0 +1,184 @@
|
||||
/* Copyright (C) 2014 InfiniDB, Inc.
|
||||
|
||||
This program is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU General Public License
|
||||
as published by the Free Software Foundation; version 2 of
|
||||
the License.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||
MA 02110-1301, USA. */
|
||||
|
||||
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/un.h>
|
||||
#include <boost/thread.hpp>
|
||||
#include <unistd.h>
|
||||
#include <list>
|
||||
#include <string>
|
||||
|
||||
#include "SMFileSystem.h"
|
||||
#include "SMDataFile.h"
|
||||
#include "SMFileFactory.h"
|
||||
#include "sm_exceptions.h"
|
||||
#include "messageFormat.h"
|
||||
|
||||
/* The purpose of this is to test the idbdatafile cloud classes using a dummy
|
||||
SessionManager defined in this file. */
|
||||
|
||||
#undef NDEBUG
|
||||
#include <assert.h>
|
||||
|
||||
using namespace idbdatafile;
|
||||
using namespace std;
|
||||
|
||||
volatile bool die = false;
|
||||
int errCode = 0;
|
||||
|
||||
/* Right now this will just accept one connection & send a fixed error response */
|
||||
void error_server_thread()
|
||||
{
|
||||
struct sockaddr_un addr;
|
||||
memset(&addr, 0, sizeof(addr));
|
||||
addr.sun_family = AF_UNIX;
|
||||
strcpy(&addr.sun_path[1], &storagemanager::socket_name[1]);
|
||||
|
||||
int server_socket = ::socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
assert(server_socket >= 0);
|
||||
|
||||
int err = ::bind(server_socket, (struct sockaddr *) &addr, sizeof(addr));
|
||||
assert(err == 0);
|
||||
|
||||
err = ::listen(server_socket, 1);
|
||||
assert(err == 0);
|
||||
|
||||
socklen_t addrlen;
|
||||
struct sockaddr_un client_addr;
|
||||
memset(&client_addr, 0, sizeof(client_addr));
|
||||
int client_socket = ::accept(server_socket, (struct sockaddr *) &client_addr, &addrlen);
|
||||
assert(client_socket >= 0);
|
||||
|
||||
//cout << "server thread got a connection" << endl;
|
||||
|
||||
uint8_t buf[4096];
|
||||
uint32_t response[4] = { storagemanager::SM_MSG_START, 8, (uint32_t ) -1, EINVAL };
|
||||
uint remainingBytes = 0;
|
||||
|
||||
while (!die)
|
||||
{
|
||||
/* This just scans for SM_MSG_START, and as it finds them it sends a generic error
|
||||
response. */
|
||||
|
||||
err = ::recv(client_socket, &buf[remainingBytes], 4096 - remainingBytes, MSG_DONTWAIT);
|
||||
if (err < 0)
|
||||
{
|
||||
if (errno == EAGAIN || errno == EWOULDBLOCK)
|
||||
{
|
||||
//cout << "server looping..." << endl;
|
||||
sleep(1); // who cares
|
||||
continue;
|
||||
}
|
||||
else {
|
||||
char errbuf[80];
|
||||
//cout << "server thread got an error: " << strerror_r(errno, errbuf, 80) << endl;
|
||||
close(client_socket);
|
||||
errCode = -1;
|
||||
die = true;
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
//cout << "server thread got some data" << endl;
|
||||
uint endOfData = remainingBytes + err;
|
||||
uint i;
|
||||
if (endOfData < 4)
|
||||
{
|
||||
remainingBytes = endOfData;
|
||||
continue;
|
||||
}
|
||||
for (i = 0; i <= endOfData - 4; i++)
|
||||
{
|
||||
if (*((uint32_t *) &buf[i]) == storagemanager::SM_MSG_START) {
|
||||
//cout << "server thread found a msg start magic" << endl;
|
||||
err = ::send(client_socket, response, 16, 0);
|
||||
assert(err > 0);
|
||||
}
|
||||
}
|
||||
memmove(buf, &buf[i], endOfData - i); // should be the trailing 3 bytes of the data
|
||||
remainingBytes = endOfData - i;
|
||||
}
|
||||
}
|
||||
|
||||
int test1()
|
||||
{
|
||||
/* Start a server thread that returns generic errors */
|
||||
boost::thread server_thread(error_server_thread);
|
||||
|
||||
/* Instantiate each of SM subclasses, call each function, and verify the expected error response */
|
||||
int err;
|
||||
SMFileFactory factory;
|
||||
|
||||
cout << "open" << endl;
|
||||
IDBDataFile *f = factory.open("dummy", "r", 0, 0);
|
||||
assert(f == NULL && errno == EINVAL && !die);
|
||||
|
||||
SMFileSystem filesystem;
|
||||
|
||||
cout << "compressedSize" << endl;
|
||||
bool gotException = false;
|
||||
try {
|
||||
filesystem.compressedSize("dummy");
|
||||
}
|
||||
catch (NotImplementedYet &) {
|
||||
gotException = true;
|
||||
}
|
||||
assert(gotException && !die);
|
||||
|
||||
cout << "copyFile" << endl;
|
||||
try {
|
||||
filesystem.copyFile("dummy1", "dummy2");
|
||||
}
|
||||
catch (NotImplementedYet &) {
|
||||
gotException = true;
|
||||
}
|
||||
assert(gotException && !die);
|
||||
|
||||
cout << "exists" << endl;
|
||||
err = filesystem.exists("dummy");
|
||||
assert(!err);
|
||||
|
||||
cout << "filesystemisup" << endl;
|
||||
err = filesystem.filesystemIsUp();
|
||||
assert(!err && !die);
|
||||
|
||||
cout << "isdir" << endl;
|
||||
err = filesystem.isDir("dummy");
|
||||
assert(!err && !die);
|
||||
|
||||
cout << "listdirectory" << endl;
|
||||
list<string> filenames;
|
||||
err = filesystem.listDirectory("dummy", filenames);
|
||||
assert(err == -1 && filenames.empty() && !die);
|
||||
|
||||
// done, return errCode
|
||||
die = true;
|
||||
server_thread.join();
|
||||
return errCode;
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
int ret;
|
||||
|
||||
ret = test1();
|
||||
return ret;
|
||||
}
|
||||
|
Reference in New Issue
Block a user