From 9e3eef341e6db40ed706df5ff84c31b7ec1a7834 Mon Sep 17 00:00:00 2001
From: Yohan Boniface <yohanboniface@free.fr>
Date: Fri, 22 Mar 2024 10:13:47 +0100
Subject: [PATCH] chore: move exports tests from js to playwright

---
 umap/static/umap/test/Map.Export.js       | 106 ----------
 umap/static/umap/test/index.html          |   1 -
 umap/tests/integration/test_export_map.py | 235 +++++++++++++++++++++-
 3 files changed, 226 insertions(+), 116 deletions(-)
 delete mode 100644 umap/static/umap/test/Map.Export.js

diff --git a/umap/static/umap/test/Map.Export.js b/umap/static/umap/test/Map.Export.js
deleted file mode 100644
index 006cedef..00000000
--- a/umap/static/umap/test/Map.Export.js
+++ /dev/null
@@ -1,106 +0,0 @@
-describe('U.Map.Export', function () {
-  let map
-  before(async () => {
-    await fetchMock.mock(
-      /\/datalayer\/62\/\?.*/,
-      JSON.stringify(RESPONSES.datalayer62_GET)
-    )
-    this.options = {
-      umap_id: 99,
-    }
-    map = initMap({ umap_id: 99 })
-    const datalayer_options = defaultDatalayerData()
-    await map.initDataLayers([datalayer_options])
-  })
-  after(function () {
-    fetchMock.restore()
-    clickCancel()
-    resetMap()
-  })
-
-  describe('#formatters()', function () {
-    it('should export to geojson', function () {
-      const { content, filetype, filename } = map.share.format('geojson')
-      assert.equal(filetype, 'application/json')
-      assert.equal(filename, 'name_of_the_map.geojson')
-      assert.deepEqual(JSON.parse(content), {
-        type: 'FeatureCollection',
-        features: [
-          {
-            type: 'Feature',
-            properties: {
-              name: 'name poly',
-            },
-            geometry: {
-              type: 'Polygon',
-              coordinates: [
-                [
-                  [11.25, 53.585984],
-                  [10.151367, 52.975108],
-                  [12.689209, 52.167194],
-                  [14.084473, 53.199452],
-                  [12.634277, 53.618579],
-                  [11.25, 53.585984],
-                  [11.25, 53.585984],
-                ],
-              ],
-            },
-          },
-          {
-            type: 'Feature',
-            properties: {
-              _umap_options: {
-                color: 'OliveDrab',
-              },
-              name: 'test',
-            },
-            geometry: {
-              type: 'Point',
-              coordinates: [-0.274658, 52.57635],
-            },
-          },
-          {
-            type: 'Feature',
-            properties: {
-              _umap_options: {
-                fill: false,
-                opacity: 0.6,
-              },
-              name: 'test',
-            },
-            geometry: {
-              type: 'LineString',
-              coordinates: [
-                [-0.571289, 54.476422],
-                [0.439453, 54.610255],
-                [1.724854, 53.448807],
-                [4.163818, 53.988395],
-                [5.306396, 53.533778],
-                [6.591797, 53.709714],
-                [7.042236, 53.350551],
-              ],
-            },
-          },
-        ],
-      })
-    })
-
-    it('should export to gpx', function () {
-      const { content, filetype, filename } = map.share.format('gpx')
-      assert.equal(filetype, 'application/gpx+xml')
-      assert.equal(filename, 'name_of_the_map.gpx')
-      const expected =
-        '<gpx xmlns="http://www.topografix.com/GPX/1/1" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.topografix.com/GPX/1/1 http://www.topografix.com/GPX/1/1/gpx.xsd" version="1.1" creator="togpx"><metadata/><wpt lat="52.57635" lon="-0.274658"><name>test</name><desc>name=test</desc></wpt><trk><name>name poly</name><desc>name=name poly</desc><trkseg><trkpt lat="53.585984" lon="11.25"/><trkpt lat="52.975108" lon="10.151367"/><trkpt lat="52.167194" lon="12.689209"/><trkpt lat="53.199452" lon="14.084473"/><trkpt lat="53.618579" lon="12.634277"/><trkpt lat="53.585984" lon="11.25"/><trkpt lat="53.585984" lon="11.25"/></trkseg></trk><trk><name>test</name><desc>name=test</desc><trkseg><trkpt lat="54.476422" lon="-0.571289"/><trkpt lat="54.610255" lon="0.439453"/><trkpt lat="53.448807" lon="1.724854"/><trkpt lat="53.988395" lon="4.163818"/><trkpt lat="53.533778" lon="5.306396"/><trkpt lat="53.709714" lon="6.591797"/><trkpt lat="53.350551" lon="7.042236"/></trkseg></trk></gpx>'
-      assert.equal(content, expected)
-    })
-
-    it('should export to kml', function () {
-      const { content, filetype, filename } = map.share.format('kml')
-      assert.equal(filetype, 'application/vnd.google-earth.kml+xml')
-      assert.equal(filename, 'name_of_the_map.kml')
-      const expected =
-        '<?xml version="1.0" encoding="UTF-8"?><kml xmlns="http://www.opengis.net/kml/2.2"><Document><Placemark><name>name poly</name><ExtendedData><Data name="name"><value>name poly</value></Data></ExtendedData><Polygon><outerBoundaryIs><LinearRing><coordinates>11.25,53.585984 10.151367,52.975108 12.689209,52.167194 14.084473,53.199452 12.634277,53.618579 11.25,53.585984 11.25,53.585984</coordinates></LinearRing></outerBoundaryIs></Polygon></Placemark><Placemark><name>test</name><ExtendedData><Data name="_umap_options"><value>[object Object]</value></Data><Data name="name"><value>test</value></Data></ExtendedData><Point><coordinates>-0.274658,52.57635</coordinates></Point></Placemark><Placemark><name>test</name><ExtendedData><Data name="_umap_options"><value>[object Object]</value></Data><Data name="name"><value>test</value></Data></ExtendedData><LineString><coordinates>-0.571289,54.476422 0.439453,54.610255 1.724854,53.448807 4.163818,53.988395 5.306396,53.533778 6.591797,53.709714 7.042236,53.350551</coordinates></LineString></Placemark></Document></kml>'
-      assert.equal(content, expected)
-    })
-  })
-})
diff --git a/umap/static/umap/test/index.html b/umap/static/umap/test/index.html
index e01794a3..254728ce 100644
--- a/umap/static/umap/test/index.html
+++ b/umap/static/umap/test/index.html
@@ -88,7 +88,6 @@
     </script>
     <script src="./_pre.js" defer></script>
     <script src="./Map.js" defer></script>
