12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727 |
- #include <gmock/gmock.h>
- #include <gtest/gtest.h>
- #include "../include/Config.h"
- #include "../include/JsonCommander.h"
- #include "../include/Notifications.h"
- #include "ChannelControlsMock.h"
- #include "FileManagerMock.h"
- #include "QueueMock.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);
- }
- TEST(ExtendedStatus, NoUploadOrDownload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(false));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(false));
- 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["error"].asString(), "");
- EXPECT_TRUE(response.json["transfersclientserver"].isNull());
- EXPECT_TRUE(response.json["transfersserverserver"].isNull());
- }
- TEST(ExtendedStatus, ClientServerDownload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(true));
- EXPECT_CALL(fileManager, getGetBaseFileName()).WillOnce(testing::Return("asdf"));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(false));
- 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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 1);
- EXPECT_FALSE(response.json["transfersclientserver"][0]["upload"].asBool());
- EXPECT_EQ(response.json["transfersclientserver"][0]["file"].asString(), "asdf");
- EXPECT_EQ(response.json["transfersclientserver"][0]["progress"].asInt(), 0);
- }
- TEST(ExtendedStatus, ClientServerUpload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(true));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, getPutBaseFileName()).WillOnce(testing::Return("asdf"));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(false));
- 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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 1);
- EXPECT_TRUE(response.json["transfersclientserver"][0]["upload"].asBool());
- EXPECT_EQ(response.json["transfersclientserver"][0]["file"].asString(), "asdf");
- EXPECT_EQ(response.json["transfersclientserver"][0]["progress"].asInt(), 0);
- }
- TEST(ExtendedStatus, ClientServerDonwloadAndUpload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(true));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(true));
- EXPECT_CALL(fileManager, getGetBaseFileName()).WillOnce(testing::Return("asdfGet"));
- EXPECT_CALL(fileManager, getPutBaseFileName()).WillOnce(testing::Return("asdfPut"));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(false));
- 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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 2);
- EXPECT_TRUE(response.json["transfersclientserver"][0]["upload"].asBool());
- EXPECT_EQ(response.json["transfersclientserver"][0]["file"].asString(), "asdfPut");
- EXPECT_EQ(response.json["transfersclientserver"][0]["progress"].asInt(), 0);
- EXPECT_FALSE(response.json["transfersclientserver"][1]["upload"].asBool());
- EXPECT_EQ(response.json["transfersclientserver"][1]["file"].asString(), "asdfGet");
- EXPECT_EQ(response.json["transfersclientserver"][1]["progress"].asInt(), 0);
- }
- TEST(ExtendedStatus, ServerServerDownload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- Queue::fileName = "a.txt";
- Config::storage.clear();
- Config::storage.insert(std::pair<std::string, std::string>("passiveMode", "true"));
- Config::storage.insert(std::pair<std::string, std::string>("covertChannelMode", "m"));
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(false));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(true));
- EXPECT_CALL(channelControls, getProgress()).WillOnce(testing::Return(std::pair<int, int>(1, 2)));
- EXPECT_CALL(channelControls, getTransferStart()).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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 0);
- EXPECT_EQ(response.json["transfersserverserver"].size(), 1);
- EXPECT_EQ(response.json["transfersserverserver"][0]["type"].asString(), "download");
- EXPECT_EQ(response.json["transfersserverserver"][0]["file"].asString(), "a.txt");
- EXPECT_EQ(response.json["transfersserverserver"][0]["progress"].asInt(), 50);
- EXPECT_TRUE(response.json["transfersserverserver"][0]["speed"].isDouble());
- EXPECT_EQ(response.json["transfersserverserver"][0]["method"].asString(), "m");
- }
- TEST(ExtendedStatus, ServerServerUpload) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- Queue::fileName = "a.txt";
- Config::storage.clear();
- Config::storage.insert(std::pair<std::string, std::string>("passiveMode", "false"));
- Config::storage.insert(std::pair<std::string, std::string>("covertChannelMode", "m"));
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(false));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(true));
- EXPECT_CALL(channelControls, getProgress()).WillOnce(testing::Return(std::pair<int, int>(1, 2)));
- EXPECT_CALL(channelControls, getTransferStart()).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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 0);
- EXPECT_EQ(response.json["transfersserverserver"].size(), 1);
- EXPECT_EQ(response.json["transfersserverserver"][0]["type"].asString(), "upload");
- EXPECT_EQ(response.json["transfersserverserver"][0]["file"].asString(), "a.txt");
- EXPECT_EQ(response.json["transfersserverserver"][0]["progress"].asInt(), 50);
- EXPECT_TRUE(response.json["transfersserverserver"][0]["speed"].isDouble());
- EXPECT_EQ(response.json["transfersserverserver"][0]["method"].asString(), "m");
- }
- TEST(ExtendedStatus, QueueNotEmpty) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- ChannelControlsMock channelControls;
- Queue::channel = &channelControls;
- Queue::queue.clear();
- Queue::queue.push_back("a");
- Queue::queue.push_back("b");
- Config::storage.clear();
- Config::storage.insert(std::pair<std::string, std::string>("covertChannelMode", "m"));
- const std::string command = "extendedstatus";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, isUploading()).WillOnce(testing::Return(false));
- EXPECT_CALL(fileManager, isDownloading()).WillOnce(testing::Return(false));
- EXPECT_CALL(channelControls, isTransferRunning()).WillOnce(testing::Return(false));
- 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["error"].asString(), "");
- EXPECT_EQ(response.json["transfersclientserver"].size(), 0);
- EXPECT_EQ(response.json["transfersserverserver"].size(), 2);
- EXPECT_EQ(response.json["transfersserverserver"][0]["type"], "queued");
- EXPECT_EQ(response.json["transfersserverserver"][0]["file"], "a");
- EXPECT_EQ(response.json["transfersserverserver"][0]["progress"], 0);
- EXPECT_EQ(response.json["transfersserverserver"][0]["speed"], 0);
- EXPECT_EQ(response.json["transfersserverserver"][0]["method"], "m");
- EXPECT_EQ(response.json["transfersserverserver"][1]["type"], "queued");
- EXPECT_EQ(response.json["transfersserverserver"][1]["file"], "b");
- EXPECT_EQ(response.json["transfersserverserver"][1]["progress"], 0);
- EXPECT_EQ(response.json["transfersserverserver"][1]["speed"], 0);
- EXPECT_EQ(response.json["transfersserverserver"][1]["method"], "m");
- }
- TEST(Notifications, NoMessage) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "notifications";
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- 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_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- EXPECT_TRUE(response.json["messages"].isNull());
- }
- TEST(Notifications, OneMessage) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "notifications";
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- Notifications::newNotification("asdf");
- 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_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- EXPECT_TRUE(response.json["messages"].isArray());
- EXPECT_EQ(response.json["messages"][0], "asdf");
- // cleaning up
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- }
- TEST(Notifications, OneMessageMultipleTimesCalled) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "notifications";
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- Notifications::newNotification("asdf");
- 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_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- EXPECT_TRUE(response.json["messages"].isArray());
- EXPECT_EQ(response.json["messages"][0], "asdf");
- // make suhre that timestamp from message is older
- Notifications::messages.at(0).first--;
- 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["error"].asString(), "");
- EXPECT_TRUE(response.json["messages"].isNull());
- // cleaning up
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- }
- TEST(Notifications, TwoMessages) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "notifications";
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- Notifications::newNotification("asdf");
- Notifications::newNotification("qwer");
- 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_TRUE(response.json["accept"].asBool());
- EXPECT_EQ(response.json["error"].asString(), "");
- EXPECT_TRUE(response.json["messages"].isArray());
- EXPECT_EQ(response.json["messages"][0], "asdf");
- EXPECT_EQ(response.json["messages"][1], "qwer");
- // cleaning up
- Notifications::messages.clear();
- Notifications::userTimeStamps.clear();
- }
- TEST(ExtendedList, positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlist";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(0));
- EXPECT_CALL(fileManager, openExtendedList()).WillOnce(testing::Return(1));
- EXPECT_CALL(fileManager, getExtendedListSize()).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(ExtendedList, negative) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlist";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(0));
- EXPECT_CALL(fileManager, openExtendedList()).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(ExtendedList, EmptyList) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlist";
- Json::Value message;
- message["command"] = command;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(0));
- EXPECT_CALL(fileManager, openExtendedList()).WillOnce(testing::Return(0));
- EXPECT_CALL(fileManager, getExtendedListSize()).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(), "");
- }
- TEST(ExtendedListData, Positive) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlistdata";
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 0;
- message["cancel"] = false;
- std::vector<std::tuple<std::string, std::string, double>> chunks;
- std::tuple<std::string, std::string, double> chunk = std::tuple("asdf", "asdfghjkasdfghjkasdfghjkasdfghjk", 41.2);
- chunks.push_back(chunk);
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(1));
- EXPECT_CALL(fileManager, getNextChunkFromExtendedList()).WillOnce(testing::Return(chunks));
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_FALSE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), 0);
- EXPECT_TRUE(response.json["files"].isArray());
- EXPECT_EQ(response.json["files"].size(), 1);
- EXPECT_EQ(response.json["files"][0]["name"].asString(), "asdf");
- EXPECT_NE(response.json["files"][0]["head"].asString(), "");
- EXPECT_EQ(response.json["files"][0]["size"].asDouble(), 41.2);
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(ExtendedListData, Cancel) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlistdata";
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 0;
- message["cancel"] = true;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(1));
- EXPECT_CALL(fileManager, cancelExtendedList());
- JsonCommander::Response response = jsonCommander.execute(message);
- EXPECT_EQ(response.json["command"].asString(), command);
- EXPECT_TRUE(response.json["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), 0);
- EXPECT_TRUE(response.json["files"].isArray());
- EXPECT_EQ(response.json["error"].asString(), "");
- }
- TEST(ExtendedListData, WrongChunkNumber) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlistdata";
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 1;
- message["cancel"] = false;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(1));
- 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["files"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(ExtendedListData, NoChunksToBeSend) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlistdata";
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 1;
- message["cancel"] = false;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).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["cancel"].asBool());
- EXPECT_EQ(response.json["remaining"].asInt(), -1);
- EXPECT_TRUE(response.json["files"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- TEST(ExtendedListData, InvalidRequest) {
- FileManagerMock fileManager;
- JsonCommander jsonCommander(fileManager);
- const std::string command = "extendedlistdata";
- Json::Value message;
- message["command"] = command;
- message["chunk"] = 5;
- EXPECT_CALL(fileManager, getRemainingExtendedListChunks()).WillOnce(testing::Return(0));
- 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["files"].isArray());
- EXPECT_TRUE(response.json["error"].asString().compare("") != 0);
- }
- } // namespace
|