Skip to content

Commit

Permalink
Renamed messaging module
Browse files Browse the repository at this point in the history
  • Loading branch information
lewis-chambers committed May 30, 2024
1 parent bf2cf77 commit 6d19cc7
Show file tree
Hide file tree
Showing 12 changed files with 20 additions and 48 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -45,4 +45,4 @@ markers = [
]

[tool.coverage.run]
omit = ["*example.py", "*__init__.py"]
omit = ["*example.py", "*__init__.py", "queries.py"]
12 changes: 0 additions & 12 deletions src/iotdevicesimulator/__assets__/configs.cfg

This file was deleted.

2 changes: 1 addition & 1 deletion src/iotdevicesimulator/devices.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import logging
from iotdevicesimulator.queries import CosmosQuery
from iotdevicesimulator.db import Oracle
from iotdevicesimulator.mqtt.aws import IotCoreMQTTConnection
from iotdevicesimulator.messaging.aws import IotCoreMQTTConnection
import random

logger = logging.getLogger(__name__)
Expand Down
2 changes: 1 addition & 1 deletion src/iotdevicesimulator/example.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from iotdevicesimulator.queries import CosmosQuery
from iotdevicesimulator.swarm import CosmosSwarm
from iotdevicesimulator.mqtt.aws import IotCoreMQTTConnection
from iotdevicesimulator.messaging.aws import IotCoreMQTTConnection
import asyncio
import config
from pathlib import Path
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import time
import json
from awscrt.exceptions import AwsCrtError
from iotdevicesimulator.mqtt.core import MessagingBaseClass
from iotdevicesimulator.messaging.core import MessagingBaseClass
import backoff
import logging

Expand Down Expand Up @@ -157,18 +157,20 @@ def _on_connection_closed(connection, callback_data): # pragma: no cover
print("Connection closed")

@backoff.on_exception(backoff.expo, exception=AwsCrtError, logger=logger)
def _connect(self):
def _connect(self): # pragma: no cover
connect_future = self.connection.connect()
connect_future.result()
print("Connected!")

@backoff.on_exception(backoff.expo, exception=AwsCrtError, logger=logger)
def _disconnect(self):
def _disconnect(self): # pragma: no cover
print("Disconnecting...")
disconnect_future = self.connection.disconnect()
disconnect_future.result()

def send_message(self, message: str, topic: str, count: int = 1) -> None:
def send_message(
self, message: str, topic: str, count: int = 1
) -> None: # pragma: no cover
"""Sends a message to the endpoint.
Args:
Expand Down
File renamed without changes.
2 changes: 1 addition & 1 deletion src/iotdevicesimulator/swarm.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from iotdevicesimulator.devices import SensorSite
from iotdevicesimulator.db import Oracle
from iotdevicesimulator.queries import CosmosQuery
from iotdevicesimulator.mqtt.core import MessagingBaseClass
from iotdevicesimulator.messaging.core import MessagingBaseClass
import logging.config
from typing import List
from pathlib import Path
Expand Down
2 changes: 1 addition & 1 deletion src/tests/test_devices.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from iotdevicesimulator.devices import SensorSite
from iotdevicesimulator.db import Oracle
from iotdevicesimulator.queries import CosmosQuery
from iotdevicesimulator.mqtt.core import MockMessageConnection
from iotdevicesimulator.messaging.core import MockMessageConnection
from parameterized import parameterized
import pathlib, config

Expand Down
16 changes: 8 additions & 8 deletions src/tests/test_messaging.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
import pytest
import unittest
from unittest.mock import patch
from iotdevicesimulator.mqtt.core import MockMessageConnection, MessagingBaseClass
from iotdevicesimulator.mqtt.aws import IotCoreMQTTConnection
from iotdevicesimulator.messaging.core import MockMessageConnection, MessagingBaseClass
from iotdevicesimulator.messaging.aws import IotCoreMQTTConnection
from config import Config
from pathlib import Path
import awscrt.mqtt
Expand Down Expand Up @@ -62,7 +62,7 @@ def test_instantiation(self):
def test_non_string_arguments(self):

with self.assertRaises(TypeError):
instance = IotCoreMQTTConnection(
IotCoreMQTTConnection(
1,
self.config["cert_path"],
self.config["key_path"],
Expand All @@ -71,16 +71,16 @@ def test_non_string_arguments(self):
)

with self.assertRaises(TypeError):
instance = IotCoreMQTTConnection(
IotCoreMQTTConnection(
self.config["endpoint"],
self.config["cert_path"],
1,
self.config["key_path"],
self.config["ca_cert_path"],
"client_id",
)

with self.assertRaises(TypeError):
instance = IotCoreMQTTConnection(
IotCoreMQTTConnection(
self.config["endpoint"],
self.config["cert_path"],
1,
Expand All @@ -89,7 +89,7 @@ def test_non_string_arguments(self):
)

with self.assertRaises(TypeError):
instance = IotCoreMQTTConnection(
IotCoreMQTTConnection(
self.config["endpoint"],
self.config["cert_path"],
self.config["key_path"],
Expand All @@ -98,7 +98,7 @@ def test_non_string_arguments(self):
)

with self.assertRaises(TypeError):
instance = IotCoreMQTTConnection(
IotCoreMQTTConnection(
self.config["endpoint"],
self.config["cert_path"],
self.config["key_path"],
Expand Down
18 changes: 0 additions & 18 deletions src/tests/test_queries.py

This file was deleted.

2 changes: 1 addition & 1 deletion src/tests/test_swarm.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from iotdevicesimulator.db import Oracle
from iotdevicesimulator.devices import SensorSite
from iotdevicesimulator.queries import CosmosQuery
from iotdevicesimulator.mqtt.core import MockMessageConnection
from iotdevicesimulator.messaging.core import MockMessageConnection

from pathlib import Path
from config import Config
Expand Down

0 comments on commit 6d19cc7

Please sign in to comment.