142 lines
3.3 KiB
C++
142 lines
3.3 KiB
C++
#include <AUnit.h>
|
|
#include <TinyMqtt.h>
|
|
#include <map>
|
|
|
|
/**
|
|
* TinyMqtt nowifi unit tests.
|
|
*
|
|
* No wifi connection unit tests.
|
|
* Checks with a local broker. Clients must connect to the local client
|
|
**/
|
|
|
|
using namespace std;
|
|
|
|
MqttBroker broker(1883);
|
|
|
|
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
|
|
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
|
{
|
|
if (srce)
|
|
published[srce->id()][topic]++;
|
|
}
|
|
|
|
test(nowifi_client_should_unregister_when_destroyed)
|
|
{
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
{
|
|
MqttClient client(&broker);
|
|
assertEqual(broker.clientsCount(), (size_t)1);
|
|
}
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
}
|
|
|
|
test(nowifi_connect)
|
|
{
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
|
|
MqttClient client(&broker);
|
|
assertTrue(client.connected());
|
|
assertEqual(broker.clientsCount(), (size_t)1);
|
|
}
|
|
|
|
test(nowifi_publish_should_be_dispatched)
|
|
{
|
|
published.clear();
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
|
|
MqttClient subscriber(&broker);
|
|
subscriber.subscribe("a/b");
|
|
subscriber.subscribe("a/c");
|
|
subscriber.setCallback(onPublish);
|
|
|
|
MqttClient publisher(&broker);
|
|
publisher.publish("a/b");
|
|
publisher.publish("a/c");
|
|
publisher.publish("a/c");
|
|
|
|
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
|
assertTrue(published[""]["a/b"] == 1);
|
|
assertTrue(published[""]["a/c"] == 2);
|
|
}
|
|
|
|
test(nowifi_publish_should_be_dispatched_to_nowifi_clients)
|
|
{
|
|
published.clear();
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
|
|
MqttClient subscriber_a(&broker, "A");
|
|
subscriber_a.setCallback(onPublish);
|
|
subscriber_a.subscribe("a/b");
|
|
subscriber_a.subscribe("a/c");
|
|
|
|
MqttClient subscriber_b(&broker, "B");
|
|
subscriber_b.setCallback(onPublish);
|
|
subscriber_b.subscribe("a/b");
|
|
|
|
MqttClient publisher(&broker);
|
|
publisher.publish("a/b");
|
|
publisher.publish("a/c");
|
|
|
|
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
|
assertTrue(published["A"]["a/b"] == 1);
|
|
assertTrue(published["A"]["a/c"] == 1);
|
|
assertTrue(published["B"]["a/b"] == 1);
|
|
assertTrue(published["B"]["a/c"] == 0);
|
|
}
|
|
|
|
test(nowifi_unsubscribe)
|
|
{
|
|
published.clear();
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
|
|
MqttClient subscriber(&broker);
|
|
subscriber.setCallback(onPublish);
|
|
subscriber.subscribe("a/b");
|
|
|
|
MqttClient publisher(&broker);
|
|
publisher.publish("a/b");
|
|
|
|
subscriber.unsubscribe("a/b");
|
|
|
|
publisher.publish("a/b");
|
|
publisher.publish("a/b");
|
|
|
|
assertTrue(published[""]["a/b"] == 1); // Only one publish has been received
|
|
}
|
|
|
|
test(nowifi_nocallback_when_destroyed)
|
|
{
|
|
published.clear();
|
|
assertEqual(broker.clientsCount(), (size_t)0);
|
|
|
|
MqttClient publisher(&broker);
|
|
|
|
{
|
|
MqttClient subscriber(&broker);
|
|
subscriber.setCallback(onPublish);
|
|
subscriber.subscribe("a/b");
|
|
publisher.publish("a/b");
|
|
}
|
|
|
|
publisher.publish("a/b");
|
|
|
|
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// setup() and loop()
|
|
void setup() {
|
|
delay(1000);
|
|
Serial.begin(115200);
|
|
while(!Serial);
|
|
|
|
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
|
}
|
|
|
|
void loop() {
|
|
aunit::TestRunner::run();
|
|
|
|
if (Serial.available()) ESP.reset();
|
|
}
|