Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 12 additions & 12 deletions test/test_datasource.py
Original file line number Diff line number Diff line change
Expand Up @@ -608,15 +608,14 @@ def test_delete(server) -> None:
server.datasources.delete("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb")


def test_download(server) -> None:
def test_download(server, tmp_path) -> None:
with requests_mock.mock() as m:
m.get(
server.datasources.baseurl + "/9dbd2263-16b5-46e1-9c43-a76bb8ab65fb/content",
headers={"Content-Disposition": 'name="tableau_datasource"; filename="Sample datasource.tds"'},
)
file_path = server.datasources.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb")
file_path = server.datasources.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", filepath=tmp_path)
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_object(server) -> None:
Expand All @@ -630,21 +629,20 @@ def test_download_object(server) -> None:
assert isinstance(file_path, BytesIO)


def test_download_sanitizes_name(server) -> None:
def test_download_sanitizes_name(server, tmp_path) -> None:
filename = "Name,With,Commas.tds"
disposition = f'name="tableau_workbook"; filename="{filename}"'
with requests_mock.mock() as m:
m.get(
server.datasources.baseurl + "/1f951daf-4061-451a-9df1-69a8062664f2/content",
headers={"Content-Disposition": disposition},
)
file_path = server.datasources.download("1f951daf-4061-451a-9df1-69a8062664f2")
file_path = server.datasources.download("1f951daf-4061-451a-9df1-69a8062664f2", filepath=tmp_path)
assert os.path.basename(file_path) == "NameWithCommas.tds"
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_extract_only(server) -> None:
def test_download_extract_only(server, tmp_path) -> None:
# Pretend we're 2.5 for 'extract_only'
server.version = "2.5"

Expand All @@ -654,12 +652,13 @@ def test_download_extract_only(server) -> None:
headers={"Content-Disposition": 'name="tableau_datasource"; filename="Sample datasource.tds"'},
complete_qs=True,
)
file_path = server.datasources.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", include_extract=False)
file_path = server.datasources.download(
"9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", include_extract=False, filepath=tmp_path
)
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_no_extract_emits_deprecation_warning(server) -> None:
def test_download_no_extract_emits_deprecation_warning(server, tmp_path) -> None:
"""no_extract=True should emit a DeprecationWarning and map to includeExtract=False."""
server.version = "2.5"

Expand All @@ -670,9 +669,10 @@ def test_download_no_extract_emits_deprecation_warning(server) -> None:
complete_qs=True,
)
with pytest.warns(DeprecationWarning, match="deprecated and will be removed"):
file_path = server.datasources.download("9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", no_extract=True)
file_path = server.datasources.download(
"9dbd2263-16b5-46e1-9c43-a76bb8ab65fb", no_extract=True, filepath=tmp_path
)
assert os.path.exists(file_path)
os.remove(file_path)


def test_update_missing_id(server) -> None:
Expand Down
5 changes: 2 additions & 3 deletions test/test_flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,15 +33,14 @@ def server():
return server


def test_download(server: TSC.Server) -> None:
def test_download(server: TSC.Server, tmp_path: Path) -> None:
with requests_mock.mock() as m:
m.get(
server.flows.baseurl + "/587daa37-b84d-4400-a9a2-aa90e0be7837/content",
headers={"Content-Disposition": 'name="tableau_flow"; filename="FlowOne.tfl"'},
)
file_path = server.flows.download("587daa37-b84d-4400-a9a2-aa90e0be7837")
file_path = server.flows.download("587daa37-b84d-4400-a9a2-aa90e0be7837", filepath=tmp_path)
assert os.path.exists(file_path) is True
os.remove(file_path)


def test_download_object(server: TSC.Server) -> None:
Expand Down
24 changes: 12 additions & 12 deletions test/test_workbook.py
Original file line number Diff line number Diff line change
Expand Up @@ -318,15 +318,14 @@ def test_update_tags(server: TSC.Server) -> None:
assert single_workbook._initial_tags == updated_workbook._initial_tags


def test_download(server: TSC.Server) -> None:
def test_download(server: TSC.Server, tmp_path: Path) -> None:
with requests_mock.mock() as m:
m.get(
server.workbooks.baseurl + "/1f951daf-4061-451a-9df1-69a8062664f2/content",
headers={"Content-Disposition": 'name="tableau_workbook"; filename="RESTAPISample.twbx"'},
)
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2")
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2", filepath=tmp_path)
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_object(server: TSC.Server) -> None:
Expand All @@ -340,21 +339,20 @@ def test_download_object(server: TSC.Server) -> None:
assert isinstance(file_path, BytesIO)


def test_download_sanitizes_name(server: TSC.Server) -> None:
def test_download_sanitizes_name(server: TSC.Server, tmp_path: Path) -> None:
filename = "Name,With,Commas.twbx"
disposition = f'name="tableau_workbook"; filename="{filename}"'
with requests_mock.mock() as m:
m.get(
server.workbooks.baseurl + "/1f951daf-4061-451a-9df1-69a8062664f2/content",
headers={"Content-Disposition": disposition},
)
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2")
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2", filepath=tmp_path)
assert os.path.basename(file_path) == "NameWithCommas.twbx"
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_extract_only(server: TSC.Server) -> None:
def test_download_extract_only(server: TSC.Server, tmp_path: Path) -> None:
# Pretend we're 2.5 for 'extract_only'
server.version = "2.5"

Expand All @@ -365,12 +363,13 @@ def test_download_extract_only(server: TSC.Server) -> None:
complete_qs=True,
)
# Technically this shouldn't download a twbx, but we are interested in the qs, not the file
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2", include_extract=False)
file_path = server.workbooks.download(
"1f951daf-4061-451a-9df1-69a8062664f2", include_extract=False, filepath=tmp_path
)
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_no_extract_emits_deprecation_warning(server: TSC.Server) -> None:
def test_download_no_extract_emits_deprecation_warning(server: TSC.Server, tmp_path: Path) -> None:
"""no_extract=True should emit a DeprecationWarning and map to includeExtract=False."""
server.version = "2.5"

Expand All @@ -381,9 +380,10 @@ def test_download_no_extract_emits_deprecation_warning(server: TSC.Server) -> No
complete_qs=True,
)
with pytest.warns(DeprecationWarning, match="deprecated and will be removed"):
file_path = server.workbooks.download("1f951daf-4061-451a-9df1-69a8062664f2", no_extract=True)
file_path = server.workbooks.download(
"1f951daf-4061-451a-9df1-69a8062664f2", no_extract=True, filepath=tmp_path
)
assert os.path.exists(file_path)
os.remove(file_path)


def test_download_missing_id(server: TSC.Server) -> None:
Expand Down
Loading