add tests for extrametadata query; pep8 format imports

Author: t-book

geonode/documents/api/tests.py

@@ -17,18 +17,19 @@
 #
 #########################################################################
 import os
-from django.contrib.auth import get_user_model
 import logging
 
+from django.contrib.auth import get_user_model
 from urllib.parse import urljoin
 
 from django.urls import reverse
 from rest_framework.test import APITestCase
 
 from guardian.shortcuts import assign_perm, get_anonymous_user
 from geonode import settings
-from geonode.documents.models import Document
+
 from geonode.base.populate_test_data import create_models
+from geonode.documents.models import Document
 
 logger = logging.getLogger(__name__)
 
@@ -76,6 +77,17 @@ def test_documents(self):
         # import json
         # logger.error(f"{json.dumps(layers_data)}")
 
+    def test_extra_metadata_included_with_param(self):
+        resource = Document.objects.first()
+        url = urljoin(f"{reverse('documents-list')}/", f"{resource.pk}")
+        data = {"include[]": "metadata"}
+
+        response = self.client.get(url, format="json", data=data)
+        self.assertIsNotNone(response.data["document"].get("metadata"))
+
+        response = self.client.get(url, format="json")
+        self.assertNotIn("metadata", response.data["document"])
+
     def test_creation_return_error_if_file_is_not_passed(self):
         """
         If file_path is not available, should raise error

geonode/geoapps/api/tests.py

@@ -19,15 +19,16 @@
 import json
 import logging
 from unittest.mock import MagicMock
+from urllib.parse import urljoin
 
-from django.urls import reverse
 from django.contrib.auth import get_user_model
+from django.urls import reverse
 from rest_framework.test import APITestCase
+
+from geonode.base.populate_test_data import create_models
 from geonode.geoapps.api.exceptions import DuplicateGeoAppException, InvalidGeoAppException
 from geonode.geoapps.api.serializers import GeoAppSerializer
-
 from geonode.geoapps.models import GeoApp
-from geonode.base.populate_test_data import create_models
 
 logger = logging.getLogger(__name__)
 
@@ -76,6 +77,17 @@ def test_geoapps_list(self):
             json.loads(response.data["geoapps"][0]["data"]), {"test_data": {"test": ["test_1", "test_2", "test_3"]}}
         )
 
+    def test_extra_metadata_included_with_param(self):
+        _app = GeoApp.objects.first()
+        url = urljoin(f"{reverse('geoapps-list')}/", f"{_app.pk}")
+        data = {"include[]": "metadata"}
+
+        response = self.client.get(url, format="json", data=data)
+        self.assertIsNotNone(response.data["geoapp"].get("metadata"))
+
+        response = self.client.get(url, format="json")
+        self.assertNotIn("metadata", response.data["geoapp"])
+
     def test_geoapps_crud(self):
         """
         Ensure we can create/update GeoApps.

geonode/layers/api/tests.py

@@ -21,17 +21,17 @@
 
 from unittest.mock import patch
 from django.contrib.auth import get_user_model
-
 from urllib.parse import urljoin
 
+from django.conf import settings
 from django.urls import reverse
 from rest_framework.test import APITestCase
-from geonode.geoserver.createlayer.utils import create_dataset
+
 from guardian.shortcuts import assign_perm, get_anonymous_user
+from geonode.geoserver.createlayer.utils import create_dataset
 
-from django.conf import settings
-from geonode.layers.models import Dataset, Attribute
 from geonode.base.populate_test_data import create_models, create_single_dataset
+from geonode.layers.models import Attribute, Dataset
 from geonode.maps.models import Map, MapLayer
 
 logger = logging.getLogger(__name__)
@@ -121,6 +121,17 @@ def test_datasets(self):
             _dataset.use_featureinfo_custom_template = False
             _dataset.save()
 
+    def test_extra_metadata_included_with_param(self):
+        _dataset = Dataset.objects.first()
+        url = urljoin(f"{reverse('datasets-list')}/", f"{_dataset.pk}")
+        data = {"include[]": "metadata"}
+
+        response = self.client.get(url, format="json", data=data)
+        self.assertIsNotNone(response.data["dataset"].get("metadata"))
+
+        response = self.client.get(url, format="json")
+        self.assertNotIn("metadata", response.data["dataset"])
+
     def test_get_dataset_related_maps_and_maplayers(self):
         dataset = Dataset.objects.first()
         assign_perm("base.view_resourcebase", get_anonymous_user(), dataset.get_self_resource())

geonode/maps/api/tests.py

@@ -21,9 +21,10 @@
 
 from django.conf import settings
 from django.urls import reverse
+
+from guardian.shortcuts import assign_perm, get_anonymous_user
 from mock import patch
 from rest_framework.test import APITestCase
-from guardian.shortcuts import assign_perm, get_anonymous_user
 
 from geonode.base.populate_test_data import create_models
 from geonode.layers.models import Dataset
@@ -112,6 +113,17 @@ def test_maps(self):
                 self.assertEqual(response.data["map"]["maplayers"][0]["extra_params"], {"foo": "bar"})
                 self.assertIsNotNone(response.data["map"]["maplayers"][0]["dataset"])
 
+    def test_extra_metadata_included_with_param(self):
+        resource = Map.objects.first()
+        url = urljoin(f"{reverse('maps-list')}/", f"{resource.pk}")
+        data = {"include[]": "metadata,data"}
+
+        response = self.client.get(url, format="json", data=data)
+        self.assertIsNotNone(response.data["map"].get("metadata"))
+
+        response = self.client.get(url, format="json")
+        self.assertNotIn("map", response.data["document"])
+
     def test_patch_map(self):
         """
         Patch to maps/<pk>/