Minor changes
This commit is contained in:
@@ -38,6 +38,7 @@
|
|||||||
4. Simple _FMT mechanism ala printf, but without the typeunsafetyness
|
4. Simple _FMT mechanism ala printf, but without the typeunsafetyness
|
||||||
and no internal buffers for replaceable stream printing
|
and no internal buffers for replaceable stream printing
|
||||||
*/
|
*/
|
||||||
|
#pragma once
|
||||||
|
|
||||||
#ifndef ARDUINO_STREAMING
|
#ifndef ARDUINO_STREAMING
|
||||||
#define ARDUINO_STREAMING
|
#define ARDUINO_STREAMING
|
||||||
|
|||||||
@@ -84,7 +84,7 @@ void MqttClient::connect(MqttBroker* parentBroker)
|
|||||||
|
|
||||||
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
||||||
{
|
{
|
||||||
debug("cnx: closing");
|
debug("MqttClient::connect");
|
||||||
keep_alive = ka;
|
keep_alive = ka;
|
||||||
close();
|
close();
|
||||||
if (client) delete client;
|
if (client) delete client;
|
||||||
|
|||||||
@@ -235,7 +235,7 @@ class MqttClient
|
|||||||
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
||||||
|
|
||||||
void clientAlive(uint32_t more_seconds);
|
void clientAlive(uint32_t more_seconds);
|
||||||
void processMessage(MqttMessage* message);
|
void processMessage(const MqttMessage* message);
|
||||||
|
|
||||||
bool mqtt_connected = false;
|
bool mqtt_connected = false;
|
||||||
char mqtt_flags;
|
char mqtt_flags;
|
||||||
|
|||||||
@@ -143,7 +143,7 @@ void setup() {
|
|||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
while(!Serial);
|
while(!Serial);
|
||||||
|
|
||||||
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
Serial.println("=============[ LOCAL TinyMqtt TESTS ]========================");
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop() {
|
void loop() {
|
||||||
|
|||||||
13
tests/network-tests/Makefile
Normal file
13
tests/network-tests/Makefile
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
EXTRA_CXXFLAGS=-g3 -O0
|
||||||
|
|
||||||
|
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||||
|
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||||
|
|
||||||
|
APP_NAME := network-tests
|
||||||
|
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP
|
||||||
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
236
tests/network-tests/network-tests.ino
Normal file
236
tests/network-tests/network-tests.ino
Normal file
@@ -0,0 +1,236 @@
|
|||||||
|
#include <Arduino.h>
|
||||||
|
#include <AUnit.h>
|
||||||
|
#include <TinyMqtt.h>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TinyMqtt network unit tests.
|
||||||
|
*
|
||||||
|
* No wifi connection unit tests.
|
||||||
|
* Checks with a local broker. Clients must connect to the local broker
|
||||||
|
**/
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
|
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||||
|
|
||||||
|
char* lastPayload = nullptr;
|
||||||
|
size_t lastLength;
|
||||||
|
|
||||||
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||||
|
{
|
||||||
|
if (srce)
|
||||||
|
published[srce->id()][topic]++;
|
||||||
|
|
||||||
|
if (lastPayload) free(lastPayload);
|
||||||
|
lastPayload = strdup(payload);
|
||||||
|
lastLength = length;
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_single_broker_begin)
|
||||||
|
{
|
||||||
|
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
broker.begin();
|
||||||
|
|
||||||
|
// TODO Nothing is tested here !
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_client_to_broker_connexion)
|
||||||
|
{
|
||||||
|
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
broker.begin();
|
||||||
|
|
||||||
|
Serial << "TODO IP = " << WiFi.localIP() << endl;
|
||||||
|
|
||||||
|
MqttClient client;
|
||||||
|
client.connect(WiFi.localIP().toString().c_str(), 1883);
|
||||||
|
broker.loop();
|
||||||
|
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
test(network_one_broker_one_client)
|
||||||
|
{
|
||||||
|
published.clear();
|
||||||
|
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
broker.begin();
|
||||||
|
|
||||||
|
Serial << "TODO IP = " << WiFi.localIP() << endl;
|
||||||
|
|
||||||
|
MqttClient client;
|
||||||
|
client.connect(WiFi.localIP().toString().c_str(), 1883);
|
||||||
|
client.setCallback(onPublish);
|
||||||
|
client.subscribe("a/b");
|
||||||
|
client.publish("a/b", "ab");
|
||||||
|
// client.loop();
|
||||||
|
|
||||||
|
assertEqual(published.size(), (size_t)1);
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
test(network_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(network_connect)
|
||||||
|
{
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
|
MqttClient client(&broker);
|
||||||
|
assertTrue(client.connected());
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_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
|
||||||
|
assertEqual(published[""]["a/b"], 1);
|
||||||
|
assertEqual(published[""]["a/c"], 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_publish_should_be_dispatched_to_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"); // A and B should receive this
|
||||||
|
publisher.publish("a/c"); // A should receive this
|
||||||
|
|
||||||
|
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
||||||
|
assertEqual(published["A"]["a/b"], 1);
|
||||||
|
assertEqual(published["A"]["a/c"], 1);
|
||||||
|
assertEqual(published["B"]["a/b"], 1);
|
||||||
|
assertEqual(published["B"]["a/c"], 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_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"); // This publish is received
|
||||||
|
|
||||||
|
subscriber.unsubscribe("a/b");
|
||||||
|
|
||||||
|
publisher.publish("a/b"); // Those one, no (unsubscribed)
|
||||||
|
publisher.publish("a/b");
|
||||||
|
|
||||||
|
assertEqual(published[""]["a/b"], 1); // Only one publish has been received
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_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
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_small_payload)
|
||||||
|
{
|
||||||
|
published.clear();
|
||||||
|
|
||||||
|
const char* payload="abcd";
|
||||||
|
|
||||||
|
MqttClient subscriber(&broker);
|
||||||
|
subscriber.setCallback(onPublish);
|
||||||
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
|
MqttClient publisher(&broker);
|
||||||
|
publisher.publish("a/b", payload, strlen(payload)); // This publish is received
|
||||||
|
|
||||||
|
// coming from MqttClient::publish(...)
|
||||||
|
assertEqual(payload, lastPayload);
|
||||||
|
assertEqual(lastLength, (size_t)4);
|
||||||
|
}
|
||||||
|
|
||||||
|
test(network_hudge_payload)
|
||||||
|
{
|
||||||
|
const char* payload="This payload is hudge, just because its length exceeds 127. Thus when encoding length, we have to encode it on two bytes at min. This should not prevent the message from being encoded and decoded successfully !";
|
||||||
|
|
||||||
|
MqttClient subscriber(&broker);
|
||||||
|
subscriber.setCallback(onPublish);
|
||||||
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
|
MqttClient publisher(&broker);
|
||||||
|
publisher.publish("a/b", payload); // This publish is received
|
||||||
|
|
||||||
|
// onPublish should have filled lastPayload and lastLength
|
||||||
|
assertEqual(payload, lastPayload);
|
||||||
|
assertEqual(lastLength, strlen(payload));
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
//----------------------------------------------------------------------------
|
||||||
|
// setup() and loop()
|
||||||
|
void setup() {
|
||||||
|
delay(1000);
|
||||||
|
Serial.begin(115200);
|
||||||
|
while(!Serial);
|
||||||
|
|
||||||
|
Serial.println("=============[ FAKE NETWORK TinyMqtt TESTS ]========================");
|
||||||
|
|
||||||
|
WiFi.mode(WIFI_STA);
|
||||||
|
WiFi.begin("network", "password");
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop() {
|
||||||
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
|
if (Serial.available()) ESP.reset();
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user