Update tests

Author: jelly

.coveragerc

@@ -10,3 +10,4 @@ omit =
 	conftest.py
         local_settings.py
 	/home/travis/virtualenv*
+	manage.py

conftest.py

@@ -2,6 +2,7 @@
 
 from django.core.management import call_command
 
+from main.models import Repo
 from devel.models import UserProfile
 
 
@@ -31,8 +32,9 @@ def staff_groups(db):
     call_command('loaddata', 'devel/fixtures/staff_groups.json')
 
 
+# TODO: test with non-admin user fixture
 @pytest.fixture
-def admin_user_profile(admin_user):
+def admin_user_profile(admin_user, arches, repos):
     profile = UserProfile.objects.create(user=admin_user,
                                          public_email="[email protected]")
     yield profile

packages/tests.py

@@ -13,7 +13,6 @@ def request(client, pkgid, adopt=True):
     return client.post('/packages/update/', data, follow=True)
 
 
-
 def test_wrong_request(admin_client, arches, repos, package):
     # TODO(jelle): fix 
     pkg = Package.objects.first()

packages/tests/__init__.py

@@ -0,0 +1,34 @@
+def assert_flag_package(client):
+    data = {
+        'website': '',
+        'email': '[email protected]',
+        'message': 'new linux version',
+    }
+
+    # TODO: hardcoded package name
+    response = client.post('/packages/core/x86_64/linux/flag/',
+                                data,
+                                follow=True)
+    assert response.status_code == 200
+
+
+def test_unflag_package_404(admin_user_profile, admin_client):
+    response = admin_client.get('/packages/core/x86_64/fooobar/unflag/')
+    assert response.status_code == 404
+
+    response = admin_client.get('/packages/core/x86_64/fooobar/unflag/all/')
+    assert response.status_code == 404
+
+
+def test_unflag_package(admin_user_profile, admin_client, package):
+    assert_flag_package(admin_client)
+    response = admin_client.get('/packages/core/x86_64/linux/unflag/', follow=True)
+    assert response.status_code == 200
+    assert 'Flag linux as out-of-date' in response.content.decode()
+
+
+def test_unflag_all_package(admin_user_profile, admin_client, package):
+    assert_flag_package(admin_client)
+    response = admin_client.get('/packages/core/x86_64/linux/unflag/all/', follow=True)
+    assert response.status_code == 200
+    assert 'Flag linux as out-of-date' in response.content.decode()