Implement package states for easier reviews

[cheatdb.git] / app / tests / test_api.py

diff --git a/app/tests/test_api.py b/app/tests/test_api.py
index 8009656051c082c92a347f5473a4fed845b04734..154e17815d92f0f22bf83f87377f7ada3e7e1ef7 100644 (file)
--- a/app/tests/test_api.py
+++ b/app/tests/test_api.py
@@ -46,7 +46,7 @@ def test_packages_with_contents(client):
        packages = parse_json(rv.data)
 
        assert len(packages) > 0
-       assert len(packages) == Package.query.filter_by(approved=True).count()
+       assert len(packages) == Package.query.filter_by(state=PackageState.APPROVED).count()
 
        validate_package_list(packages)