tox linting

Author: flacoman91

complaint_search/tests/test_es_interface.py

@@ -381,7 +381,7 @@ def test_search_with_company_agg_exclude__valid(self):
     def test_search_with_product__valid(self):
         self.request_test(
             "search_with_product__valid",
-            agg_exclude=["congressional_district", "company","msa", "zip_code"],
+            agg_exclude=["congressional_district", "company", "msa", "zip_code"],
             product=["Payday loan", "Mortgage\u2022FHA mortgage"],
         )
 
@@ -398,7 +398,7 @@ def test_search_with_not_product__valid(self):
     def test_search_with_issue__valid(self):
         self.request_test(
             "search_with_issue__valid",
-            agg_exclude=["congressional_district", "company","msa", "zip_code"],
+            agg_exclude=["congressional_district", "company", "msa", "zip_code"],
             issue=[
                 "Communication tactics\u2022Frequent or repeated calls",
                 "Loan servicing, payments, escrow account",

complaint_search/tests/test_views_suggest_congressional_district.py

@@ -42,7 +42,7 @@ def test_suggest_text__valid(self, mock_essuggest):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         mock_essuggest.assert_called_once_with(
             "congressional_district",
-            None,
+            "congressional_district",
             field="complaint_what_happened",
             format="default",
             frm=0,

complaint_search/tests/test_views_suggest_msa.py

@@ -42,7 +42,7 @@ def test_suggest_text__valid(self, mock_essuggest):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         mock_essuggest.assert_called_once_with(
             "msa",
-            None,
+            "msa",
             field="complaint_what_happened",
             format="default",
             frm=0,

complaint_search/views.py

@@ -238,7 +238,7 @@ def removekey(d, key):
     if data.get("text"):
         data["text"] = data["text"].upper()
 
-    return _suggest_field(data, "congressional_district")
+    return _suggest_field(data, "congressional_district", "congressional_district")
 
 
 @api_view(["GET"])
@@ -264,6 +264,7 @@ def removekey(d, key):
 
     return _suggest_field(data, "company.suggest", "company.raw")
 
+
 @api_view(["GET"])
 @catch_es_error
 def suggest_msa(request):
@@ -285,7 +286,8 @@ def removekey(d, key):
     if data.get("text"):
         data["text"] = data["text"].upper()
 
-    return _suggest_field(data, "msa")
+    return _suggest_field(data, "msa", "msa")
+
 
 @api_view(["GET"])
 @throttle_classes(