-    <script src="./Map.Export.js" defer></script>
     <script src="./DataLayer.js" defer></script>
     <script src="./TableEditor.js" defer></script>
     <script src="./Feature.js" defer></script>
diff --git a/umap/tests/integration/test_export_map.py b/umap/tests/integration/test_export_map.py
index 69599dde..1f65d46f 100644
--- a/umap/tests/integration/test_export_map.py
+++ b/umap/tests/integration/test_export_map.py
@@ -4,10 +4,84 @@ from pathlib import Path
 import pytest
 from playwright.sync_api import expect
 
+from ..base import DataLayerFactory
+
 pytestmark = pytest.mark.django_db
 
+DATALAYER_DATA = {
+    "type": "FeatureCollection",
+    "features": [
+        {
+            "type": "Feature",
+            "properties": {
+                "name": "name poly",
+            },
+            "id": "gyNzM",
+            "geometry": {
+                "type": "Polygon",
+                "coordinates": [
+                    [
+                        [11.25, 53.585984],
+                        [10.151367, 52.975108],
+                        [12.689209, 52.167194],
+                        [14.084473, 53.199452],
+                        [12.634277, 53.618579],
+                        [11.25, 53.585984],
+                        [11.25, 53.585984],
+                    ],
+                ],
+            },
+        },
+        {
+            "type": "Feature",
+            "properties": {
+                "_umap_options": {
+                    "color": "OliveDrab",
+                },
+                "name": "test",
+                "description": "Some description",
+            },
+            "id": "QwNjg",
+            "geometry": {
+                "type": "Point",
+                "coordinates": [-0.274658, 52.57635],
+            },
+        },
+        {
+            "type": "Feature",
+            "properties": {
+                "_umap_options": {
+                    "fill": False,
+                    "opacity": 0.6,
+                },
+                "name": "test",
+            },
+            "id": "YwMTM",
+            "geometry": {
+                "type": "LineString",
+                "coordinates": [
+                    [-0.571289, 54.476422],
+                    [0.439453, 54.610255],
+                    [1.724854, 53.448807],
+                    [4.163818, 53.988395],
+                    [5.306396, 53.533778],
+                    [6.591797, 53.709714],
+                    [7.042236, 53.350551],
+                ],
+            },
+        },
+    ],
+}
 
