Exemplo n.º 1
0
	def test_count(self):
		q = SQSQueueMock('test')
		m = Message()
		m.set_body('this is a test')
		status = q.write(m)
		self.assertTrue(status)
		self.assertTrue(q.count() == 1)
		
		status = q.write(m)
		self.assertTrue(status)
		self.assertTrue(q.count() == 2)
		
		status = q.write(m)
		self.assertTrue(status)
		self.assertTrue(q.count() == 3)
		
		messages = q.get_messages(num_messages=1)
		self.assertTrue(q.delete_message(messages[0]))
		self.assertTrue(q.count() == 2)
		
		messages = q.get_messages(num_messages=1)
		self.assertTrue(q.delete_message(messages[0]))
		self.assertTrue(q.count() == 1)
		
		messages = q.get_messages(num_messages=1)
		self.assertTrue(q.delete_message(messages[0]))
		self.assertTrue(q.count() == 0)
Exemplo n.º 2
0
    def test_count(self):
        q = SQSQueueMock('test')
        m = Message()
        m.set_body('this is a test')
        status = q.write(m)
        self.assertTrue(status)
        self.assertTrue(q.count() == 1)

        status = q.write(m)
        self.assertTrue(status)
        self.assertTrue(q.count() == 2)

        status = q.write(m)
        self.assertTrue(status)
        self.assertTrue(q.count() == 3)

        messages = q.get_messages(num_messages=1)
        self.assertTrue(q.delete_message(messages[0]))
        self.assertTrue(q.count() == 2)

        messages = q.get_messages(num_messages=1)
        self.assertTrue(q.delete_message(messages[0]))
        self.assertTrue(q.count() == 1)

        messages = q.get_messages(num_messages=1)
        self.assertTrue(q.delete_message(messages[0]))
        self.assertTrue(q.count() == 0)
Exemplo n.º 3
0
	def test_delete_message(self):
		m = Message()
		m.set_body('this is a test')
		q = SQSQueueMock('test')
		status = q.write(m)
		self.assertTrue(status)
		
		messages = q.get_messages(num_messages=2)
		self.assertTrue(len(messages) == 1)
		self.assertTrue(messages[0].get_body() == 'this is a test')
		
		self.assertTrue(q.delete_message(messages[0]))
		
		messages = q.get_messages(num_messages=1)
		self.assertTrue(len(messages) == 0)
Exemplo n.º 4
0
    def test_delete_message(self):
        m = Message()
        m.set_body('this is a test')
        q = SQSQueueMock('test')
        status = q.write(m)
        self.assertTrue(status)

        messages = q.get_messages(num_messages=2)
        self.assertTrue(len(messages) == 1)
        self.assertTrue(messages[0].get_body() == 'this is a test')

        self.assertTrue(q.delete_message(messages[0]))

        messages = q.get_messages(num_messages=1)
        self.assertTrue(len(messages) == 0)