Delete customer by ID not name

alerta · Jan 2, 2017 · 296d1af · 296d1af
1 parent df8ce5b
commit 296d1af
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/alerta/app/database/mongo.py b/alerta/app/database/mongo.py
@@ -1363,7 +1363,7 @@ def get_customers(self, query=None):
 
     def delete_customer(self, customer):
 
-        response = self.db.customers.delete_one({"customer": customer})
+        response = self.db.customers.delete_one({"_id": customer})
         return True if response.deleted_count == 1 else False
 
     def get_keys(self, query=None):

diff --git a/tests/test_auth.py b/tests/test_auth.py
@@ -147,6 +147,9 @@ def test_login(self):
         }
         response = self.app.post('/customer', data=json.dumps(payload), headers=self.headers)
         self.assertEqual(response.status_code, 201)
+        data = json.loads(response.data.decode('utf-8'))
+
+        customer_id = data['id']
 
         payload = {
             'email': 'josephine@debeauharnais.fr',
@@ -203,5 +206,5 @@ def test_login(self):
         self.assertEqual(response.status_code, 200)
 
         # delete customer mapping
-        response = self.app.delete('/customer/' + 'Bonaparte Industries', headers=self.headers)
+        response = self.app.delete('/customer/' + customer_id, headers=self.headers)
         self.assertEqual(response.status_code, 200)