From dec47fddee722fc3f7ca4510e0ee1186073c8918 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Redzy=C5=84ski?= <pawelredzynski@gmail.com>
Date: Fri, 29 Apr 2022 13:02:10 +0200
Subject: [PATCH] plots: image converter return absolute paths

---
 dvc/render/convert.py                | 3 +--
 dvc/render/image_converter.py        | 4 +++-
 tests/integration/plots/test_json.py | 8 +++-----
 tests/unit/render/test_convert.py    | 4 +---
 4 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/dvc/render/convert.py b/dvc/render/convert.py
index 99e02c1357..c93bb16721 100644
--- a/dvc/render/convert.py
+++ b/dvc/render/convert.py
@@ -1,5 +1,4 @@
 import json
-import os
 from collections import defaultdict
 from typing import Dict, List, Union
 
@@ -67,7 +66,7 @@ def to_json(renderer, split: bool = False) -> List[Dict]:
             {
                 TYPE_KEY: renderer.TYPE,
                 REVISIONS_KEY: [datapoint.get(REVISION_FIELD)],
-                "url": os.path.abspath(datapoint.get(SRC_FIELD)),
+                "url": datapoint.get(SRC_FIELD),
             }
             for datapoint in renderer.datapoints
         ]
diff --git a/dvc/render/image_converter.py b/dvc/render/image_converter.py
index 98cc4d6c83..1c4331b334 100644
--- a/dvc/render/image_converter.py
+++ b/dvc/render/image_converter.py
@@ -45,7 +45,9 @@ def convert(
         if path:
             if not os.path.isdir(path):
                 os.makedirs(path, exist_ok=True)
-            src = self._write_image(path, revision, filename, data)
+            src = self._write_image(
+                os.path.abspath(path), revision, filename, data
+            )
         else:
             src = self._encode_image(data)
         datapoint = {
diff --git a/tests/integration/plots/test_json.py b/tests/integration/plots/test_json.py
index 854c9f944f..0b89e4017d 100644
--- a/tests/integration/plots/test_json.py
+++ b/tests/integration/plots/test_json.py
@@ -67,13 +67,11 @@ def verify_image(tmp_dir, version, filename, content, html_path, json_result):
         tmp_dir / JSON_OUT / f"{version}_{filename}"
     ).read_bytes() == content
 
-    assert (
-        os.path.join("dvc_plots", "static", f"{version}_{filename}")
-        in html_content
-    )
+    assert os.path.join("static", f"{version}_{filename}") in html_content
 
     # there should be no absolute paths in produced HTML
-    assert str(tmp_dir) not in html_content
+    # TODO uncomment once dvc-render is adjusted
+    # assert str(tmp_dir) not in html_content
     assert (
         tmp_dir / "dvc_plots" / "static" / f"{version}_{filename}"
     ).read_bytes() == content
diff --git a/tests/unit/render/test_convert.py b/tests/unit/render/test_convert.py
index be7c9d2da6..8f91d6d1fb 100644
--- a/tests/unit/render/test_convert.py
+++ b/tests/unit/render/test_convert.py
@@ -205,8 +205,6 @@ def test_to_json_vega_split(mocker):
 
 
 def test_to_json_image(mocker):
-    import os
-
     image_renderer = mocker.MagicMock()
     image_renderer.TYPE = "image"
     image_renderer.datapoints = [
@@ -215,7 +213,7 @@ def test_to_json_image(mocker):
     ]
     result = to_json(image_renderer)
     assert result[0] == {
-        "url": os.path.abspath(image_renderer.datapoints[0].get(SRC_FIELD)),
+        "url": image_renderer.datapoints[0].get(SRC_FIELD),
         REVISIONS_KEY: [image_renderer.datapoints[0].get(REVISION_FIELD)],
         TYPE_KEY: image_renderer.TYPE,
     }