| 
									
										
										
										
											2017-02-23 19:43:48 -05:00
										 |  |  | from __future__ import unicode_literals | 
					
						
							|  |  |  | import sure  # noqa | 
					
						
							| 
									
										
										
										
											2020-10-06 07:54:49 +02:00
										 |  |  | import pytest | 
					
						
							| 
									
										
										
										
											2017-02-23 19:43:48 -05:00
										 |  |  | import requests | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import boto3 | 
					
						
							|  |  |  | from moto import mock_sqs, settings | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  | base_url = ( | 
					
						
							|  |  |  |     "http://localhost:5000" | 
					
						
							|  |  |  |     if settings.TEST_SERVER_MODE | 
					
						
							|  |  |  |     else "http://motoapi.amazonaws.com" | 
					
						
							|  |  |  | ) | 
					
						
							| 
									
										
										
										
											2017-02-23 19:43:48 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | @mock_sqs | 
					
						
							|  |  |  | def test_reset_api(): | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     conn = boto3.client("sqs", region_name="us-west-1") | 
					
						
							| 
									
										
										
										
											2017-02-23 19:43:48 -05:00
										 |  |  |     conn.create_queue(QueueName="queue1") | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     conn.list_queues()["QueueUrls"].should.have.length_of(1) | 
					
						
							| 
									
										
										
										
											2017-02-23 19:43:48 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  |     res = requests.post("{base_url}/moto-api/reset".format(base_url=base_url)) | 
					
						
							|  |  |  |     res.content.should.equal(b'{"status": "ok"}') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     conn.list_queues().shouldnt.contain("QueueUrls")  # No more queues | 
					
						
							| 
									
										
										
										
											2017-03-11 22:45:42 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | @mock_sqs | 
					
						
							|  |  |  | def test_data_api(): | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     conn = boto3.client("sqs", region_name="us-west-1") | 
					
						
							| 
									
										
										
										
											2017-03-11 22:45:42 -05:00
										 |  |  |     conn.create_queue(QueueName="queue1") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     res = requests.post("{base_url}/moto-api/data.json".format(base_url=base_url)) | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     queues = res.json()["sqs"]["Queue"] | 
					
						
							| 
									
										
										
										
											2017-03-11 22:45:42 -05:00
										 |  |  |     len(queues).should.equal(1) | 
					
						
							|  |  |  |     queue = queues[0] | 
					
						
							| 
									
										
										
										
											2019-10-31 08:44:26 -07:00
										 |  |  |     queue["name"].should.equal("queue1") |