Hawkin
/
arangodb
mirror of https://gitee.com/bigwinds/arangodb
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added test case

This commit is contained in:
Jan Steemann 2013-04-12 15:45:28 +02:00
parent b3e8313a5e
commit 5ff92dddb6
2 changed files with 124 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
UnitTests/Makefile.unittests
View File

@ -564,6 +564,7 @@ unittests-arangob:
@builddir@/bin/arangob --configuration none --quiet --server.username "$(USERNAME)" --server.password "$(PASSWORD)" --server.endpoint unix://$(VOCDIR)/arango.sock --requests 5000 --concurrency 2 --test skiplist --complexity 1 || test "x$(FORCE)" == "x1"
@builddir@/bin/arangob --configuration none --quiet --server.username "$(USERNAME)" --server.password "$(PASSWORD)" --server.endpoint unix://$(VOCDIR)/arango.sock --requests 500 --concurrency 3 --test aqltrx --complexity 1 || test "x$(FORCE)" == "x1"
@builddir@/bin/arangob --configuration none --quiet --server.username "$(USERNAME)" --server.password "$(PASSWORD)" --server.endpoint unix://$(VOCDIR)/arango.sock --requests 100 --concurrency 3 --test counttrx || test "x$(FORCE)" == "x1"
@builddir@/bin/arangob --configuration none --quiet --server.username "$(USERNAME)" --server.password "$(PASSWORD)" --server.endpoint unix://$(VOCDIR)/arango.sock --requests 500 --concurrency 3 --test multitrx || test "x$(FORCE)" == "x1"
kill `cat $(PIDFILE)`

123
arangosh/Benchmark/test-cases.h
View File

@ -36,6 +36,8 @@ static bool DeleteCollection (SimpleClient*, const string&);
static bool CreateCollection (SimpleClient*, const string&, const int);
static bool CreateDocument (SimpleClient*, const string&, const string&);
static bool CreateIndex (SimpleClient*, const string&, const string&, const string&);
// -----------------------------------------------------------------------------
@ -1017,6 +1019,93 @@ struct TransactionCountTest : public BenchmarkOperation {
/// @}
////////////////////////////////////////////////////////////////////////////////
// -----------------------------------------------------------------------------
// --SECTION-- transaction test
// -----------------------------------------------------------------------------
////////////////////////////////////////////////////////////////////////////////
/// @addtogroup Benchmark
/// @{
////////////////////////////////////////////////////////////////////////////////
struct TransactionMultiTest : public BenchmarkOperation {
TransactionMultiTest ()
: BenchmarkOperation () {
}
~TransactionMultiTest () {
}
bool setUp (SimpleClient* client) {
_c1 = string(Collection + "1");
_c2 = string(Collection + "2");
return DeleteCollection(client, _c1) &&
DeleteCollection(client, _c2) &&
CreateCollection(client, _c1, 2) &&
CreateCollection(client, _c2, 2) &&
CreateDocument(client, _c2, "{ \"_key\": \"sum\", \"count\": 0 }");
}
void tearDown () {
}
string url (const int threadNumber, const size_t threadCounter, const size_t globalCounter) {
return string("/_admin/execute");
}
HttpRequest::HttpRequestType type (const int threadNumber, const size_t threadCounter, const size_t globalCounter) {
return HttpRequest::HTTP_REQUEST_POST;
}
const char* payload (size_t* length, const int threadNumber, const size_t threadCounter, const size_t globalCounter, bool* mustFree) {
const size_t mod = globalCounter % 2;
TRI_string_buffer_t* buffer;
buffer = TRI_CreateSizedStringBuffer(TRI_UNKNOWN_MEM_ZONE, 256);
TRI_AppendStringStringBuffer(buffer, "var c1 = require(\"internal\").db._collection(\"");
TRI_AppendStringStringBuffer(buffer, _c1.c_str());
TRI_AppendStringStringBuffer(buffer, "\"); ");
TRI_AppendStringStringBuffer(buffer, "var c2 = require(\"internal\").db._collection(\"");
TRI_AppendStringStringBuffer(buffer, _c2.c_str());
TRI_AppendStringStringBuffer(buffer, "\"); TRANSACTION({ collections: { ");
if (mod == 0) {
TRI_AppendStringStringBuffer(buffer, "write: [ \"");
TRI_AppendStringStringBuffer(buffer, _c1.c_str());
TRI_AppendStringStringBuffer(buffer, "\", \"");
TRI_AppendStringStringBuffer(buffer, _c2.c_str());
TRI_AppendStringStringBuffer(buffer, "\" ] }, ");
TRI_AppendStringStringBuffer(buffer, "action: function () { var n = Math.floor(Math.random() * 25) + 1; c1.save({ count: n }); var d = c2.document(\"sum\"); c2.update(d, { count: d.count + n }); return true; } });");
}
else {
TRI_AppendStringStringBuffer(buffer, "read: [ \"");
TRI_AppendStringStringBuffer(buffer, _c1.c_str());
TRI_AppendStringStringBuffer(buffer, "\", \"");
TRI_AppendStringStringBuffer(buffer, _c2.c_str());
TRI_AppendStringStringBuffer(buffer, "\" ] }, ");
TRI_AppendStringStringBuffer(buffer, "action: function () { var r1 = 0; c1.toArray().forEach(function (d) { r1 += d.count }); var r2 = c2.document(\"sum\").count; if (r1 !== r2) { throw \"error\"; } return true; } });");
}
*length = TRI_LengthStringBuffer(buffer);
*mustFree = true;
char* ptr = buffer->_buffer;
buffer->_buffer = NULL;
TRI_FreeStringBuffer(TRI_UNKNOWN_MEM_ZONE, buffer);
return (const char*) ptr;
}
const map<string, string>& headers () {
static const map<string, string> headers;
return headers;
}
string _c1;
string _c2;
};
// -----------------------------------------------------------------------------
// --SECTION-- private variables
// -----------------------------------------------------------------------------
@ -1128,6 +1217,37 @@ static bool CreateIndex (SimpleClient* client,
return ! failed;
}
////////////////////////////////////////////////////////////////////////////////
/// @brief create a document
////////////////////////////////////////////////////////////////////////////////
static bool CreateDocument (SimpleClient* client,
const string& collection,
const string& payload) {
map<string, string> headerFields;
SimpleHttpResult* result = 0;
result = client->request(HttpRequest::HTTP_REQUEST_POST,
"/_api/document?collection=" + collection,
payload.c_str(),
payload.size(),
headerFields);
bool failed = true;
if (result != 0) {
if (result->getHttpReturnCode() == 200 ||
result->getHttpReturnCode() == 201 ||
result->getHttpReturnCode() == 202) {
failed = false;
}
delete result;
}
return ! failed;
}
////////////////////////////////////////////////////////////////////////////////
/// @brief return the test case for a name
////////////////////////////////////////////////////////////////////////////////
@ -1163,6 +1283,9 @@ static BenchmarkOperation* GetTestCase (const string& name) {
if (name == "counttrx") {
return new TransactionCountTest();
}
if (name == "multitrx") {
return new TransactionMultiTest();
}
return 0;
}