Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixing errors on get_job_document

Browse Source
This commit is contained in:
Stephan Huber 2018-09-24 13:04:39 +02:00
parent c49a8387bd
commit 1c5c5036e3
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/iot/responses.py
View File

@ -153,7 +153,7 @@ class IoTResponse(BaseResponse):
job = self.iot_backend.get_job_document(job_id=self._get_param("jobId")) job = self.iot_backend.get_job_document(job_id=self._get_param("jobId"))
if job.document is not None: if job.document is not None:
json.dumps({'document': job.document}) return json.dumps({'document': job.document})
else: else:
# job.document_source is not None: # job.document_source is not None:
# TODO: needs to be implemented to get document_source's content from S3 # TODO: needs to be implemented to get document_source's content from S3

17
tests/test_iot/test_iot.py
View File

@ -1,6 +1,7 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import json import json
import sure # noqa
import boto3 import boto3
@ -555,7 +556,10 @@ def test_create_job():
client = boto3.client('iot', region_name='eu-west-1') client = boto3.client('iot', region_name='eu-west-1')
name = "my-thing" name = "my-thing"
job_id = "TestJob" job_id = "TestJob"
# thing # thing# job document
# job_document = {
# "field": "value"
# }
thing = client.create_thing(thingName=name) thing = client.create_thing(thingName=name)
thing.should.have.key('thingName').which.should.equal(name) thing.should.have.key('thingName').which.should.equal(name)
thing.should.have.key('thingArn') thing.should.have.key('thingArn')
@ -718,16 +722,21 @@ def test_get_job_document_with_document_source():
def test_get_job_document_with_document(): def test_get_job_document_with_document():
client = boto3.client('iot', region_name='eu-west-1') client = boto3.client('iot', region_name='eu-west-1')
name = "my-thing" name = "my-thing"
job_id = "TestJob1" job_id = "TestJob"
# thing # thing
thing = client.create_thing(thingName=name) thing = client.create_thing(thingName=name)
thing.should.have.key('thingName').which.should.equal(name) thing.should.have.key('thingName').which.should.equal(name)
thing.should.have.key('thingArn') thing.should.have.key('thingArn')
# job document
job_document = {
"field": "value"
}
job = client.create_job( job = client.create_job(
jobId=job_id, jobId=job_id,
targets=[thing["thingArn"]], targets=[thing["thingArn"]],
document=json.dumps({'foo': 'bar'}), document=json.dumps(job_document),
presignedUrlConfig={ presignedUrlConfig={
'roleArn': 'arn:aws:iam::1:role/service-role/iot_job_role', 'roleArn': 'arn:aws:iam::1:role/service-role/iot_job_role',
'expiresInSec': 123 'expiresInSec': 123
@ -742,4 +751,4 @@ def test_get_job_document_with_document():
job.should.have.key('jobArn') job.should.have.key('jobArn')
job_document = client.get_job_document(jobId=job_id) job_document = client.get_job_document(jobId=job_id)
job_document.should.have.key('document').which.should.equal('') job_document.should.have.key('document').which.should.equal("{\"field\": \"value\"}")