def test_basic_publish(self):
reactor, router, server_factory, session_factory = build_mqtt_server()
session, pump = connect_application_session(
server_factory, ObservingSession, component_config=ComponentConfig(realm=u"mqtt"))
client_transport, client_protocol, mqtt_pump = connect_mqtt_server(server_factory)
client_transport.write(
Connect(client_id=u"testclient", username=u"test123", password=u"password",
flags=ConnectFlags(clean_session=False, username=True, password=True)).serialise())
mqtt_pump.flush()
# We get a CONNECT
self.assertEqual(client_protocol.data,
ConnACK(session_present=False, return_code=0).serialise())
client_protocol.data = b""
client_transport.write(
Publish(duplicate=False, qos_level=0, retain=False, topic_name=u"test", payload=b'{"kwargs": {"bar": "baz"}}').serialise())
mqtt_pump.flush()
pump.flush()
# This needs to be replaced with the real deal, see https://github.com/crossbario/crossbar/issues/885
self.assertEqual(len(session.events), 1)
self.assertEqual(
session.events,
[{"args": tuple(),
"kwargs": {u'bar': u'baz'}}])
def test_self_subscribe(host, port):
record = [
Frame(send=True,
data=Connect(client_id=u"test_selfsub",
flags=ConnectFlags(clean_session=True))),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(send=True,
data=Subscribe(
packet_identifier=1234,
topic_requests=[SubscriptionTopicRequest(u"foo", 2)])),
Frame(send=False,
data=SubACK(packet_identifier=1234, return_codes=[2])),
Frame(send=True,
data=Publish(duplicate=False,
qos_level=0,
topic_name=u"foo",
payload=b"abc",
retain=False)),
Frame(send=False,
data=Publish(duplicate=False,
qos_level=0,
topic_name=u"foo",
payload=b"abc",
retain=False)),
Frame(send=True, data=Disconnect()),
ConnectionLoss(),
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("self_subscribe", f.success, f.reason, f.client_transcript)
def _test_retained(self):
"""
The MQTT client can set and receive retained messages.
"""
reactor, router, server_factory, session_factory = build_mqtt_server()
client_transport, client_protocol, mqtt_pump = connect_mqtt_server(
server_factory)
client_transport.write(
Connect(client_id=u"testclient",
username=u"test123",
password=u"password",
flags=ConnectFlags(clean_session=False,
username=True,
password=True)).serialise())
client_transport.write(
Publish(duplicate=False,
qos_level=1,
retain=True,
topic_name=u"com/test/wamp",
packet_identifier=123,
payload=b'{}').serialise())
mqtt_pump.flush()
self.assertEqual(
client_protocol.data,
(ConnACK(session_present=False, return_code=0).serialise() +
PubACK(packet_identifier=123).serialise()))
client_protocol.data = b""
client_transport.write(
Subscribe(packet_identifier=1,
topic_requests=[
SubscriptionTopicRequest(
topic_filter=u"com/test/wamp", max_qos=0)
]).serialise())
mqtt_pump.flush()
self.assertEqual(
client_protocol.data,
SubACK(packet_identifier=1, return_codes=[0]).serialise())
client_protocol.data = b""
reactor.advance(0.1)
mqtt_pump.flush()
# This needs to be replaced with the real deal, see https://github.com/crossbario/crossbar/issues/885
self.assertEqual(
client_protocol.data,
Publish(duplicate=False,
qos_level=0,
retain=True,
topic_name=u"com/test/wamp",
payload=json.dumps(
{}, sort_keys=True).encode('utf8')).serialise())
class TestProtocol(Protocol):
data = b""
expected = (
ConnACK(session_present=False, return_code=1).serialise())
def dataReceived(self_, data):
self_.data = self_.data + data
if len(self_.data) == len(self_.expected):
self.assertEqual(self_.data, self_.expected)
real_reactor.stop()
def _test_basic_subscribe(self):
"""
The MQTT client can subscribe to a WAMP topic and get messages.
"""
reactor, router, server_factory, session_factory = build_mqtt_server()
client_transport, client_protocol, mqtt_pump = connect_mqtt_server(
server_factory)
session, pump = connect_application_session(
server_factory,
ApplicationSession,
component_config=ComponentConfig(realm=u"mqtt"))
client_transport.write(
Connect(client_id=u"testclient",
username=u"test123",
password=u"password",
flags=ConnectFlags(clean_session=False,
username=True,
password=True)).serialise())
client_transport.write(
Subscribe(packet_identifier=1,
topic_requests=[
SubscriptionTopicRequest(
topic_filter=u"com/test/wamp", max_qos=0)
]).serialise())
mqtt_pump.flush()
self.assertEqual(
client_protocol.data,
(ConnACK(session_present=False, return_code=0).serialise() +
SubACK(packet_identifier=1, return_codes=[0]).serialise()))
client_protocol.data = b""
session.publish(u"com.test.wamp", u"bar")
pump.flush()
reactor.advance(0.1)
mqtt_pump.flush()
self.assertEqual(
client_protocol.data,
Publish(duplicate=False,
qos_level=0,
retain=False,
topic_name=u"com/test/wamp",
payload=b'{"args":["bar"]}').serialise())
def test_quirks_mode_connect(host, port):
record = [
Frame(
send=True,
data=b"\x10\x15\x00\x04MQTT\x04\x02\x00x\x00\x07testqrk\x00\x00"),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(send=True, data=Disconnect()),
ConnectionLoss(),
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("connect_quirks", f.success, f.reason, f.client_transcript)
def test_connect(host, port):
record = [
Frame(send=True,
data=Connect(client_id=u"test_cleanconnect",
flags=ConnectFlags(clean_session=True))),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(send=True, data=Disconnect()),
ConnectionLoss(),
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("connect", f.success, f.reason, f.client_transcript)
def test_lastwill(self):
"""
The MQTT client can set a last will message which will be published
when it disconnects.
"""
reactor, router, server_factory, session_factory = build_mqtt_server()
session, pump = connect_application_session(
server_factory,
ObservingSession,
component_config=ComponentConfig(realm=u"mqtt"))
client_transport, client_protocol, mqtt_pump = connect_mqtt_server(
server_factory)
client_transport.write(
Connect(client_id=u"testclient",
username=u"test123",
password=u"password",
will_topic=u"test",
will_message=b'{"args": ["foobar"]}',
flags=ConnectFlags(clean_session=False,
username=True,
password=True,
will=True)).serialise())
mqtt_pump.flush()
# We get a CONNECT
self.assertEqual(
client_protocol.data,
ConnACK(session_present=False, return_code=0).serialise())
client_protocol.data = b""
client_transport.write(Disconnect().serialise())
mqtt_pump.flush()
pump.flush()
self.assertEqual(client_transport.disconnected, True)
# This needs to be replaced with the real deal, see https://github.com/crossbario/crossbar/issues/885
self.assertEqual(len(session.events), 1)
self.assertEqual(session.events, [{
"args": (u"foobar", ),
"kwargs": {}
}])
def test_qos1_send_wrong_confirm(host, port):
record = [
Frame(send=True,
data=Connect(client_id=u"test_wrong_confirm_qos1",
flags=ConnectFlags(clean_session=True))),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(send=True,
data=Subscribe(
packet_identifier=1234,
topic_requests=[SubscriptionTopicRequest(u"foo", 2)])),
Frame(send=False,
data=SubACK(packet_identifier=1234, return_codes=[2])),
Frame(send=True,
data=Publish(duplicate=False,
qos_level=1,
topic_name=u"foo",
payload=b"abc",
retain=False,
packet_identifier=12)),
Frame(send=False,
data=[
PubACK(packet_identifier=12),
Publish(duplicate=False,
qos_level=1,
topic_name=u"foo",
payload=b"abc",
retain=False,
packet_identifier=1)
]),
# We send a pubrel to the packet_id expecting a puback
Frame(send=True, data=PubREL(packet_identifier=1)),
# ..aaaaand we get a pubcomp back (even though mosquitto warns).
Frame(send=False, data=PubCOMP(packet_identifier=1)),
Frame(send=True, data=Disconnect()),
ConnectionLoss(),
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("qos1_wrong_confirm", f.success, f.reason,
f.client_transcript)
def test_uninvited_pubrel(host, port):
record = [
Frame(send=True,
data=Connect(client_id=u"test_pubrel",
flags=ConnectFlags(clean_session=True))),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(send=True, data=PubREL(packet_identifier=1234)),
Frame(send=False, data=PubCOMP(packet_identifier=1234)),
Frame(send=True, data=Disconnect()),
ConnectionLoss(),
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("uninvited_pubrel", f.success, f.reason, f.client_transcript)
def test_reserved_packet_15(host, port):
record = [
Frame(send=True,
data=Connect(client_id=u"test_reserved15",
flags=ConnectFlags(clean_session=True))),
Frame(send=False, data=ConnACK(session_present=False, return_code=0)),
Frame(
send=True,
# v pkt 15 right here
data=b"\xf0\x13\x00\x04MQTT\x04\x02\x00\x02\x00\x07test123"),
ConnectionLoss()
]
r = SelectReactor()
f = ReplayClientFactory(r, record)
e = TCP4ClientEndpoint(r, host, port)
e.connect(f)
r.run()
return Result("reserved_pkt15", f.success, f.reason, f.client_transcript)
def test_tls_auth_denied(self):
"""
A MQTT client offering the wrong certificate won't be authenticated.
"""
reactor, router, server_factory, session_factory = build_mqtt_server()
real_reactor = selectreactor.SelectReactor()
logger = make_logger()
session, pump = connect_application_session(
server_factory, ObservingSession, component_config=ComponentConfig(realm=u"mqtt"))
endpoint = create_listening_endpoint_from_config({
"type": "tcp",
"port": 1099,
"interface": "0.0.0.0",
"tls": {
"certificate": "server.crt",
"key": "server.key",
"dhparam": "dhparam",
"ca_certificates": [
"ca.cert.pem",
"intermediate.cert.pem"
]},
}, FilePath(__file__).sibling('certs').path, real_reactor, logger)
client_endpoint = create_connecting_endpoint_from_config({
"type": "tcp",
"host": "127.0.0.1",
"port": 1099,
"tls": {
# BAD key: trusted by the CA, but wrong ID
"certificate": "client_1.crt",
"hostname": u"localhost",
"key": "client_1.key",
"ca_certificates": [
"ca.cert.pem",
"intermediate.cert.pem"
]},
}, FilePath(__file__).sibling('certs').path, real_reactor, logger)
p = []
l = endpoint.listen(server_factory)
class TestProtocol(Protocol):
data = b""
expected = (
ConnACK(session_present=False, return_code=1).serialise())
def dataReceived(self_, data):
self_.data = self_.data + data
if len(self_.data) == len(self_.expected):
self.assertEqual(self_.data, self_.expected)
real_reactor.stop()
@l.addCallback
def _listening(factory):
d = client_endpoint.connect(Factory.forProtocol(TestProtocol))
@d.addCallback
def _(proto):
p.append(proto)
proto.transport.write(
Connect(client_id=u"test123",
flags=ConnectFlags(clean_session=False)).serialise())
proto.transport.write(
Publish(duplicate=False, qos_level=1, retain=False, topic_name=u"test", payload=b"{}", packet_identifier=1).serialise())
lc = LoopingCall(pump.flush)
lc.clock = real_reactor
lc.start(0.01)
def timeout():
print("Timing out :(")
real_reactor.stop()
print(self.logs.log_text.getvalue())
# Timeout, just in case
real_reactor.callLater(10, timeout)
real_reactor.run()
client_protocol = p[0]
# We get a CONNECT
self.assertEqual(client_protocol.data,
ConnACK(session_present=False, return_code=1).serialise())
client_protocol.data = b""
pump.flush()
# No events!
self.assertEqual(len(session.events), 0)
