2015-03-06 15:34:42 +00:00
|
|
|
#include <DB/IO/WriteBufferAIO.h>
|
|
|
|
|
|
|
|
#include <iostream>
|
|
|
|
#include <fstream>
|
|
|
|
#include <streambuf>
|
|
|
|
#include <cstdlib>
|
|
|
|
|
2015-03-10 16:34:00 +00:00
|
|
|
namespace
|
|
|
|
{
|
2015-03-06 15:34:42 +00:00
|
|
|
|
|
|
|
void die(const std::string & msg)
|
|
|
|
{
|
|
|
|
std::cout << msg;
|
|
|
|
::exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2015-03-10 16:34:00 +00:00
|
|
|
bool test1()
|
|
|
|
{
|
|
|
|
static const std::string symbols = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
|
|
|
|
|
|
|
|
char pattern[] = "/tmp/fileXXXXXX";
|
|
|
|
char * dir = ::mkdtemp(pattern);
|
|
|
|
if (dir == nullptr)
|
|
|
|
die("Could not create directory");
|
|
|
|
|
|
|
|
const std::string filename = std::string(dir) + "/foo";
|
|
|
|
|
|
|
|
size_t n = 10 * DB::WriteBufferAIO::BLOCK_SIZE;
|
|
|
|
|
|
|
|
std::string buf;
|
|
|
|
buf.reserve(n);
|
|
|
|
|
|
|
|
for (size_t i = 0; i < n; ++i)
|
|
|
|
buf += symbols[i % symbols.length()];
|
|
|
|
|
|
|
|
{
|
|
|
|
DB::WriteBufferAIO out(filename, 3 * DB::WriteBufferAIO::BLOCK_SIZE);
|
|
|
|
|
|
|
|
if (out.getFileName() != filename)
|
|
|
|
return false;
|
|
|
|
if (out.getFD() == -1)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
out.write(&buf[0], buf.length());
|
|
|
|
}
|
|
|
|
|
|
|
|
std::ifstream in(filename.c_str());
|
|
|
|
if (!in.is_open())
|
|
|
|
die("Could not open file");
|
|
|
|
|
|
|
|
std::string received{ std::istreambuf_iterator<char>(in), std::istreambuf_iterator<char>() };
|
|
|
|
|
|
|
|
in.close();
|
|
|
|
::unlink(filename.c_str());
|
|
|
|
|
|
|
|
return (received == buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
void run()
|
2015-03-06 15:34:42 +00:00
|
|
|
{
|
|
|
|
bool ok;
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
2015-03-10 16:34:00 +00:00
|
|
|
ok = test1();
|
2015-03-06 15:34:42 +00:00
|
|
|
}
|
|
|
|
catch (const DB::Exception & ex)
|
|
|
|
{
|
|
|
|
ok = false;
|
|
|
|
std::cout << "Caught exception " << ex.displayText() << "\n";
|
|
|
|
}
|
|
|
|
catch (const std::exception & ex)
|
|
|
|
{
|
|
|
|
ok = false;
|
|
|
|
std::cout << "Caught exception " << ex.what() << "\n";
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ok)
|
|
|
|
std::cout << "Test passed\n";
|
|
|
|
else
|
|
|
|
std::cout << "Test failed\n";
|
2015-03-10 16:34:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
2015-03-06 15:34:42 +00:00
|
|
|
|
2015-03-10 16:34:00 +00:00
|
|
|
int main()
|
|
|
|
{
|
|
|
|
run();
|
2015-03-06 15:34:42 +00:00
|
|
|
return 0;
|
|
|
|
}
|