TrustMemoryStorage: Add security policies

1 files changed, 36 insertions, 0 deletions

diff --git a/tests/qxmpptrustmemorystorage/tst_qxmpptrustmemorystorage.cpp b/tests/qxmpptrustmemorystorage/tst_qxmpptrustmemorystorage.cpp
index 63932f07..9d803752 100644
--- a/tests/qxmpptrustmemorystorage/tst_qxmpptrustmemorystorage.cpp
+++ b/tests/qxmpptrustmemorystorage/tst_qxmpptrustmemorystorage.cpp
@@ -33,6 +33,7 @@ class tst_QXmppTrustMemoryStorage : public QObject
     Q_OBJECT
 
 private slots:
+    void testSecurityPolicies();
     void testOwnKeys();
     void testKeys();
     void testTrustLevels();
@@ -42,6 +43,41 @@ private:
     QXmppTrustMemoryStorage m_trustStorage;
 };
 
+void tst_QXmppTrustMemoryStorage::testSecurityPolicies()
+{
+    auto future = m_trustStorage.securityPolicy(ns_ox);
+    QVERIFY(future.isFinished());
+    auto result = future.result();
+    QCOMPARE(result, QXmppTrustStorage::NoSecurityPolicy);
+
+    m_trustStorage.setSecurityPolicies(ns_ox, QXmppTrustStorage::Toakafa);
+    m_trustStorage.setSecurityPolicies(ns_omemo, QXmppTrustStorage::Toakafa);
+
+    future = m_trustStorage.securityPolicy(ns_ox);
+    QVERIFY(future.isFinished());
+    result = future.result();
+    QCOMPARE(result, QXmppTrustStorage::Toakafa);
+
+    future = m_trustStorage.securityPolicy(ns_omemo);
+    QVERIFY(future.isFinished());
+    result = future.result();
+    QCOMPARE(result, QXmppTrustStorage::Toakafa);
+
+    m_trustStorage.setSecurityPolicies(ns_ox);
+
+    future = m_trustStorage.securityPolicy(ns_ox);
+    QVERIFY(future.isFinished());
+    result = future.result();
+    QCOMPARE(result, QXmppTrustStorage::NoSecurityPolicy);
+
+    m_trustStorage.setSecurityPolicies();
+
+    future = m_trustStorage.securityPolicy(ns_omemo);
+    QVERIFY(future.isFinished());
+    result = future.result();
+    QCOMPARE(result, QXmppTrustStorage::NoSecurityPolicy);
+}
+
 void tst_QXmppTrustMemoryStorage::testOwnKeys()
 {
     m_trustStorage.addOwnKey(ns_ox, QStringLiteral("6850019d7ed0feb6d3823072498ceb4f616c6025586f8f666dc6b9c81ef7e0a4"));