添加广播MDNS服务

FactoryTestTool/ThirdParty/qmdnsengine-master/tests/TestBrowser.cpp

@@ -0,0 +1,172 @@
+/*
+ * The MIT License (MIT)
+ *
+ * Copyright (c) 2017 Nathan Osman
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to
+ * deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+ * sell copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ */
+
+#include <QSignalSpy>
+#include <QTest>
+
+#include <qmdnsengine/browser.h>
+#include <qmdnsengine/dns.h>
+#include <qmdnsengine/mdns.h>
+#include <qmdnsengine/message.h>
+#include <qmdnsengine/query.h>
+#include <qmdnsengine/record.h>
+#include <qmdnsengine/service.h>
+
+#include "common/testserver.h"
+#include "common/util.h"
+
+Q_DECLARE_METATYPE(QMdnsEngine::Service)
+
+const QByteArray Name = "Test";
+const QByteArray Type = "_test._tcp.local.";
+const QByteArray Fqdn = Name + "." + Type;
+const QByteArray Target = "Test.local.";
+const quint16 Port = 1234;
+const QByteArray Key = "key";
+const QByteArray Value = "value";
+
+class TestBrowser : public QObject
+{
+    Q_OBJECT
+
+private Q_SLOTS:
+
+    void initTestCase();
+    void testBrowser();
+    void testBrowsePtr();
+};
+
+void TestBrowser::initTestCase()
+{
+    qRegisterMetaType<QMdnsEngine::Service>("Service");
+}
+
+void TestBrowser::testBrowser()
+{
+    TestServer server;
+    QMdnsEngine::Browser browser(&server, Type);
+
+    QSignalSpy serviceAddedSpy(&browser, SIGNAL(serviceAdded(Service)));
+    QSignalSpy serviceUpdatedSpy(&browser, SIGNAL(serviceUpdated(Service)));
+    QSignalSpy serviceRemovedSpy(&browser, SIGNAL(serviceRemoved(Service)));
+
+    // Wait for the PTR query
+    QTRY_VERIFY(queryReceived(&server, Type, QMdnsEngine::PTR));
+    server.clearReceivedMessages();
+
+    // Transmit the PTR record
+    {
+        QMdnsEngine::Record record;
+        record.setName(Type);
+        record.setType(QMdnsEngine::PTR);
+        record.setTarget(Fqdn);
+        QMdnsEngine::Message message;
+        message.setResponse(true);
+        message.addRecord(record);
+        server.deliverMessage(message);
+    }
+
+    // Wait for a SRV query
+    QTRY_VERIFY(queryReceived(&server, Fqdn, QMdnsEngine::SRV));
+    server.clearReceivedMessages();
+
+    // Nothing should have been added yet
+    QCOMPARE(serviceAddedSpy.count(), 0);
+
+    // Transmit the SRV record
+    {
+        QMdnsEngine::Record record;
+        record.setName(Fqdn);
+        record.setType(QMdnsEngine::SRV);
+        record.setTarget(Target);
+        record.setPort(Port);
+        QMdnsEngine::Message message;
+        message.setResponse(true);
+        message.addRecord(record);
+        server.deliverMessage(message);
+    }
+
+    // The serviceAdded signal should have been emitted
+    QCOMPARE(serviceAddedSpy.count(), 1);
+
+    // Transmit a TXT record
+    {
+        QMdnsEngine::Record record;
+        record.setName(Fqdn);
+        record.setType(QMdnsEngine::TXT);
+        record.setAttributes({{Key, Value}});
+        QMdnsEngine::Message message;
+        message.setResponse(true);
+        message.addRecord(record);
+        server.deliverMessage(message);
+    }
+
+    // The serviceAdded signal should NOT have been emitted and the
+    // serviceUpdated signal should have been
+    QCOMPARE(serviceAddedSpy.count(), 1);
+    QCOMPARE(serviceUpdatedSpy.count(), 1);
+
+    // Remove the SRV record
+    {
+        QMdnsEngine::Record record;
+        record.setName(Fqdn);
+        record.setType(QMdnsEngine::SRV);
+        record.setTarget(Target);
+        record.setPort(Port);
+        record.setTtl(0);
+        QMdnsEngine::Message message;
+        message.setResponse(true);
+        message.addRecord(record);
+        server.deliverMessage(message);
+    }
+
+    // The serviceRemoved signal should have been emitted
+    QCOMPARE(serviceRemovedSpy.count(), 1);
+}
+
+void TestBrowser::testBrowsePtr()
+{
+    TestServer server;
+    QMdnsEngine::Browser browser(&server, QMdnsEngine::MdnsBrowseType);
+    Q_UNUSED(browser);
+
+    // Wait for a query for service types
+    QTRY_VERIFY(queryReceived(&server, QMdnsEngine::MdnsBrowseType, QMdnsEngine::PTR));
+
+    // Send a PTR and SRV record
+    QMdnsEngine::Record record;
+    record.setName(QMdnsEngine::MdnsBrowseType);
+    record.setType(QMdnsEngine::PTR);
+    record.setTarget(Type);
+    QMdnsEngine::Message message;
+    message.setResponse(true);
+    message.addRecord(record);
+    server.deliverMessage(message);
+
+    // Wait for the query for records of the specified type
+    QTRY_VERIFY(queryReceived(&server, Type, QMdnsEngine::PTR));
+}
+
+QTEST_MAIN(TestBrowser)
+#include "TestBrowser.moc"