-def test_umap_export(map, live_server, datalayer, page):
+
+@pytest.fixture
+def bootstrap(map, live_server):
+    map.settings["properties"]["onLoadPanel"] = "databrowser"
+    map.save()
+    DataLayerFactory(map=map, data=DATALAYER_DATA)
+
+
+def test_umap_export(map, live_server, bootstrap, page):
     page.goto(f"{live_server.url}{map.get_absolute_url()}?share")
     link = page.get_by_role("link", name="full backup")
     expect(link).to_be_visible()
@@ -34,16 +108,56 @@ def test_umap_export(map, live_server, datalayer, page):
                 "features": [
                     {
                         "geometry": {
-                            "coordinates": [14.68896484375, 48.55297816440071],
+                            "coordinates": [
+                                [
+                                    [11.25, 53.585984],
+                                    [10.151367, 52.975108],
+                                    [12.689209, 52.167194],
+                                    [14.084473, 53.199452],
+                                    [12.634277, 53.618579],
+                                    [11.25, 53.585984],
+                                    [11.25, 53.585984],
+                                ]
+                            ],
+                            "type": "Polygon",
+                        },
+                        "id": "gyNzM",
+                        "properties": {"name": "name poly"},
+                        "type": "Feature",
+                    },
+                    {
+                        "geometry": {
+                            "coordinates": [-0.274658, 52.57635],
                             "type": "Point",
                         },
+                        "id": "QwNjg",
                         "properties": {
-                            "_umap_options": {"color": "DarkCyan", "iconClass": "Ball"},
-                            "description": "Da place anonymous " "again 755",
-                            "name": "Here",
+                            "_umap_options": {"color": "OliveDrab"},
+                            "name": "test",
+                            "description": "Some description",
                         },
                         "type": "Feature",
-                    }
+                    },
+                    {
+                        "geometry": {
+                            "coordinates": [
+                                [-0.571289, 54.476422],
+                                [0.439453, 54.610255],
+                                [1.724854, 53.448807],
+                                [4.163818, 53.988395],
+                                [5.306396, 53.533778],
+                                [6.591797, 53.709714],
+                                [7.042236, 53.350551],
+                            ],
+                            "type": "LineString",
+                        },
+                        "id": "YwMTM",
+                        "properties": {
+                            "_umap_options": {"fill": False, "opacity": 0.6},
+                            "name": "test",
+                        },
+                        "type": "Feature",
+                    },
                 ],
                 "type": "FeatureCollection",
             }
@@ -66,12 +180,13 @@ def test_umap_export(map, live_server, datalayer, page):
             "tilelayersControl": True,
             "zoom": 7,
             "zoomControl": True,
+            "onLoadPanel": "databrowser",
         },
         "type": "umap",
     }
 
 
-def test_csv_export(map, live_server, datalayer, page):
+def test_csv_export(map, live_server, bootstrap, page):
     page.goto(f"{live_server.url}{map.get_absolute_url()}?share")
     button = page.get_by_role("button", name="csv")
     expect(button).to_be_visible()
@@ -83,6 +198,108 @@ def test_csv_export(map, live_server, datalayer, page):
     download.save_as(path)
     assert (
         path.read_text()
-        == """name,description,Latitude,Longitude
-Here,Da place anonymous again 755,48.55297816440071,14.68896484375"""
+        == """name,Latitude,Longitude,description
+name poly,53.0072070131872,12.182431646910137,
+test,52.57635,-0.274658,Some description
+test,53.725145179688646,2.9700064980570517,"""
     )
