catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Run black on moto & test directories.

Browse source
This commit is contained in:
Asher Foa 2019-10-31 08:44:26 -07:00
commit 96e5b1993d
507 changed files with 52541 additions and 47814 deletions
Download patch file Download diff file Expand all files Collapse all files

92
tests/test_iotdata/test_iotdata.py
View file

@ -11,9 +11,9 @@ from moto import mock_iotdata, mock_iot
@mock_iot
@mock_iotdata
def test_basic():
iot_client = boto3.client('iot', region_name='ap-northeast-1')
client = boto3.client('iot-data', region_name='ap-northeast-1')
name = 'my-thing'
iot_client = boto3.client("iot", region_name="ap-northeast-1")
client = boto3.client("iot-data", region_name="ap-northeast-1")
name = "my-thing"
raw_payload = b'{"state": {"desired": {"led": "on"}}}'
iot_client.create_thing(thingName=name)
@ -22,20 +22,24 @@ def test_basic():
res = client.update_thing_shadow(thingName=name, payload=raw_payload)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = '{"desired": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('desired').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(1)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"desired"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(1)
payload.should.have.key("timestamp")
res = client.get_thing_shadow(thingName=name)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = b'{"desired": {"led": "on"}, "delta": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('desired').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(1)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"desired"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(1)
payload.should.have.key("timestamp")
client.delete_thing_shadow(thingName=name)
with assert_raises(ClientError):
@ -45,55 +49,63 @@ def test_basic():
@mock_iot
@mock_iotdata
def test_update():
iot_client = boto3.client('iot', region_name='ap-northeast-1')
client = boto3.client('iot-data', region_name='ap-northeast-1')
name = 'my-thing'
iot_client = boto3.client("iot", region_name="ap-northeast-1")
client = boto3.client("iot-data", region_name="ap-northeast-1")
name = "my-thing"
raw_payload = b'{"state": {"desired": {"led": "on"}}}'
iot_client.create_thing(thingName=name)
# first update
res = client.update_thing_shadow(thingName=name, payload=raw_payload)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = '{"desired": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('desired').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(1)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"desired"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(1)
payload.should.have.key("timestamp")
res = client.get_thing_shadow(thingName=name)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = b'{"desired": {"led": "on"}, "delta": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('desired').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(1)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"desired"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(1)
payload.should.have.key("timestamp")
# reporting new state
new_payload = b'{"state": {"reported": {"led": "on"}}}'
res = client.update_thing_shadow(thingName=name, payload=new_payload)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = '{"reported": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('reported').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(2)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"reported"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(2)
payload.should.have.key("timestamp")
res = client.get_thing_shadow(thingName=name)
payload = json.loads(res['payload'].read())
payload = json.loads(res["payload"].read())
expected_state = b'{"desired": {"led": "on"}, "reported": {"led": "on"}}'
payload.should.have.key('state').which.should.equal(json.loads(expected_state))
payload.should.have.key('metadata').which.should.have.key('desired').which.should.have.key('led')
payload.should.have.key('version').which.should.equal(2)
payload.should.have.key('timestamp')
payload.should.have.key("state").which.should.equal(json.loads(expected_state))
payload.should.have.key("metadata").which.should.have.key(
"desired"
).which.should.have.key("led")
payload.should.have.key("version").which.should.equal(2)
payload.should.have.key("timestamp")
raw_payload = b'{"state": {"desired": {"led": "on"}}, "version": 1}'
with assert_raises(ClientError) as ex:
client.update_thing_shadow(thingName=name, payload=raw_payload)
ex.exception.response['ResponseMetadata']['HTTPStatusCode'].should.equal(409)
ex.exception.response['Error']['Message'].should.equal('Version conflict')
ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(409)
ex.exception.response["Error"]["Message"].should.equal("Version conflict")
@mock_iotdata
def test_publish():
client = boto3.client('iot-data', region_name='ap-northeast-1')
client.publish(topic='test/topic', qos=1, payload=b'')
client = boto3.client("iot-data", region_name="ap-northeast-1")
client.publish(topic="test/topic", qos=1, payload=b"")

9
tests/test_iotdata/test_server.py
View file

@ -5,9 +5,10 @@ import sure # noqa
import moto.server as server
from moto import mock_iotdata
'''
"""
Test the different server responses
'''
"""
@mock_iotdata
def test_iotdata_list():
@ -15,6 +16,6 @@ def test_iotdata_list():
test_client = backend.test_client()
# just making sure that server is up
thing_name = 'nothing'
res = test_client.get('/things/{}/shadow'.format(thing_name))
thing_name = "nothing"
res = test_client.get("/things/{}/shadow".format(thing_name))
res.status_code.should.equal(404)