fix tests

newsroom/tests/fixtures.py

@@ -294,7 +294,7 @@ def setup_user_company(app):
                 "email": "[email protected]",
                 "first_name": "Restrict",
                 "last_name": "User",
-                "company": COMPANY_1_ID,
+                "company": COMPANY_3_ID,
                 "is_enabled": True,
                 "is_approved": True,
                 "_created": utcnow(),

tests/core/test_reports.py

@@ -119,7 +119,7 @@ def test_company_products(client):
     resp = client.get("reports/company-products")
     report = json.loads(resp.get_data())
     assert report["name"] == "Products per company"
-    assert len(report["results"]) == 4
+    assert len(report["results"]) == 5
     assert report["results"][0]["name"] == "Example 2 Company"
     assert len(report["results"][0]["products"]) == 1
     assert report["results"][1]["name"] == "Example Company"
@@ -165,10 +165,11 @@ def test_companies(client):
     resp = client.get("reports/company")
     report = json.loads(resp.get_data())
     assert report["name"] == "Company"
-    assert len(report["results"]) == 4
+    assert len(report["results"]) == 5
     assert report["results"][0]["name"] == "Example 2 Company"
     assert report["results"][1]["name"] == "Example Company"
-    assert report["results"][2]["name"] == "Paper Co."
+    assert report["results"][2]["name"] == "News Co."
+    assert report["results"][3]["name"] == "Paper Co."
 
 
 def test_content_activity_csv(client):

tests/core/test_users.py

@@ -659,7 +659,7 @@ def test_create_user_inherit_sections(app):
 
 def test_filter_and_sorting_user(app, client):
     users = client.get("/users/search?q=").get_json()
-    assert len(users) == 3
+    assert len(users) == 4
     response = client.get("/users/search?q=admin")
     assert "admin" in response.get_data(as_text=True)
     user_data = response.get_json()[0]
@@ -676,7 +676,7 @@ def test_filter_and_sorting_user(app, client):
     # sort by First_name
     users = client.get("/users/search?q=&sort=[('first_name', 1)]").get_json()
     assert users[0]["first_name"] == "Foo"
-    assert users[2]["first_name"] == "Zoe"
+    assert users[3]["first_name"] == "Zoe"
 
     # sort by Last_name
     users = client.get("/users/search?q=&sort=[('last_name', 1)]").get_json()