12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199 |
- #include <gmock/gmock.h>
- #include <gtest/gtest.h>
- #include "../include/JsonCommander.h"
- #include "FileManagerMock.h"
- namespace {
- /* Version tests */
- TEST(testVersion, PositiveAllEqual) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- Json::Value message;
- message["major"] = 0;
- message["minor"] = 1;
- JsonCommander::Response response = jsonCommander.testVersion(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["major"].asInt(), jsonCommander.protocolMajorVersion);
- EXPECT_EQ(response.json["minor"].asInt(), jsonCommander.protocolMinorVersion);
- }
- TEST(testVersion, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- Json::Value message;
- message["major"] = jsonCommander.protocolMajorVersion;
- message["minor"] = jsonCommander.protocolMinorVersion - 1;
- JsonCommander::Response response = jsonCommander.testVersion(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["major"].asInt(), jsonCommander.protocolMajorVersion);
- EXPECT_EQ(response.json["minor"].asInt(), jsonCommander.protocolMinorVersion);
- }
- TEST(testVersion, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- Json::Value message;
- message["major"] = "def";
- message["minor"] = "abc";
- JsonCommander::Response response = jsonCommander.testVersion(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["major"].asInt(), jsonCommander.protocolMajorVersion);
- EXPECT_EQ(response.json["minor"].asInt(), jsonCommander.protocolMinorVersion);
- }
- TEST(testVersion, NotEqualMajorNumber) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- Json::Value message;
- message["major"] = jsonCommander.protocolMajorVersion + 1;
- message["minor"] = jsonCommander.protocolMinorVersion;
- JsonCommander::Response response = jsonCommander.testVersion(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["major"].asInt(), jsonCommander.protocolMajorVersion);
- EXPECT_EQ(response.json["minor"].asInt(), jsonCommander.protocolMinorVersion);
- }
- TEST(testVersion, BiggerMinorNumber) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- Json::Value message;
- message["major"] = jsonCommander.protocolMajorVersion;
- message["minor"] = jsonCommander.protocolMinorVersion + 1;
- JsonCommander::Response response = jsonCommander.testVersion(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["major"].asInt(), jsonCommander.protocolMajorVersion);
- EXPECT_EQ(response.json["minor"].asInt(), jsonCommander.protocolMinorVersion);
- }
- /* Status tests */
- TEST(Status, Ok) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "status";
- Json::Value message;
- message["command"] = command;
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["response"].asString(), "ok");
- }
- TEST(Status, Downloading) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "status";
- Json::Value message;
- message["command"] = command;
- ON_CALL(fileManager, isDownloading()).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["response"].asString(), "download running");
- }
- TEST(Status, Uploading) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "status";
- Json::Value message;
- message["command"] = command;
- ON_CALL(fileManager, isUploading()).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["response"].asString(), "upload running");
- }
- TEST(Status, UploadingAndDownloading) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "status";
- Json::Value message;
- message["command"] = command;
- ON_CALL(fileManager, isDownloading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, isUploading()).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["response"].asString(), "download and upload running");
- }
- /* Close tests */
- TEST(Close, Close) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "close";
- Json::Value message;
- message["command"] = command;
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["response"].asString(), "bye");
- }
- /* Put tests */
- TEST(Put, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "put";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- message["size"] = 1337;
- message["chunks"] = 1;
- ON_CALL(fileManager, openPutFile(testing::_)).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Put, Negative) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "put";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- message["size"] = 1337;
- message["chunks"] = 1;
- ON_CALL(fileManager, openPutFile(testing::_)).WillByDefault(testing::Return(false));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_TRUE(response.json["error"].asString().length() > 0);
- }
- /* Putdata tests */
- TEST(Putdata, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- /* start with put */
- std::string command = "put";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- message["size"] = 1337;
- const int chunks = 3;
- message["chunks"] = chunks;
- ON_CALL(fileManager, openPutFile(testing::_)).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* putdata */
- command = "putdata";
- ON_CALL(fileManager, isUploading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getPutBaseFileName()).WillByDefault(testing::Return(filename));
- for (int remaining = chunks - 1; remaining >= 0; remaining--) {
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["data"] = "MTMzNw==";
- message["remaining"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["received"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- }
- TEST(Putdata, Cancel) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- /* start with put */
- std::string command = "put";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- message["size"] = 1337;
- const int chunks = 3;
- message["chunks"] = chunks;
- ON_CALL(fileManager, openPutFile(testing::_)).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* putdata */
- command = "putdata";
- ON_CALL(fileManager, isUploading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getPutBaseFileName()).WillByDefault(testing::Return(filename));
- int remaining = chunks - 1;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["data"] = "MTMzNw==";
- message["remaining"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["received"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- // cancel transfer
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["data"] = "MTMzNw==";
- message["remaining"] = --remaining;
- message["cancel"] = true;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["received"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Putdata, WrongRemaining) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- /* start with put */
- std::string command = "put";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- message["size"] = 1337;
- const int chunks = 3;
- message["chunks"] = chunks;
- ON_CALL(fileManager, openPutFile(testing::_)).WillByDefault(testing::Return(true));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* putdata */
- command = "putdata";
- ON_CALL(fileManager, isUploading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getPutBaseFileName()).WillByDefault(testing::Return(filename));
- int remaining = chunks - 1;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["data"] = "MTMzNw==";
- message["remaining"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["received"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["error"].asString(), "");
- message = Json::Value();
- // skip remaining=1 and provoke an error
- remaining = 0;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["data"] = "MTMzNw==";
- message["remaining"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["received"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_TRUE(response.json["error"].asString().length() > 0);
- }
- /* Get tests */
- TEST(Get, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "get";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- const int chunks = 3;
- EXPECT_CALL(fileManager, openGetFile(testing::_)).WillOnce(testing::Return(std::pair<bool, int>(true, chunks)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["chunks"].asInt(), chunks);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Get, Negative) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "get";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- EXPECT_CALL(fileManager, openGetFile(testing::_)).WillOnce(testing::Return(std::pair<bool, int>(false, -1)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["chunks"].asInt(), -1);
- EXPECT_TRUE(response.json["error"].asString().length() > 0);
- }
- /* Getdata tests */
- TEST(Getdata, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- std::string command = "get";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- const int chunks = 3;
- EXPECT_CALL(fileManager, openGetFile(testing::_)).WillOnce(testing::Return(std::pair<bool, int>(true, chunks)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["chunks"].asInt(), chunks);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* getdata */
- command = "getdata";
- ON_CALL(fileManager, isDownloading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getGetBaseFileName()).WillByDefault(testing::Return(filename));
- std::vector<char> data;
- data.push_back('1');
- data.push_back('3');
- data.push_back('3');
- data.push_back('7');
- ON_CALL(fileManager, readGet()).WillByDefault(testing::Return(data));
- for (int remaining = chunks - 1; remaining >= 0; remaining--) {
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["chunk"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["data"].asString(), "MTMzNw==");
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- }
- TEST(Getdata, Cancle) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- std::string command = "get";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- const int chunks = 3;
- EXPECT_CALL(fileManager, openGetFile(testing::_)).WillOnce(testing::Return(std::pair<bool, int>(true, chunks)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["chunks"].asInt(), chunks);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* getdata */
- command = "getdata";
- ON_CALL(fileManager, isDownloading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getGetBaseFileName()).WillByDefault(testing::Return(filename));
- std::vector<char> data;
- data.push_back('1');
- data.push_back('3');
- data.push_back('3');
- data.push_back('7');
- ON_CALL(fileManager, readGet()).WillByDefault(testing::Return(data));
- int remaining = chunks - 1;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["chunk"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["data"].asString(), "MTMzNw==");
- EXPECT_EQ(response.json["error"].asString(), "");
- // set cancel to true
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["chunk"] = --remaining;
- message["cancel"] = true;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["data"].asString(), "");
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Getdata, WrongChunk) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- std::string command = "get";
- const std::string filename = "cool.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = filename;
- const int chunks = 3;
- EXPECT_CALL(fileManager, openGetFile(testing::_)).WillOnce(testing::Return(std::pair<bool, int>(true, chunks)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["chunks"].asInt(), chunks);
- EXPECT_EQ(response.json["error"].asString(), "");
- /* getdata */
- command = "getdata";
- ON_CALL(fileManager, isDownloading()).WillByDefault(testing::Return(true));
- ON_CALL(fileManager, getGetBaseFileName()).WillByDefault(testing::Return(filename));
- std::vector<char> data;
- data.push_back('1');
- data.push_back('3');
- data.push_back('3');
- data.push_back('7');
- ON_CALL(fileManager, readGet()).WillByDefault(testing::Return(data));
- int remaining = chunks - 1;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["chunk"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["data"].asString(), "MTMzNw==");
- EXPECT_EQ(response.json["error"].asString(), "");
- // skip chunk=0
- remaining = 0;
- message = Json::Value();
- message["command"] = command;
- message["file"] = filename;
- message["chunk"] = remaining;
- message["cancel"] = false;
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining);
- EXPECT_EQ(response.json["file"].asString(), filename);
- EXPECT_EQ(response.json["data"].asString(), "");
- EXPECT_TRUE(response.json["error"].asString().length() > 0);
- }
- /* List tests */
- TEST(List, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "list";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, openList()).WillOnce(testing::Return(1));
- EXPECT_CALL(fileManager, getListSize()).WillOnce(testing::Return(5));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["chunks"].asInt(), 1);
- EXPECT_EQ(response.json["items"].asInt(), 5);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(List, Negative) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "list";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, openList()).WillOnce(testing::Return(-1));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["chunks"].asInt(), -1);
- EXPECT_EQ(response.json["items"].asInt(), -1);
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(List, EmptyList) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "list";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, openList()).WillOnce(testing::Return(0));
- EXPECT_CALL(fileManager, getListSize()).WillOnce(testing::Return(0));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["chunks"].asInt(), 0);
- EXPECT_EQ(response.json["items"].asInt(), 0);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- /* Listdata tests */
- void fillExampleFileList(std::vector<std::string> (&chunk)[3]) {
- chunk[0].push_back("file01.txt");
- chunk[0].push_back("bumdibumps");
- chunk[0].push_back("1");
- chunk[0].push_back("Ich habe Hunger.txt");
- chunk[0].push_back("answerIs42");
- chunk[0].push_back("123456789456115811");
- chunk[0].push_back("kek");
- chunk[1].push_back("1337");
- chunk[1].push_back("cats.png");
- chunk[1].push_back("more_cats.png");
- chunk[1].push_back("ugly dog.tiff");
- chunk[1].push_back("hello.txt");
- chunk[1].push_back("bye.exe");
- chunk[1].push_back("poster.pdf");
- chunk[2].push_back("headbang.gif");
- chunk[2].push_back("feelsbad.jpg");
- chunk[2].push_back("hack.s");
- chunk[2].push_back("SodiumChloride");
- chunk[2].push_back("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrst"
- "uvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMN"
- "OPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
- }
- TEST(Listdata, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "listdata";
- const int chunks = 3;
- std::vector<std::string> chunk[chunks];
- fillExampleFileList(chunk);
- int remaining = chunks - 1;
- for (int k = 0; k < chunks; k++) {
- Json::Value message;
- message["command"] = command;
- message["chunk"] = remaining;
- message["cancel"] = false;
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(remaining + 1));
- EXPECT_CALL(fileManager, getNextChunkFromList()).WillOnce(testing::Return(chunk[k]));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining--);
- EXPECT_TRUE(response.json["names"].isArray());
- Json::Value array = response.json["names"];
- EXPECT_EQ(array.size(), chunk[k].size());
- for (int i = 0; i < 3; i++) {
- EXPECT_EQ(array[i].asString(), chunk[k][i]);
- }
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- }
- TEST(Listdata, Cancel) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "listdata";
- const int chunks = 3;
- std::vector<std::string> chunk[chunks];
- fillExampleFileList(chunk);
- int remaining = chunks - 1;
- Json::Value message;
- message["command"] = command;
- message["chunk"] = remaining;
- message["cancel"] = false;
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(remaining + 1));
- EXPECT_CALL(fileManager, getNextChunkFromList()).WillOnce(testing::Return(chunk[0]));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining--);
- EXPECT_TRUE(response.json["names"].isArray());
- Json::Value array = response.json["names"];
- EXPECT_EQ(array.size(), chunk[0].size());
- for (int i = 0; i < 3; i++) {
- EXPECT_EQ(array[i].asString(), chunk[0][i]);
- }
- EXPECT_EQ(response.json["error"].asString(), "");
- message = Json::Value();
- message["command"] = command;
- message["chunk"] = remaining;
- message["cancel"] = true;
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(remaining + 1));
- response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), remaining--);
- EXPECT_TRUE(response.json["names"].isArray());
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Listdata, WrongChunkNumber) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "listdata";
- const int chunks = 3;
- int remaining = chunks - 1;
- Json::Value message;
- message["command"] = command;
- message["chunk"] = remaining;
- message["cancel"] = false;
- // return smaller remaining
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(remaining));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), -1);
- EXPECT_TRUE(response.json["names"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(Listdata, NoChunksToBeSend) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "listdata";
- const int chunks = 0;
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 1;
- message["cancel"] = false;
- // return smaller remaining
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(chunks));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), -1);
- EXPECT_TRUE(response.json["names"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(Listdata, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "listdata";
- const int chunks = 3;
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 1;
- // return smaller remaining
- EXPECT_CALL(fileManager, getRemainingListChunks()).WillOnce(testing::Return(chunks));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), -1);
- EXPECT_TRUE(response.json["names"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(Head, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "head";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- std::vector<char> bytes = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'};
- const std::string bytesAsString = "YWJjZGVmZ2g=";
- EXPECT_CALL(fileManager, getBytesFromFile(testing::_, testing::_)).WillOnce(testing::Return(std::make_pair(bytes, FileManager::Error::no_error)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_EQ(response.json["data"].asString(), bytesAsString);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Head, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "head";
- const int file = 3641;
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), "");
- EXPECT_EQ(response.json["data"].asString(), "");
- EXPECT_NE(response.json["error"].asString(), "");
- }
- TEST(Head, NoSuchFile) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "head";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- std::vector<char> bytes;
- EXPECT_CALL(fileManager, getBytesFromFile(testing::_, testing::_)).WillOnce(testing::Return(std::make_pair(bytes, FileManager::Error::no_such_file)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_EQ(response.json["data"].asString(), "");
- EXPECT_NE(response.json["error"].asString(), "");
- }
- TEST(Head, FileTooSmall) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "head";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- std::vector<char> bytes;
- EXPECT_CALL(fileManager, getBytesFromFile(testing::_, testing::_)).WillOnce(testing::Return(std::make_pair(bytes, FileManager::Error::file_too_small)));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_EQ(response.json["data"].asString(), "");
- EXPECT_NE(response.json["error"].asString(), "");
- }
- TEST(Deleteme, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- // need to set currentUser in jsonCommander via calling checkLogin
- Json::Value login;
- login["login"] = true;
- login["user"] = "positive";
- login["pass"] = "positive";
- login["cancel"] = false;
- JsonCommander::Response loginRes = jsonCommander.checkLogin(login);
- EXPECT_TRUE(loginRes.json["accept"].asBool());
- EXPECT_EQ(loginRes.json["error"].asString(), "");
- // now the actual test
- const std::string command = "deleteme";
- Json::Value message;
- message["command"] = command;
- message["pass"] = "positive";
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_EQ(response.action, JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(Deleteme, Negative) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- // need to set currentUser in jsonCommander via calling checkLogin
- Json::Value login;
- login["login"] = true;
- login["user"] = "positive";
- login["pass"] = "positive";
- login["cancel"] = false;
- JsonCommander::Response loginRes = jsonCommander.checkLogin(login);
- EXPECT_TRUE(loginRes.json["accept"].asBool());
- EXPECT_EQ(loginRes.json["error"].asString(), "");
- // now the actual test
- const std::string command = "deleteme";
- Json::Value message;
- message["command"] = command;
- message["pass"] = "negative";
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_EQ(response.action, JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_NE(response.json["error"].asString(), "");
- }
- TEST(Deleteme, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- // need to set currentUser in jsonCommander via calling checkLogin
- Json::Value login;
- login["login"] = true;
- login["user"] = "positive";
- login["pass"] = "positive";
- login["cancel"] = false;
- JsonCommander::Response loginRes = jsonCommander.checkLogin(login);
- EXPECT_TRUE(loginRes.json["accept"].asBool());
- EXPECT_EQ(loginRes.json["error"].asString(), "");
- // now the actual test
- const std::string command = "deleteme";
- Json::Value message;
- message["command"] = command;
- message["pass"] = 123;
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_EQ(response.action, JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_NE(response.json["error"].asString(), "");
- }
- TEST(DeleteFile, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "deletefile";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- EXPECT_CALL(fileManager, deleteFile(file)).WillOnce(testing::Return(FileManager::Error::no_error));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(DeleteFile, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "deletefile";
- const int file = 3641;
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::closeAndSend);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["file"].asString(), "");
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(DeleteFile, FileDoesNotExist) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "deletefile";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- EXPECT_CALL(fileManager, deleteFile(file)).WillOnce(testing::Return(FileManager::Error::no_such_file));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(DeleteFile, DisabledInConfig) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "deletefile";
- const std::string file = "asdf.txt";
- Json::Value message;
- message["command"] = command;
- message["file"] = file;
- EXPECT_CALL(fileManager, deleteFile(file)).WillOnce(testing::Return(FileManager::Error::not_allowed));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_TRUE(response.action == JsonCommander::Action::send);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_EQ(response.json["file"].asString(), file);
- EXPECT_FALSE(response.json["accept"].asBool());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- } // namespace
|