pytest simple tests, disable screenshots in tests

Author: yvess

django/Makefile

@@ -29,6 +29,12 @@ migrate:
 	$(DOCKER_EXEC) python3 manage.py migrate
 	make collectstatic
 
+run-tests:
+	$(DOCKER_EXEC) DJANGO_SETTINGS_MODULE=main.settings_tests pytest --create-db
+
+run-tests-dev:
+	$(DOCKER_EXEC) DJANGO_SETTINGS_MODULE=main.settings_tests pytest --reuse-db -vvv --nomigrations
+
 delete-gemeinden-django:
 	$(DOCKER_EXEC) python3 manage.py shell --command="from gsmap.models import Municipality;Municipality.objects.all().delete()"
 

django/gsmap/models.py

@@ -10,6 +10,7 @@
 from django.contrib.postgres import fields as pg_fields
 from django.contrib.sites.models import Site
 from django.utils.html import format_html
+from django.conf import settings
 from sortedm2m.fields import SortedManyToManyField
 from sorl.thumbnail import ImageField
 from gsuser.models import User
@@ -212,8 +213,8 @@ def save_screenshot():
             finally:
                 # always reconnect signal
                 post_save.connect(save_screenshot_handler, sender=Snapshot)
-
-    save_screenshot()
+    if hasattr(settings, 'SAVE_SCREENSHOT_ENABLED') and settings.SAVE_SCREENSHOT_ENABLED is True:
+        save_screenshot()
 
 
 class Workspace(models.Model):

django/gsmap/tests.py

@@ -1,3 +1,137 @@
-from django.test import TestCase
+import json
+import pytest
+from graphene.test import Client
+from main.schema import schema
 
-# Create your tests here.
+@pytest.mark.django_db
+def test_municipalities_exists():
+    client = Client(schema)
+    result = client.execute('''
+query {
+  municipalities {
+    edges {
+      node {
+        id
+        name
+        bfsNumber
+      }
+    }
+  }
+}''',
+                            context={})
+    result = json.loads(json.dumps(result)) # remove OrderedDics, default in python 3.7+
+    assert result == {
+        "data": {
+            "municipalities": {
+            "edges": [
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTo2MjUy",
+                    "name": "Anniviers",
+                    "bfsNumber": 6252
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTo2MDMx",
+                    "name": "Bagnes",
+                    "bfsNumber": 6031
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTozNzky",
+                    "name": "Bregaglia",
+                    "bfsNumber": 3792
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTozODUx",
+                    "name": "Davos",
+                    "bfsNumber": 3851
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZToxNjMx",
+                    "name": "Glarus Süd",
+                    "bfsNumber": 1631
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTo3ODQ=",
+                    "name": "Innertkirchen",
+                    "bfsNumber": 784
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTozNzYy",
+                    "name": "Scuol",
+                    "bfsNumber": 3762
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTozNTQz",
+                    "name": "Surses",
+                    "bfsNumber": 3543
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTo2MzAw",
+                    "name": "Zermatt",
+                    "bfsNumber": 6300
+                }
+                },
+                {
+                "node": {
+                    "id": "TXVuaWNpcGFsaXR5Tm9kZTozNzQ2",
+                    "name": "Zernez",
+                    "bfsNumber": 3746
+                }
+                }
+            ]
+            }
+        }
+    }
+
+
+@pytest.mark.django_db
+def test_snapshots_exists():
+    client = Client(schema)
+    result = client.execute('''
+query {
+    snapshots(isShowcase: true) {
+        edges {
+            node {
+                id
+                pk
+                title
+                topic
+                screenshot
+            }
+        }
+    }
+}''',
+                            context={})
+    result = json.loads(
+        json.dumps(result))  # remove OrderedDics, default in python 3.7+
+    assert result == {
+        'data': {
+            'snapshots': {
+                'edges': [{
+                    'node': {
+                        'id': 'U25hcHNob3ROb2RlOlI0UlBHQw==',
+                        'pk': 'R4RPGC',
+                        'title': 'test snapshot',
+                        'topic': 'test topic',
+                        'screenshot': ''
+                    }
+                }]
+            }
+        }
+    }

django/main/settings.py

@@ -138,3 +138,4 @@
     }
 }
 THUMBNAIL_BACKEND = 'main.utils.PermalinkThumbnailBackend'
+SAVE_SCREENSHOT_ENABLED = True