beam/node/functionaltests/send_wrong_newtip_test.cpp at master · pilot-SyS/beam · GitHub

History

96 lines (79 loc) · 2.19 KB

Raw

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

// http://www.apache.org/licenses/LICENSE-2.0

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

// See the License for the specific language governing permissions and

// limitations under the License.

#include "node/node.h"

#include "utility/logger.h"

#include "tools/base_node_connection.h"

using namespace beam;

class TestNodeConnection : public BaseTestNode

{

public:

TestNodeConnection(int argc, char* argv[]);

private:

bool OnMsg2(proto::Boolean&& msg) override;

bool OnMsg2(proto::NewTip&& msg) override;

private:

Height m_Height;

};

TestNodeConnection::TestNodeConnection(int argc, char* argv[])

: BaseTestNode(argc, argv)

, m_Height(0)

{

m_Timeout = 60 * 1000;

}

bool TestNodeConnection::OnMsg2(proto::Boolean&& msg)

{

LOG_INFO() << "Node returned: " << msg.m_Value;

return true;

}

bool TestNodeConnection::OnMsg2(proto::NewTip&& msg)

{

Block::SystemState::ID id;

msg.m_Description.get_ID(id);

LOG_INFO() << "NewTip: " << id;

if (m_Height == 0)

{

LOG_INFO() << "Send NewTip with height = " << msg.m_Description.m_Height + 5;

proto::NewTip newMsg;

newMsg.m_Description = msg.m_Description;

newMsg.m_Description.m_Height += 5;

m_Height = newMsg.m_Description.m_Height;

Send(newMsg);

m_Timer->start(3 * 60 * 1000, false, [this]()

{

m_Failed = true;

io::Reactor::get_Current().stop();

});

}

else

{

if (msg.m_Description.m_Height == m_Height)

{

LOG_INFO() << "Ok";

}

else

{

LOG_INFO() << "Failed";

}

io::Reactor::get_Current().stop();

}

return true;

}

int main(int argc, char* argv[])

{

int logLevel = LOG_LEVEL_DEBUG;

auto logger = Logger::create(logLevel, logLevel);

TestNodeConnection connection(argc, argv);

connection.Run();

return connection.CheckOnFailed();

}

beam/node/functionaltests/send_wrong_newtip_test.cpp at master · pilot-SyS/beam · GitHub

Uh oh!

Expand file tree

Breadcrumbs

send_wrong_newtip_test.cpp

Latest commit

History

Breadcrumbs

send_wrong_newtip_test.cpp

File metadata and controls