def test_parseSiteId(self):
class MQTTMessage:
def __init__(self, payload):
self.payload = payload
self.assertEqual(
CommonsManager.parseSiteId(MQTTMessage('{"siteId": "site_id", "IPAddress": "127.0.0.1"}')),
'site id')
self.assertEqual(
CommonsManager.parseSiteId(MQTTMessage('{"IPAddress": "127.0.0.1"}')),
'127.0.0.1')
def test_parseSiteId(self, mock_superManager):
class MQTTMessage:
def __init__(self, payload):
self.payload = payload
self.topic = 'test'
# mock SuperManager
mock_instance = MagicMock()
mock_superManager.getInstance.return_value = mock_instance
mock_instance.configManager.getAliceConfigByName.return_value = 'uuid'
self.assertEqual(
CommonsManager.parseSiteId(
MQTTMessage(
'{"siteId": "site_id", "IPAddress": "127.0.0.1"}')),
'site id')
self.assertEqual(
CommonsManager.parseSiteId(
MQTTMessage('{"IPAddress": "127.0.0.1"}')), '127.0.0.1')
self.assertEqual(CommonsManager.parseSiteId(MQTTMessage('{}')), 'uuid')
