初始化提交

arduino-cli/libraries/painlessMesh-master/test/catch/catch_plugin.cpp

@@ -0,0 +1,174 @@
+#define CATCH_CONFIG_MAIN
+
+#include "catch2/catch.hpp"
+
+#include "Arduino.h"
+
+#include "catch_utils.hpp"
+
+#include "painlessmesh/plugin.hpp"
+#include "plugin/performance.hpp"
+
+using namespace painlessmesh;
+
+logger::LogClass Log;
+
+class CustomPackage : public plugin::SinglePackage {
+ public:
+  double sensor = 1.0;
+
+  CustomPackage() : SinglePackage(20) {}
+
+  CustomPackage(JsonObject jsonObj) : SinglePackage(jsonObj) {
+    sensor = jsonObj["sensor"];
+  }
+
+  JsonObject addTo(JsonObject&& jsonObj) const {
+    jsonObj = SinglePackage::addTo(std::move(jsonObj));
+    jsonObj["sensor"] = sensor;
+    return jsonObj;
+  }
+
+  size_t jsonObjectSize() const { return JSON_OBJECT_SIZE(noJsonFields + 1); }
+};
+
+class BCustomPackage : public plugin::BroadcastPackage {
+ public:
+  double sensor = 1.0;
+
+  BCustomPackage() : BroadcastPackage(21) {}
+
+  BCustomPackage(JsonObject jsonObj) : BroadcastPackage(jsonObj) {
+    sensor = jsonObj["sensor"];
+  }
+
+  JsonObject addTo(JsonObject&& jsonObj) const {
+    jsonObj = BroadcastPackage::addTo(std::move(jsonObj));
+    jsonObj["sensor"] = sensor;
+    return jsonObj;
+  }
+
+  size_t jsonObjectSize() const { return JSON_OBJECT_SIZE(noJsonFields + 1); }
+};
+
+class MockConnection : public layout::Neighbour {
+ public:
+  bool addMessage(TSTRING msg) { return true; }
+};
+
+SCENARIO("We can send a custom package") {
+  GIVEN("A package") {
+    auto pkg = CustomPackage();
+    pkg.from = 1;
+    pkg.dest = 2;
+    pkg.sensor = 0.5;
+    REQUIRE(pkg.routing == router::SINGLE);
+    REQUIRE(pkg.type == 20);
+    WHEN("Converting it to and from Variant") {
+      auto var = protocol::Variant(&pkg);
+      auto pkg2 = var.to<CustomPackage>();
+      THEN("Should result in the same values") {
+        REQUIRE(pkg2.sensor == pkg.sensor);
+        REQUIRE(pkg2.from == pkg.from);
+        REQUIRE(pkg2.dest == pkg.dest);
+        REQUIRE(pkg2.routing == pkg.routing);
+        REQUIRE(pkg2.type == pkg.type);
+      }
+    }
+  }
+
+  GIVEN("A broadcast package") {
+    auto pkg = BCustomPackage();
+    pkg.from = 1;
+    pkg.sensor = 0.5;
+    REQUIRE(pkg.routing == router::BROADCAST);
+    REQUIRE(pkg.type == 21);
+    WHEN("Converting it to and from Variant") {
+      auto var = protocol::Variant(&pkg);
+      auto pkg2 = var.to<CustomPackage>();
+      THEN("Should result in the same values") {
+        REQUIRE(pkg2.sensor == pkg.sensor);
+        REQUIRE(pkg2.from == pkg.from);
+        REQUIRE(pkg2.routing == pkg.routing);
+        REQUIRE(pkg2.type == pkg.type);
+      }
+    }
+  }
+
+  GIVEN("A package handler function") {
+    auto handler = plugin::PackageHandler<MockConnection>();
+    auto func = [](protocol::Variant variant) {
+      auto pkg = variant.to<CustomPackage>();
+      REQUIRE(pkg.routing == router::BROADCAST);
+      return false;
+    };
+    THEN("We can pass it to handler") { handler.onPackage(20, func); }
+  }
+
+  GIVEN("A package") {
+    auto handler = plugin::PackageHandler<MockConnection>();
+    auto pkg = CustomPackage();
+    THEN("We can call sendPackage") {
+      auto res = handler.sendPackage(&pkg);
+      REQUIRE(!res);
+    }
+  }
+}
+
+SCENARIO("We can add tasks to the taskscheduler") {
+  GIVEN("A couple of tasks added") {
+    Scheduler mScheduler;
+    auto handler = plugin::PackageHandler<MockConnection>();
+    int i = 0;
+    int j = 0;
+    int k = 0;
+    auto task1 = handler.addTask(mScheduler, 0, 1, [&i]() { ++i; });
+    auto task2 = handler.addTask(mScheduler, 0, 3, [&j]() { ++j; });
+    auto task3 = handler.addTask(mScheduler, 0, 3, [&k]() { ++k; });
+    auto task4 = handler.addTask(mScheduler, 0, 3, []() {});
+
+    WHEN("Executing the tasks") {
+      THEN("They should be called and automatically removed") {
+        REQUIRE(i == 0);
+        REQUIRE(j == 0);
+        mScheduler.execute();
+        REQUIRE(i == 1);
+        mScheduler.execute();
+        REQUIRE(i == 1);
+        REQUIRE(j == 2);
+        // Still kept in handler, because hasn't been executed 3 times yet
+        task3->disable();
+        handler.stop();
+      }
+    }
+  }
+}
+
+SCENARIO("We can add anonymous tasks to the taskscheduler") {
+  GIVEN("A couple of tasks added") {
+    Scheduler mScheduler;
+    auto handler = plugin::PackageHandler<MockConnection>();
+    int i = 0;
+    int j = 0;
+    int k = 0;
+    handler.addTask(mScheduler, 0, 1, [&i]() { ++i; });
+    handler.addTask(mScheduler, 0, 3, [&j]() { ++j; });
+    handler.addTask(mScheduler, 0, 3, [&k]() { ++k; });
+
+    WHEN("Executing the tasks") {
+      THEN("They should be called and automatically removed") {
+        REQUIRE(i == 0);
+        REQUIRE(j == 0);
+        mScheduler.execute();
+        REQUIRE(i == 1);
+        mScheduler.execute();
+        REQUIRE(i == 1);
+        REQUIRE(j == 2);
+        handler.addTask(mScheduler, 0, 1, [&i]() { ++i; });
+        mScheduler.execute();
+        REQUIRE(i == 2);
+        handler.stop();
+      }
+    }
+  }
+}