+
+
+def test_gpx_export(map, live_server, bootstrap, page):
+    page.goto(f"{live_server.url}{map.get_absolute_url()}?share")
+    button = page.get_by_role("button", name="gpx")
+    expect(button).to_be_visible()
+    with page.expect_download() as download_info:
+        button.click()
+    download = download_info.value
+    # FIXME assert mimetype (find no way to access it throught PW)
+    assert download.suggested_filename == "test_map.gpx"
+    path = Path("/tmp/") / download.suggested_filename
+    download.save_as(path)
+    assert (
+        path.read_text()
+        == """<gpx xmlns="http://www.topografix.com/GPX/1/1" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.topografix.com/GPX/1/1 http://www.topografix.com/GPX/1/1/gpx.xsd" version="1.1" creator="togpx"><metadata/><wpt lat="52.57635" lon="-0.274658"><name>test</name><desc>name=test
+description=Some description</desc></wpt><trk><name>name poly</name><desc>name=name poly</desc><trkseg><trkpt lat="53.585984" lon="11.25"/><trkpt lat="52.975108" lon="10.151367"/><trkpt lat="52.167194" lon="12.689209"/><trkpt lat="53.199452" lon="14.084473"/><trkpt lat="53.618579" lon="12.634277"/><trkpt lat="53.585984" lon="11.25"/><trkpt lat="53.585984" lon="11.25"/></trkseg></trk><trk><name>test</name><desc>name=test</desc><trkseg><trkpt lat="54.476422" lon="-0.571289"/><trkpt lat="54.610255" lon="0.439453"/><trkpt lat="53.448807" lon="1.724854"/><trkpt lat="53.988395" lon="4.163818"/><trkpt lat="53.533778" lon="5.306396"/><trkpt lat="53.709714" lon="6.591797"/><trkpt lat="53.350551" lon="7.042236"/></trkseg></trk></gpx>"""
+    )
+
+
+def test_kml_export(map, live_server, bootstrap, page):
+    page.goto(f"{live_server.url}{map.get_absolute_url()}?share")
+    button = page.get_by_role("button", name="kml")
+    expect(button).to_be_visible()
+    with page.expect_download() as download_info:
+        button.click()
+    download = download_info.value
+    assert download.suggested_filename == "test_map.kml"
+    path = Path("/tmp/") / download.suggested_filename
+    download.save_as(path)
+    assert (
+        path.read_text()
+        == """<?xml version="1.0" encoding="UTF-8"?><kml xmlns="http://www.opengis.net/kml/2.2"><Document><Placemark><name>name poly</name><ExtendedData><Data name="name"><value>name poly</value></Data></ExtendedData><Polygon><outerBoundaryIs><LinearRing><coordinates>11.25,53.585984 10.151367,52.975108 12.689209,52.167194 14.084473,53.199452 12.634277,53.618579 11.25,53.585984 11.25,53.585984</coordinates></LinearRing></outerBoundaryIs></Polygon></Placemark><Placemark><name>test</name><description>Some description</description><ExtendedData><Data name="_umap_options"><value>[object Object]</value></Data><Data name="name"><value>test</value></Data><Data name="description"><value>Some description</value></Data></ExtendedData><Point><coordinates>-0.274658,52.57635</coordinates></Point></Placemark><Placemark><name>test</name><ExtendedData><Data name="_umap_options"><value>[object Object]</value></Data><Data name="name"><value>test</value></Data></ExtendedData><LineString><coordinates>-0.571289,54.476422 0.439453,54.610255 1.724854,53.448807 4.163818,53.988395 5.306396,53.533778 6.591797,53.709714 7.042236,53.350551</coordinates></LineString></Placemark></Document></kml>"""
+    )
+
+
+def test_geojson_export(map, live_server, bootstrap, page):
+    page.goto(f"{live_server.url}{map.get_absolute_url()}?share")
+    button = page.get_by_role("button", name="geojson")
+    expect(button).to_be_visible()
+    with page.expect_download() as download_info:
+        button.click()
+    download = download_info.value
+    assert download.suggested_filename == "test_map.geojson"
+    path = Path("/tmp/") / download.suggested_filename
+    download.save_as(path)
+    assert json.loads(path.read_text()) == {
+        "features": [
+            {
+                "geometry": {
+                    "coordinates": [
+                        [
+                            [11.25, 53.585984],
+                            [10.151367, 52.975108],
+                            [12.689209, 52.167194],
+                            [14.084473, 53.199452],
+                            [12.634277, 53.618579],
+                            [11.25, 53.585984],
+                            [11.25, 53.585984],
+                        ]
+                    ],
+                    "type": "Polygon",
+                },
+                "id": "gyNzM",
+                "properties": {"name": "name poly"},
+                "type": "Feature",
+            },
+            {
+                "geometry": {"coordinates": [-0.274658, 52.57635], "type": "Point"},
+                "id": "QwNjg",
+                "properties": {
+                    "_umap_options": {"color": "OliveDrab"},
+                    "name": "test",
+                    "description": "Some description",
+                },
+                "type": "Feature",
+            },
+            {
+                "geometry": {
+                    "coordinates": [
+                        [-0.571289, 54.476422],
+                        [0.439453, 54.610255],
+                        [1.724854, 53.448807],
+                        [4.163818, 53.988395],
+                        [5.306396, 53.533778],
+                        [6.591797, 53.709714],
+                        [7.042236, 53.350551],
+                    ],
+                    "type": "LineString",
+                },
+                "id": "YwMTM",
+                "properties": {
+                    "_umap_options": {"fill": False, "opacity": 0.6},
+                    "name": "test",
+                },
+                "type": "Feature",
+            },
+        ],
+        "type": "FeatureCollection",
+    }