mirror of https://github.com/meshcore-dev/MeshCore
Michael Lynch
1 month ago
2 changed files with 179 additions and 172 deletions
@ -0,0 +1,178 @@ |
|||||
|
#pragma once |
||||
|
|
||||
|
#include <cstddef> |
||||
|
#include <cstdint> |
||||
|
#include <cstring> |
||||
|
#include <optional> |
||||
|
|
||||
|
#include "helpers/BaseChatMesh.h" |
||||
|
#include "helpers/SimpleMeshTables.h" |
||||
|
|
||||
|
class FakeMillis final : public mesh::MillisecondClock { |
||||
|
public: |
||||
|
unsigned long getMillis() override { |
||||
|
return 0; |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
class FakeRtc final : public mesh::RTCClock { |
||||
|
public: |
||||
|
explicit FakeRtc(uint32_t initial_time) : _time(initial_time) {} |
||||
|
|
||||
|
uint32_t getCurrentTime() override { |
||||
|
return _time; |
||||
|
} |
||||
|
|
||||
|
void setCurrentTime(uint32_t time) override { |
||||
|
_time = time; |
||||
|
} |
||||
|
|
||||
|
private: |
||||
|
uint32_t _time; |
||||
|
}; |
||||
|
|
||||
|
class FakeRng final : public mesh::RNG { |
||||
|
public: |
||||
|
void random(uint8_t* dest, size_t sz) override { |
||||
|
memset(dest, 0x5A, sz); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
class FakeRadio final : public mesh::Radio { |
||||
|
public: |
||||
|
int recvRaw(uint8_t*, int) override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
uint32_t getEstAirtimeFor(int) override { |
||||
|
return 1; |
||||
|
} |
||||
|
|
||||
|
float packetScore(float, int) override { |
||||
|
return 1.0f; |
||||
|
} |
||||
|
|
||||
|
bool startSendRaw(const uint8_t*, int) override { |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
bool isSendComplete() override { |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
void onSendFinished() override {} |
||||
|
|
||||
|
bool isInRecvMode() const override { |
||||
|
return true; |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
class NoopPacketManager final : public mesh::PacketManager { |
||||
|
public: |
||||
|
mesh::Packet* allocNew() override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
void free(mesh::Packet*) override {} |
||||
|
|
||||
|
void queueOutbound(mesh::Packet*, uint8_t, uint32_t) override {} |
||||
|
|
||||
|
mesh::Packet* getNextOutbound(uint32_t) override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
int getOutboundCount(uint32_t) const override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
int getOutboundTotal() const override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
int getFreeCount() const override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
mesh::Packet* getOutboundByIdx(int) override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
mesh::Packet* removeOutboundByIdx(int) override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
void queueInbound(mesh::Packet*, uint32_t) override {} |
||||
|
|
||||
|
mesh::Packet* getNextInbound(uint32_t) override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
class TestChatMesh final : public BaseChatMesh { |
||||
|
public: |
||||
|
TestChatMesh(mesh::Radio& radio, mesh::MillisecondClock& ms, mesh::RNG& rng, mesh::RTCClock& rtc, |
||||
|
mesh::PacketManager& mgr, mesh::MeshTables& tables) |
||||
|
: BaseChatMesh(radio, ms, rng, rtc, mgr, tables) {} |
||||
|
|
||||
|
mesh::DispatcherAction recv(mesh::Packet* pkt) { |
||||
|
return onRecvPacket(pkt); |
||||
|
} |
||||
|
|
||||
|
std::optional<ContactInfo> discovered_contact; |
||||
|
|
||||
|
protected: |
||||
|
void onDiscoveredContact(ContactInfo& contact, bool, uint8_t, const uint8_t*) override { |
||||
|
discovered_contact = contact; |
||||
|
} |
||||
|
|
||||
|
ContactInfo* processAck(const uint8_t*) override { |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
void onContactPathUpdated(const ContactInfo&) override {} |
||||
|
|
||||
|
void onMessageRecv(const ContactInfo&, mesh::Packet*, uint32_t, const char*) override {} |
||||
|
|
||||
|
void onCommandDataRecv(const ContactInfo&, mesh::Packet*, uint32_t, const char*) override {} |
||||
|
|
||||
|
void onSignedMessageRecv(const ContactInfo&, mesh::Packet*, uint32_t, const uint8_t*, const char*) override {} |
||||
|
|
||||
|
uint32_t calcFloodTimeoutMillisFor(uint32_t) const override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
uint32_t calcDirectTimeoutMillisFor(uint32_t, uint8_t) const override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
void onSendTimeout() override {} |
||||
|
|
||||
|
void onChannelMessageRecv(const mesh::GroupChannel&, mesh::Packet*, uint32_t, const char*) override {} |
||||
|
|
||||
|
uint8_t onContactRequest(const ContactInfo&, uint32_t, const uint8_t*, uint8_t, uint8_t*) override { |
||||
|
return 0; |
||||
|
} |
||||
|
|
||||
|
void onContactResponse(const ContactInfo&, const uint8_t*, uint8_t) override {} |
||||
|
}; |
||||
|
|
||||
|
struct TestMeshContext { |
||||
|
explicit TestMeshContext(uint32_t current_timestamp) |
||||
|
: rtc(current_timestamp), mesh(radio, ms, rng, rtc, packet_manager, tables) {} |
||||
|
|
||||
|
TestChatMesh* operator->() { |
||||
|
return &mesh; |
||||
|
} |
||||
|
|
||||
|
const TestChatMesh* operator->() const { |
||||
|
return &mesh; |
||||
|
} |
||||
|
|
||||
|
FakeRadio radio; |
||||
|
FakeMillis ms; |
||||
|
FakeRng rng; |
||||
|
FakeRtc rtc; |
||||
|
NoopPacketManager packet_manager; |
||||
|
SimpleMeshTables tables; |
||||
|
TestChatMesh mesh; |
||||
|
}; |
||||
Loading…
Reference in new issue