From 5da3a740af25f7cb4f7101013464bed8d5c51bc1 Mon Sep 17 00:00:00 2001
From: Andrew Melton <andrew.melton@rackspace.com>
Date: Tue, 4 Jun 2013 11:52:09 -0400
Subject: [PATCH] Accept POST on usage confirm dbapi call.

---
 stacktach/dbapi.py       |  2 +-
 tests/unit/test_dbapi.py | 45 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/stacktach/dbapi.py b/stacktach/dbapi.py
index 85734d5..1a548ea 100644
--- a/stacktach/dbapi.py
+++ b/stacktach/dbapi.py
@@ -148,7 +148,7 @@ def get_usage_exist(request, exist_id):
 
 @api_call
 def exists_send_status(request, message_id):
-    if request.method != 'PUT':
+    if request.method not in ['PUT', 'POST']:
         raise BadRequestException(message="Invalid method")
 
     if request.body is None or request.body == '':
diff --git a/tests/unit/test_dbapi.py b/tests/unit/test_dbapi.py
index 8af5d01..fdc9c44 100644
--- a/tests/unit/test_dbapi.py
+++ b/tests/unit/test_dbapi.py
@@ -404,6 +404,24 @@ class DBAPITestCase(unittest.TestCase):
         self.assertEqual(exists.send_status, 200)
         self.mox.VerifyAll()
 
+    def test_send_status_accepts_post(self):
+        fake_request = self.mox.CreateMockAnything()
+        fake_request.method = 'POST'
+        body_dict = {'send_status': 200}
+        body = json.dumps(body_dict)
+        fake_request.body = body
+        exists = self.mox.CreateMockAnything()
+        result = self.mox.CreateMockAnything()
+        models.InstanceExists.objects.select_for_update().AndReturn(result)
+        result.get(message_id=MESSAGE_ID_1).AndReturn(exists)
+        exists.save()
+        self.mox.ReplayAll()
+
+        dbapi.exists_send_status(fake_request, MESSAGE_ID_1)
+
+        self.assertEqual(exists.send_status, 200)
+        self.mox.VerifyAll()
+
     def test_send_status_not_found(self):
         fake_request = self.mox.CreateMockAnything()
         fake_request.method = 'PUT'
@@ -517,6 +535,33 @@ class DBAPITestCase(unittest.TestCase):
         trans_obj.__exit__(None, None, None)
         self.mox.ReplayAll()
 
+    def test_send_status_batch_accepts_post(self):
+        fake_request = self.mox.CreateMockAnything()
+        fake_request.method = 'POST'
+        messages = {
+            MESSAGE_ID_1: 200,
+            MESSAGE_ID_2: 400
+        }
+        body_dict = {'messages': messages}
+        body = json.dumps(body_dict)
+        fake_request.body = body
+        self.mox.StubOutWithMock(transaction, 'commit_on_success')
+        trans_obj = self.mox.CreateMockAnything()
+        transaction.commit_on_success().AndReturn(trans_obj)
+        trans_obj.__enter__()
+        results1 = self.mox.CreateMockAnything()
+        models.InstanceExists.objects.select_for_update().AndReturn(results1)
+        exists1 = self.mox.CreateMockAnything()
+        results1.get(message_id=MESSAGE_ID_2).AndReturn(exists1)
+        exists1.save()
+        results2 = self.mox.CreateMockAnything()
+        models.InstanceExists.objects.select_for_update().AndReturn(results2)
+        exists2 = self.mox.CreateMockAnything()
+        results2.get(message_id=MESSAGE_ID_1).AndReturn(exists2)
+        exists2.save()
+        trans_obj.__exit__(None, None, None)
+        self.mox.ReplayAll()
+
         resp = dbapi.exists_send_status(fake_request, 'batch')
         self.assertEqual(resp.status_code, 200)
         exists1.send_status = 200