|
"source": "import json\nimport unittest\nimport datetime\nimport pytz\n\nfrom dsmr_parser import telegram_specifications, obis_references\n\nfrom dsmr_parser.objects import CosemObject\nfrom dsmr_parser.objects import MBusObject\nfrom dsmr_parser.objects import ProfileGenericObject\nfrom dsmr_parser.parsers import TelegramParser\nfrom test.example_telegrams import TELEGRAM_V4_2, TELEGRAM_V5_TWO_MBUS, TELEGRAM_V5\nfrom decimal import Decimal\n\n\nclass TelegramTest(unittest.TestCase):\n \"\"\" Test instantiation of Telegram object \"\"\"\n\n def __init__(self, *args, **kwargs):\n self.item_names_tested = []\n super(TelegramTest, self).__init__(*args, **kwargs)\n\n def verify_telegram_item(self, telegram, testitem_name, object_type, unit_val, value_type, value_val):\n testitem = eval(\"telegram.{}\".format(testitem_name))\n assert isinstance(testitem, object_type)\n assert testitem.unit == unit_val\n assert isinstance(testitem.value, value_type)\n assert testitem.value == value_val\n self.item_names_tested.append(testitem_name)\n\n def test_instantiate(self):\n parser = TelegramParser(telegram_specifications.V4)\n telegram = parser.parse(TELEGRAM_V4_2)\n\n # P1_MESSAGE_HEADER (1-3:0.2.8)\n self.verify_telegram_item(telegram,\n 'P1_MESSAGE_HEADER',\n object_type=CosemObject,\n unit_val=None,\n value_type=str,\n value_val='42')\n\n # P1_MESSAGE_TIMESTAMP (0-0:1.0.0)\n self.verify_telegram_item(telegram,\n 'P1_MESSAGE_TIMESTAMP',\n CosemObject,\n unit_val=None,\n value_type=datetime.datetime,\n value_val=datetime.datetime(2016, 11, 13, 19, 57, 57, tzinfo=pytz.UTC))\n\n # ELECTRICITY_USED_TARIFF_1 (1-0:1.8.1)\n self.verify_telegram_item(telegram,\n 'ELECTRICITY_USED_TARIFF_1',\n object_type=CosemObject,\n unit_val='kWh',\n value_type=Decimal,\n value_val=Decimal('1581.123'))\n\n # ELECTRICITY_USED_TARIFF_2 (1-0:1.8.2)\n self.verify_telegram_item(telegram,\n 'ELECTRICITY_USED_TARIFF_2',\n object_type=CosemObject,\n unit_val='kWh',\n value_type=Decimal,\n value_val=Decimal('1435.706'))\n\n # ELECTRICITY_DELIVERED_TARIFF_1 (1-0:2.8.1)\n self.verify_telegram_item(telegram,\n 'ELECTRICITY_DELIVERED_TARIFF_1',\n object_type=CosemObject,\n unit_val='kWh',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # ELECTRICITY_DELIVERED_TARIFF_2 (1-0:2.8.2)\n self.verify_telegram_item(telegram,\n 'ELECTRICITY_DELIVERED_TARIFF_2',\n object_type=CosemObject,\n unit_val='kWh',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # ELECTRICITY_ACTIVE_TARIFF (0-0:96.14.0)\n self.verify_telegram_item(telegram,\n 'ELECTRICITY_ACTIVE_TARIFF',\n object_type=CosemObject,\n unit_val=None,\n value_type=str,\n value_val='0002')\n\n # EQUIPMENT_IDENTIFIER (0-0:96.1.1)\n self.verify_telegram_item(telegram,\n 'EQUIPMENT_IDENTIFIER',\n object_type=CosemObject,\n unit_val=None,\n value_type=str,\n value_val='3960221976967177082151037881335713')\n\n # CURRENT_ELECTRICITY_USAGE (1-0:1.7.0)\n self.verify_telegram_item(telegram,\n 'CURRENT_ELECTRICITY_USAGE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('2.027'))\n\n # CURRENT_ELECTRICITY_DELIVERY (1-0:2.7.0)\n self.verify_telegram_item(telegram,\n 'CURRENT_ELECTRICITY_DELIVERY',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # SHORT_POWER_FAILURE_COUNT (1-0:96.7.21)\n self.verify_telegram_item(telegram,\n 'SHORT_POWER_FAILURE_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=15)\n\n # LONG_POWER_FAILURE_COUNT (96.7.9)\n self.verify_telegram_item(telegram,\n 'LONG_POWER_FAILURE_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=7)\n\n # VOLTAGE_SAG_L1_COUNT (1-0:32.32.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SAG_L1_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # VOLTAGE_SAG_L2_COUNT (1-0:52.32.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SAG_L2_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # VOLTAGE_SAG_L3_COUNT (1-0:72.32.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SAG_L3_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # VOLTAGE_SWELL_L1_COUNT (1-0:32.36.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SWELL_L1_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # VOLTAGE_SWELL_L2_COUNT (1-0:52.36.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SWELL_L2_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # VOLTAGE_SWELL_L3_COUNT (1-0:72.36.0)\n self.verify_telegram_item(telegram,\n 'VOLTAGE_SWELL_L3_COUNT',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=0)\n\n # TEXT_MESSAGE_CODE (0-0:96.13.1)\n self.verify_telegram_item(telegram,\n 'TEXT_MESSAGE_CODE',\n object_type=CosemObject,\n unit_val=None,\n value_type=type(None),\n value_val=None)\n\n # TEXT_MESSAGE (0-0:96.13.0)\n self.verify_telegram_item(telegram,\n 'TEXT_MESSAGE',\n object_type=CosemObject,\n unit_val=None,\n value_type=type(None),\n value_val=None)\n\n # INSTANTANEOUS_CURRENT_L1 (1-0:31.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_CURRENT_L1',\n object_type=CosemObject,\n unit_val='A',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # INSTANTANEOUS_CURRENT_L2 (1-0:51.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_CURRENT_L2',\n object_type=CosemObject,\n unit_val='A',\n value_type=Decimal,\n value_val=Decimal('6'))\n\n # INSTANTANEOUS_CURRENT_L3 (1-0:71.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_CURRENT_L3',\n object_type=CosemObject,\n unit_val='A',\n value_type=Decimal,\n value_val=Decimal('2'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L1_POSITIVE (1-0:21.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L1_POSITIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0.170'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L2_POSITIVE (1-0:41.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L2_POSITIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('1.247'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L3_POSITIVE (1-0:61.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L3_POSITIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0.209'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L1_NEGATIVE (1-0:22.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L1_NEGATIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L2_NEGATIVE (1-0:42.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L2_NEGATIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # INSTANTANEOUS_ACTIVE_POWER_L3_NEGATIVE (1-0:62.7.0)\n self.verify_telegram_item(telegram,\n 'INSTANTANEOUS_ACTIVE_POWER_L3_NEGATIVE',\n object_type=CosemObject,\n unit_val='kW',\n value_type=Decimal,\n value_val=Decimal('0'))\n\n # DEVICE_TYPE (0-1:24.1.0)\n self.verify_telegram_item(telegram,\n 'DEVICE_TYPE',\n object_type=CosemObject,\n unit_val=None,\n value_type=int,\n value_val=3)\n\n # EQUIPMENT_IDENTIFIER_GAS (0-1:96.1.0)\n self.verify_telegram_item(telegram,\n 'EQUIPMENT_IDENTIFIER_GAS',\n object_type=CosemObject,\n unit_val=None,\n value_type=str,\n value_val='4819243993373755377509728609491464')\n\n # HOURLY_GAS_METER_READING (0-1:24.2.1)\n self.verify_telegram_item(telegram,\n 'HOURLY_GAS_METER_READING',\n object_type=MBusObject,\n unit_val='m3',\n value_type=Decimal,\n value_val=Decimal('981.443'))\n\n # POWER_EVENT_FAILURE_LOG (1-0:99.97.0)\n testitem_name = 'POWER_EVENT_FAILURE_LOG'\n object_type = ProfileGenericObject\n testitem = eval(\"telegram.{}\".format(testitem_name))\n assert isinstance(testitem, object_type)\n assert testitem.buffer_length == 3\n assert testitem.buffer_type == '0-0:96.7.19'\n buffer = testitem.buffer\n assert isinstance(testitem.buffer, list)\n assert len(buffer) == 3\n assert all([isinstance(item, MBusObject) for item in buffer])\n date0 = datetime.datetime(2000, 1, 4, 17, 3, 20, tzinfo=datetime.timezone.utc)\n date1 = datetime.datetime(1999, 12, 31, 23, 0, 1, tzinfo=datetime.timezone.utc)\n date2 = datetime.datetime(2000, 1, 1, 23, 0, 3, tzinfo=datetime.timezone.utc)\n assert buffer[0].datetime == date0\n assert buffer[1].datetime == date1\n assert buffer[2].datetime == date2\n assert buffer[0].value == 237126\n assert buffer[1].value == 2147583646\n assert buffer[2].value == 2317482647\n assert all([isinstance(item.value, int) for item in buffer])\n assert all([isinstance(item.unit, str) for item in buffer])\n assert all([(item.unit == 's') for item in buffer])\n self.item_names_tested.append(testitem_name)\n\n # check if all items in telegram V4 specification are covered\n V4_name_list = [object[\"value_name\"] for object in\n telegram_specifications.V4['objects']]\n V4_name_set = set(V4_name_list)\n item_names_tested_set = set(self.item_names_tested)\n\n assert item_names_tested_set == V4_name_set\n\n def test_iter(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5)\n\n for obis_name, dsmr_object in telegram:\n break\n\n # Verify that the iterator works for at least one value\n self.assertEqual(obis_name, \"P1_MESSAGE_HEADER\")\n self.assertEqual(dsmr_object.value, '50')\n\n def test_mbus_devices(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5_TWO_MBUS)\n mbus_devices = telegram.MBUS_DEVICES\n\n self.assertEqual(len(mbus_devices), 2)\n\n mbus_device_1 = mbus_devices[0]\n self.assertEqual(mbus_device_1.MBUS_DEVICE_TYPE.value, 3)\n self.assertEqual(mbus_device_1.MBUS_EQUIPMENT_IDENTIFIER.value, None)\n self.assertEqual(mbus_device_1.MBUS_METER_READING.value, Decimal('0'))\n\n mbus_device_2 = mbus_devices[1]\n self.assertEqual(mbus_device_2.MBUS_DEVICE_TYPE.value, 3)\n self.assertEqual(mbus_device_2.MBUS_EQUIPMENT_IDENTIFIER.value, '4730303339303031393336393930363139')\n self.assertEqual(mbus_device_2.MBUS_METER_READING.value, Decimal('246.138'))\n\n def test_get_mbus_device_by_channel(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5_TWO_MBUS)\n\n mbus_device_1 = telegram.get_mbus_device_by_channel(1)\n self.assertEqual(mbus_device_1.MBUS_DEVICE_TYPE.value, 3)\n self.assertEqual(mbus_device_1.MBUS_EQUIPMENT_IDENTIFIER.value, None)\n self.assertEqual(mbus_device_1.MBUS_METER_READING.value, Decimal('0'))\n\n mbus_device_2 = telegram.get_mbus_device_by_channel(2)\n self.assertEqual(mbus_device_2.MBUS_DEVICE_TYPE.value, 3)\n self.assertEqual(mbus_device_2.MBUS_EQUIPMENT_IDENTIFIER.value, '4730303339303031393336393930363139')\n self.assertEqual(mbus_device_2.MBUS_METER_READING.value, Decimal('246.138'))\n\n def test_without_mbus_devices(self):\n parser = TelegramParser(telegram_specifications.V5, apply_checksum_validation=False)\n telegram = parser.parse('')\n\n self.assertFalse(hasattr(telegram, 'MBUS_DEVICES'))\n self.assertIsNone(telegram.get_mbus_device_by_channel(1))\n\n def test_to_json(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5)\n json_data = json.loads(telegram.to_json())\n\n self.maxDiff = None\n\n self.assertEqual(\n json_data,\n {'CURRENT_ELECTRICITY_DELIVERY': {'unit': 'kW', 'value': 0.0},\n 'CURRENT_ELECTRICITY_USAGE': {'unit': 'kW', 'value': 0.244},\n 'ELECTRICITY_ACTIVE_TARIFF': {'unit': None, 'value': '0002'},\n 'ELECTRICITY_DELIVERED_TARIFF_1': {'unit': 'kWh', 'value': 2.444},\n 'ELECTRICITY_DELIVERED_TARIFF_2': {'unit': 'kWh', 'value': 0.0},\n 'ELECTRICITY_USED_TARIFF_1': {'unit': 'kWh', 'value': 4.426},\n 'ELECTRICITY_USED_TARIFF_2': {'unit': 'kWh', 'value': 2.399},\n 'EQUIPMENT_IDENTIFIER': {'unit': None,\n 'value': '4B384547303034303436333935353037'},\n 'INSTANTANEOUS_ACTIVE_POWER_L1_NEGATIVE': {'unit': 'kW', 'value': 0.0},\n 'INSTANTANEOUS_ACTIVE_POWER_L1_POSITIVE': {'unit': 'kW', 'value': 0.07},\n 'INSTANTANEOUS_ACTIVE_POWER_L2_NEGATIVE': {'unit': 'kW', 'value': 0.0},\n 'INSTANTANEOUS_ACTIVE_POWER_L2_POSITIVE': {'unit': 'kW', 'value': 0.032},\n 'INSTANTANEOUS_ACTIVE_POWER_L3_NEGATIVE': {'unit': 'kW', 'value': 0.0},\n 'INSTANTANEOUS_ACTIVE_POWER_L3_POSITIVE': {'unit': 'kW', 'value': 0.142},\n 'INSTANTANEOUS_CURRENT_L1': {'unit': 'A', 'value': 0.48},\n 'INSTANTANEOUS_CURRENT_L2': {'unit': 'A', 'value': 0.44},\n 'INSTANTANEOUS_CURRENT_L3': {'unit': 'A', 'value': 0.86},\n 'INSTANTANEOUS_VOLTAGE_L1': {'unit': 'V', 'value': 230.0},\n 'INSTANTANEOUS_VOLTAGE_L2': {'unit': 'V', 'value': 230.0},\n 'INSTANTANEOUS_VOLTAGE_L3': {'unit': 'V', 'value': 229.0},\n 'LONG_POWER_FAILURE_COUNT': {'unit': None, 'value': 0},\n 'MBUS_DEVICES': [{'CHANNEL_ID': 1,\n 'MBUS_DEVICE_TYPE': {'unit': None, 'value': 3},\n 'MBUS_EQUIPMENT_IDENTIFIER': {'unit': None,\n 'value': '3232323241424344313233343536373839'},\n 'MBUS_METER_READING': {'datetime': '2017-01-02T15:10:05+00:00',\n 'unit': 'm3',\n 'value': 0.107}},\n {'CHANNEL_ID': 2,\n 'MBUS_DEVICE_TYPE': {'unit': None, 'value': 3},\n 'MBUS_EQUIPMENT_IDENTIFIER': {'unit': None,\n 'value': None}}],\n 'P1_MESSAGE_HEADER': {'unit': None, 'value': '50'},\n 'P1_MESSAGE_TIMESTAMP': {'unit': None, 'value': '2017-01-02T18:20:02+00:00'},\n 'POWER_EVENT_FAILURE_LOG': {'buffer': [],\n 'buffer_length': 0,\n 'buffer_type': '0-0:96.7.19'},\n 'SHORT_POWER_FAILURE_COUNT': {'unit': None, 'value': 13},\n 'TEXT_MESSAGE': {'unit': None, 'value': None},\n 'VOLTAGE_SAG_L1_COUNT': {'unit': None, 'value': 0},\n 'VOLTAGE_SAG_L2_COUNT': {'unit': None, 'value': 0},\n 'VOLTAGE_SAG_L3_COUNT': {'unit': None, 'value': 0},\n 'VOLTAGE_SWELL_L1_COUNT': {'unit': None, 'value': 0},\n 'VOLTAGE_SWELL_L2_COUNT': {'unit': None, 'value': 0},\n 'VOLTAGE_SWELL_L3_COUNT': {'unit': None, 'value': 0}}\n )\n\n def test_to_str(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5)\n\n self.maxDiff = None\n\n self.assertEqual(\n str(telegram),\n (\n 'P1_MESSAGE_HEADER: \t 50\t[None]\\n'\n 'P1_MESSAGE_TIMESTAMP: \t 2017-01-02T18:20:02+00:00\t[None]\\n'\n 'EQUIPMENT_IDENTIFIER: \t 4B384547303034303436333935353037\t[None]\\n'\n 'ELECTRICITY_USED_TARIFF_1: \t 4.426\t[kWh]\\n'\n 'ELECTRICITY_USED_TARIFF_2: \t 2.399\t[kWh]\\n'\n 'ELECTRICITY_DELIVERED_TARIFF_1: \t 2.444\t[kWh]\\n'\n 'ELECTRICITY_DELIVERED_TARIFF_2: \t 0.000\t[kWh]\\n'\n 'ELECTRICITY_ACTIVE_TARIFF: \t 0002\t[None]\\n'\n 'CURRENT_ELECTRICITY_USAGE: \t 0.244\t[kW]\\n'\n 'CURRENT_ELECTRICITY_DELIVERY: \t 0.000\t[kW]\\n'\n 'LONG_POWER_FAILURE_COUNT: \t 0\t[None]\\n'\n 'SHORT_POWER_FAILURE_COUNT: \t 13\t[None]\\n'\n 'POWER_EVENT_FAILURE_LOG: \t \t buffer length: 0\\n'\n '\t buffer type: 0-0:96.7.19\\n'\n 'VOLTAGE_SAG_L1_COUNT: \t 0\t[None]\\n'\n 'VOLTAGE_SAG_L2_COUNT: \t 0\t[None]\\n'\n 'VOLTAGE_SAG_L3_COUNT: \t 0\t[None]\\n'\n 'VOLTAGE_SWELL_L1_COUNT: \t 0\t[None]\\n'\n 'VOLTAGE_SWELL_L2_COUNT: \t 0\t[None]\\n'\n 'VOLTAGE_SWELL_L3_COUNT: \t 0\t[None]\\n'\n 'INSTANTANEOUS_VOLTAGE_L1: \t 230.0\t[V]\\n'\n 'INSTANTANEOUS_VOLTAGE_L2: \t 230.0\t[V]\\n'\n 'INSTANTANEOUS_VOLTAGE_L3: \t 229.0\t[V]\\n'\n 'INSTANTANEOUS_CURRENT_L1: \t 0.48\t[A]\\n'\n 'INSTANTANEOUS_CURRENT_L2: \t 0.44\t[A]\\n'\n 'INSTANTANEOUS_CURRENT_L3: \t 0.86\t[A]\\n'\n 'TEXT_MESSAGE: \t None\t[None]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L1_POSITIVE: \t 0.070\t[kW]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L2_POSITIVE: \t 0.032\t[kW]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L3_POSITIVE: \t 0.142\t[kW]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L1_NEGATIVE: \t 0.000\t[kW]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L2_NEGATIVE: \t 0.000\t[kW]\\n'\n 'INSTANTANEOUS_ACTIVE_POWER_L3_NEGATIVE: \t 0.000\t[kW]\\n'\n 'MBUS DEVICE (channel 1)\\n'\n '\tMBUS_DEVICE_TYPE: \t 3\t[None]\\n'\n '\tMBUS_EQUIPMENT_IDENTIFIER: \t 3232323241424344313233343536373839\t[None]\\n'\n '\tMBUS_METER_READING: \t 0.107\t[m3] at 2017-01-02T15:10:05+00:00\\n'\n 'MBUS DEVICE (channel 2)\\n'\n '\tMBUS_DEVICE_TYPE: \t 3\t[None]\\n'\n '\tMBUS_EQUIPMENT_IDENTIFIER: \t None\t[None]\\n'\n )\n )\n\n def test_getitem(self):\n parser = TelegramParser(telegram_specifications.V5)\n telegram = parser.parse(TELEGRAM_V5)\n\n self.assertEqual(telegram[obis_references.P1_MESSAGE_HEADER].value, '50')\n",
|