Advertisement
Guest User

Untitled

a guest
Mar 29th, 2017
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.61 KB | None | 0 0
  1. def get_index_from_s3(key):
  2. try:
  3. response = s3.get_object(
  4. Bucket=bucket,
  5. Key=key
  6. )
  7. body = response.get('Body')
  8. content = body.read().decode('utf-8')
  9. except ClientError as ex:
  10. # print 'EXCEPTION MESSAGE: {}'.format(ex.response['Error']['Code'])
  11. content = '{}'
  12.  
  13. message = json.loads(content)
  14. return message
  15.  
  16. def arrange(self):
  17. super(WhenCognitoAndNerfFoundTestCase, self).arrange()
  18. # self.s3_response = self.s3.get_object.return_value
  19. self.s3_body = self.s3.get_object.return_value.get.return_value
  20. self.s3_body.read.return_value.decode.side_effect = [
  21. self.cognito_content,
  22. self.nerf_content]
  23. signed_url = "https://this.is/a/signed/url/index.html"
  24. self.s3.generate_presigned_url.return_value = signed_url
  25.  
  26. def arrange(self):
  27. super(WhenCognitoOnlyFoundTestCase, self).arrange()
  28. # self.s3_response = MagicMock()
  29. # botocore.response.StreamingBody
  30. self.s3.get_object.side_effect = [{},
  31. ClientError]
  32. # self.s3_response = self.s3.get_object.return_value
  33. self.s3_body = self.s3.get_object.return_value.get.return_value
  34. self.s3_body.read.return_value.decode.return_value =
  35. self.cognito_content
  36.  
  37. def get_index_from_s3(key):
  38. try:
  39. response = s3.get_object(
  40. Bucket=bucket,
  41. Key=key
  42. )
  43. body = response.get('Body')
  44. > content = body.read().decode('utf-8')
  45. E AttributeError: 'NoneType' object has no attribute 'read'
  46.  
  47. master_profile.py:66: AttributeError
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement