test: add snapshot tests for MessageBuilder (#5598)

src/providers/ffz/FfzBadges.cpp

@@ -1,5 +1,6 @@
 #include "providers/ffz/FfzBadges.hpp"
 
+#include "Application.hpp"
 #include "common/network/NetworkRequest.hpp"
 #include "common/network/NetworkResult.hpp"
 #include "messages/Emote.hpp"
@@ -109,4 +110,30 @@ void FfzBadges::load()
         .execute();
 }
 
+void FfzBadges::registerBadge(int badgeID, Badge badge)
+{
+    assert(getApp()->isTest());
+
+    std::unique_lock lock(this->mutex_);
+
+    this->badges.emplace(badgeID, std::move(badge));
+}
+
+void FfzBadges::assignBadgeToUser(const UserId &userID, int badgeID)
+{
+    assert(getApp()->isTest());
+
+    std::unique_lock lock(this->mutex_);
+
+    auto it = this->userBadges.find(userID.string);
+    if (it != this->userBadges.end())
+    {
+        it->second.emplace(badgeID);
+    }
+    else
+    {
+        this->userBadges.emplace(userID.string, std::set{badgeID});
+    }
+}
+
 }  // namespace chatterino

src/providers/ffz/FfzBadges.hpp

@@ -30,6 +30,9 @@ public:
     std::vector<Badge> getUserBadges(const UserId &id);
     std::optional<Badge> getBadge(int badgeID) const;
 
+    void registerBadge(int badgeID, Badge badge);
+    void assignBadgeToUser(const UserId &userID, int badgeID);
+
     void load();
 
 private: