diff sat/test/test_plugin_xep_0085.py @ 2624:56f94936df1e

code style reformatting using black
author Goffi <goffi@goffi.org>
date Wed, 27 Jun 2018 20:14:46 +0200
parents 26edcf3a30eb
children 378188abe941
line wrap: on
line diff
--- a/sat/test/test_plugin_xep_0085.py	Wed Jun 27 07:51:29 2018 +0200
+++ b/sat/test/test_plugin_xep_0085.py	Wed Jun 27 20:14:46 2018 +0200
@@ -30,11 +30,16 @@
 
 
 class XEP_0085Test(helpers.SatTestCase):
-
     def setUp(self):
         self.host = helpers.FakeSAT()
         self.plugin = plugin.XEP_0085(self.host)
-        self.host.memory.setParam(plugin.PARAM_NAME, True, plugin.PARAM_KEY, C.NO_SECURITY_LIMIT, Const.PROFILE[0])
+        self.host.memory.setParam(
+            plugin.PARAM_NAME,
+            True,
+            plugin.PARAM_KEY,
+            C.NO_SECURITY_LIMIT,
+            Const.PROFILE[0],
+        )
 
     def test_messageReceived(self):
         for state in plugin.CHAT_STATES:
@@ -43,37 +48,51 @@
             %s
             <%s xmlns='%s'/>
             </message>
-            """ % (Const.JID_STR[1],
-                   Const.JID_STR[0],
-                   "<body>test</body>" if state == "active" else "",
-                   state, plugin.NS_CHAT_STATES)
+            """ % (
+                Const.JID_STR[1],
+                Const.JID_STR[0],
+                "<body>test</body>" if state == "active" else "",
+                state,
+                plugin.NS_CHAT_STATES,
+            )
             stanza = parseXml(xml.encode("utf-8"))
-            self.host.bridge.expectCall("chatStateReceived", Const.JID_STR[1], state, Const.PROFILE[0])
-            self.plugin.messageReceivedTrigger(self.host.getClient(Const.PROFILE[0]), stanza, None)
+            self.host.bridge.expectCall(
+                "chatStateReceived", Const.JID_STR[1], state, Const.PROFILE[0]
+            )
+            self.plugin.messageReceivedTrigger(
+                self.host.getClient(Const.PROFILE[0]), stanza, None
+            )
 
     def test_messageSendTrigger(self):
         def cb(data):
-            xml = data['xml'].toXml().encode("utf-8")
+            xml = data["xml"].toXml().encode("utf-8")
             self.assertEqualXML(xml, expected.toXml().encode("utf-8"))
 
         d_list = []
 
         for state in plugin.CHAT_STATES:
-            mess_data = {"to": Const.JID[0],
-                         "type": "chat",
-                         "message": "content",
-                         "extra": {} if state == "active" else {"chat_state": state}}
+            mess_data = {
+                "to": Const.JID[0],
+                "type": "chat",
+                "message": "content",
+                "extra": {} if state == "active" else {"chat_state": state},
+            }
             stanza = u"""
             <message type="chat" from="%s" to="%s" id="test_1">
             %s
             </message>
-            """ % (Const.JID_STR[1], Const.JID_STR[0],
-                   ("<body>%s</body>" % mess_data['message']) if state == "active" else "")
-            mess_data['xml'] = parseXml(stanza.encode("utf-8"))
-            expected = deepcopy(mess_data['xml'])
+            """ % (
+                Const.JID_STR[1],
+                Const.JID_STR[0],
+                ("<body>%s</body>" % mess_data["message"]) if state == "active" else "",
+            )
+            mess_data["xml"] = parseXml(stanza.encode("utf-8"))
+            expected = deepcopy(mess_data["xml"])
             expected.addElement(state, plugin.NS_CHAT_STATES)
             post_treatments = defer.Deferred()
-            self.plugin.messageSendTrigger(self.host.getClient(Const.PROFILE[0]), mess_data, None, post_treatments)
+            self.plugin.messageSendTrigger(
+                self.host.getClient(Const.PROFILE[0]), mess_data, None, post_treatments
+            )
 
             post_treatments.addCallback(cb)
             post_treatments.callback(mess_data)