130 lines
3.9 KiB
C++
130 lines
3.9 KiB
C++
/* XMRig
|
|
* Copyright 2016-2021 XMRig <https://github.com/xmrig>, <support@xmrig.com>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "proxy/workers/Worker.h"
|
|
|
|
using namespace xmrig;
|
|
|
|
class WorkerTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {
|
|
worker = new Worker(1, "test_worker", "127.0.0.1");
|
|
}
|
|
|
|
void TearDown() override {
|
|
delete worker;
|
|
}
|
|
|
|
Worker* worker = nullptr;
|
|
};
|
|
|
|
TEST_F(WorkerTest, InitialState) {
|
|
EXPECT_STREQ(worker->name(), "test_worker");
|
|
EXPECT_STREQ(worker->ip(), "127.0.0.1");
|
|
EXPECT_EQ(worker->id(), 1);
|
|
EXPECT_EQ(worker->connections(), 1); // Worker starts with 1 connection
|
|
EXPECT_EQ(worker->accepted(), 0);
|
|
EXPECT_EQ(worker->rejected(), 0);
|
|
EXPECT_EQ(worker->invalid(), 0);
|
|
}
|
|
|
|
TEST_F(WorkerTest, AddConnection) {
|
|
// Worker starts with 1 connection already
|
|
EXPECT_EQ(worker->connections(), 1);
|
|
|
|
worker->add("192.168.1.1");
|
|
EXPECT_EQ(worker->connections(), 2);
|
|
EXPECT_STREQ(worker->ip(), "192.168.1.1");
|
|
|
|
worker->add("192.168.1.2");
|
|
EXPECT_EQ(worker->connections(), 3);
|
|
}
|
|
|
|
TEST_F(WorkerTest, RemoveConnection) {
|
|
// Worker starts with 1 connection
|
|
worker->add("192.168.1.1");
|
|
worker->add("192.168.1.2");
|
|
worker->add("192.168.1.3");
|
|
EXPECT_EQ(worker->connections(), 4);
|
|
|
|
worker->remove();
|
|
EXPECT_EQ(worker->connections(), 3);
|
|
}
|
|
|
|
TEST_F(WorkerTest, AcceptedShares) {
|
|
// add(diff) increments accepted by 1, and adds diff to hashes
|
|
worker->add(1000);
|
|
EXPECT_EQ(worker->accepted(), 1); // accepted increments by 1
|
|
EXPECT_EQ(worker->hashes(), 1000); // hashes gets the difficulty
|
|
|
|
worker->add(5000);
|
|
EXPECT_EQ(worker->accepted(), 2); // accepted increments by 1 again
|
|
EXPECT_EQ(worker->hashes(), 6000); // hashes accumulates
|
|
}
|
|
|
|
TEST_F(WorkerTest, RejectedShares) {
|
|
worker->reject(false); // false = rejected (not invalid)
|
|
EXPECT_EQ(worker->rejected(), 1);
|
|
EXPECT_EQ(worker->invalid(), 0);
|
|
|
|
worker->reject(false);
|
|
EXPECT_EQ(worker->rejected(), 2);
|
|
}
|
|
|
|
TEST_F(WorkerTest, InvalidShares) {
|
|
worker->reject(true); // true = invalid
|
|
EXPECT_EQ(worker->invalid(), 1);
|
|
EXPECT_EQ(worker->rejected(), 0);
|
|
|
|
worker->reject(true);
|
|
EXPECT_EQ(worker->invalid(), 2);
|
|
}
|
|
|
|
TEST_F(WorkerTest, MixedOperations) {
|
|
// Worker starts with 1 connection
|
|
worker->add("192.168.1.1");
|
|
worker->add("192.168.1.2");
|
|
worker->add(10000); // accepted increments by 1, hashes gets 10000
|
|
worker->reject(false); // rejected
|
|
worker->reject(true); // invalid
|
|
|
|
EXPECT_EQ(worker->connections(), 3); // 1 initial + 2 added
|
|
EXPECT_EQ(worker->accepted(), 1); // add() increments by 1
|
|
EXPECT_EQ(worker->hashes(), 10000); // difficulty goes here
|
|
EXPECT_EQ(worker->rejected(), 1);
|
|
EXPECT_EQ(worker->invalid(), 1);
|
|
|
|
worker->remove();
|
|
EXPECT_EQ(worker->connections(), 2);
|
|
}
|
|
|
|
TEST_F(WorkerTest, EmptyWorkerName) {
|
|
Worker emptyWorker(2, "", "10.0.0.1");
|
|
EXPECT_STREQ(emptyWorker.name(), "");
|
|
}
|
|
|
|
TEST_F(WorkerTest, LongWorkerName) {
|
|
std::string longName(1000, 'x');
|
|
Worker longWorker(3, longName, "10.0.0.2");
|
|
EXPECT_STREQ(longWorker.name(), longName.c_str());
|
|
}
|
|
|
|
TEST_F(WorkerTest, HashrateCalculation) {
|
|
// Hashrate requires ticking and time
|
|
EXPECT_EQ(worker->hashrate(10), 0.0);
|
|
}
|