import unittest
from slixmpp import Message
from slixmpp.test import SlixTest
from slixmpp.xmlstream import register_stanza_plugin

from slixmpp.plugins.xep_0356 import stanza


class TestPermissions(SlixTest):
    def setUp(self):
        stanza.register()

    def testAdvertisePermission(self):
        xmlstring = """
            <message from='capulet.net' to='pubub.capulet.lit'>
                <privilege xmlns='urn:xmpp:privilege:2'>
                    <perm access='roster' type='both'/>
                    <perm access='message' type='outgoing'/>
                    <perm access='presence' type='managed_entity'/>
                </privilege>
            </message>
        """
        msg = self.Message()
        msg["from"] = "capulet.net"
        msg["to"] = "pubub.capulet.lit"
        # This raises AttributeError: 'NoneType' object has no attribute 'use_origin_id'
        # msg["id"] = "id"

        for access, type_ in [
            ("roster", "both"),
            ("message", "outgoing"),
            ("presence", "managed_entity"),
        ]:
            msg["privilege"].add_perm(access, type_)

        self.check(msg, xmlstring)
        # Should this one work? →        # AttributeError: 'Message' object has no attribute 'permission'
        # self.assertEqual(msg.permission["roster"], "both")


suite = unittest.TestLoader().loadTestsFromTestCase(TestPermissions)