castor-extractor 0.23.3__py3-none-any.whl → 0.24.0__py3-none-any.whl

castor_extractor/visualization/tableau/tests/unit/rest_api/credentials_test.py

@@ -1,13 +0,0 @@
-from ....client.credentials import CredentialsApi
-
-
-def test_default_site_id():
-    creds = CredentialsApi(  # noqa: S106
-        server_url="url",
-        user="test_user",
-        password="test_pwd",
-        token="token",
-        token_name="token_name",
-        site_id=None,
-    )
-    assert creds.to_dict()["site_id"] == ""

castor_extractor/visualization/tableau/tests/unit/rest_api/projects_test.py

@@ -1,59 +0,0 @@
-import os
-import unittest
-from unittest import mock
-
-import requests_mock  # type: ignore
-import tableauserverclient as TSC  # type: ignore
-
-from ....client import ApiClient
-from ..utils import KEYS
-
-TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), "../assets/rest_api/")
-
-GET_XML = os.path.join(TEST_ASSET_DIR, "project_get.xml")
-
-EXPECTED = [
-    {
-        "id": "ee8c6e70-43b6-11e6-af4f-f7b0d8e20760",
-        "name": "default",
-        "description": "The default project that was automatically created by Tableau.",
-        "parent_id": None,
-        "folder_path": "/default",
-    },
-    {
-        "id": "1d0304cd-3796-429f-b815-7258370b9b74",
-        "name": "Tableau",
-        "description": None,
-        "parent_id": None,
-        "folder_path": "/Tableau",
-    },
-    {
-        "id": "4cc52973-5e3a-4d1f-a4fb-5b5f73796edf",
-        "name": "Tableau/Child_1",
-        "description": None,
-        "parent_id": "1d0304cd-3796-429f-b815-7258370b9b74",
-        "folder_path": "/Tableau/Tableau/Child_1",
-    },
-]
-
-
-class ProjectTests(unittest.TestCase):
-    @mock.patch.dict(os.environ, KEYS)
-    def setUp(self):
-        self._client = ApiClient()
-        self._client._server = TSC.Server("http://test")
-
-        # Fake signin
-        self._client._server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
-        self._client._server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
-
-        self.baseurl = self._client._server.projects.baseurl
-
-    def test_fetch_projects(self):
-        with open(GET_XML, "rb") as f:
-            response_xml = f.read().decode("utf-8")
-        with requests_mock.mock() as m:
-            m.get(self.baseurl, text=response_xml)
-            results = self._client._fetch_projects()
-
-        self.assertEqual(results, EXPECTED)

castor_extractor/visualization/tableau/tests/unit/rest_api/usages_test.py

@@ -1,49 +0,0 @@
-import os
-import unittest
-from unittest import mock
-
-import requests_mock  # type: ignore
-import tableauserverclient as TSC  # type: ignore
-
-from ....client import ApiClient
-from ..utils import KEYS
-
-TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), "../assets/rest_api/")
-
-GET_XML = os.path.join(TEST_ASSET_DIR, "view_get_usage.xml")
-
-EXPECTED = [
-    {
-        "workbook_id": "3cc6cd06-89ce-4fdc-b935-5294135d6d42",
-        "view_counts": 11,
-    },
-    {
-        "workbook_id": "6d13b0ca-043d-4d42-8c9d-3f3313ea3a00",
-        "view_counts": 13,
-    },
-]
-
-
-class UsageTests(unittest.TestCase):
-    @mock.patch.dict(os.environ, KEYS)
-    def setUp(self):
-        self._client = ApiClient()
-        self._client._server = TSC.Server("http://test_usage")
-
-        # Fake signin
-        self._client._server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
-        self._client._server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
-
-        self.baseurl = self._client._server.views.baseurl
-
-    def test_fetch_usages(self):
-        with open(GET_XML, "rb") as f:
-            response_xml = f.read().decode("utf-8")
-        with requests_mock.mock() as m:
-            m.get(
-                self.baseurl + "?includeUsageStatistics=true",
-                text=response_xml,
-            )
-            results = self._client._fetch_usages(False)
-
-            self.assertEqual(results, EXPECTED)

castor_extractor/visualization/tableau/tests/unit/rest_api/users_test.py

@@ -1,52 +0,0 @@
-import os
-import unittest
-from unittest import mock
-
-import requests_mock  # type: ignore
-import tableauserverclient as TSC  # type: ignore
-
-from ....client import ApiClient
-from ..utils import KEYS
-
-TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), "../assets/rest_api/")
-
-GET_XML = os.path.join(TEST_ASSET_DIR, "user_get.xml")
-
-EXPECTED = [
-    {
-        "id": "dd2239f6-ddf1-4107-981a-4cf94e415794",
-        "name": "alice",
-        "email": "alicecook@test.com",
-        "fullname": "alice cook",
-        "site_role": "Publisher",
-    },
-    {
-        "id": "2a47bbf8-8900-4ebb-b0a4-2723bd7c46c3",
-        "name": "Bob",
-        "email": "bob@test.com",
-        "fullname": "Bob Smith",
-        "site_role": "Interactor",
-    },
-]
-
-
-class UserTests(unittest.TestCase):
-    @mock.patch.dict(os.environ, KEYS)
-    def setUp(self):
-        self._client = ApiClient()
-        self._client._server = TSC.Server("http://test")
-
-        # Fake signin
-        self._client._server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
-        self._client._server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
-
-        self.baseurl = self._client._server.users.baseurl
-
-    def test_fetch_users(self):
-        with open(GET_XML, "rb") as f:
-            response_xml = f.read().decode("utf-8")
-        with requests_mock.mock() as m:
-            m.get(self.baseurl + "?fields=_all_", text=response_xml)
-            results = self._client._fetch_users()
-
-        self.assertEqual(results, EXPECTED)

castor_extractor/visualization/tableau/tests/unit/rest_api/workbooks_test.py

@@ -1,60 +0,0 @@
-import os
-import unittest
-from unittest import mock
-
-import requests_mock  # type: ignore
-import tableauserverclient as TSC  # type: ignore
-
-from ....client import ApiClient
-from ..utils import KEYS
-
-TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), "../assets/rest_api/")
-
-GET_XML = os.path.join(TEST_ASSET_DIR, "workbook_get.xml")
-
-EXPECTED = [
-    {
-        "id": "6d13b0ca-043d-4d42-8c9d-3f3313ea3a00",
-        "name": "Superstore",
-        "description": "description for Superstore",
-        "tags": set(),
-        "project_id": "ee8c6e70-43b6-11e6-af4f-f7b0d8e20760",
-        "created_at": None,
-        "updated_at": None,
-        "owner_id": "5de011f8-5aa9-4d5b-b991-f462c8dd6bb7",
-        "webpage_url": "http://tableauserver/#/workbooks/1/views",
-    },
-    {
-        "id": "3cc6cd06-89ce-4fdc-b935-5294135d6d42",
-        "name": "SafariSample",
-        "description": "description for SafariSample",
-        "tags": {"Sample", "Safari"},
-        "project_id": "ee8c6e70-43b6-11e6-af4f-f7b0d8e20760",
-        "created_at": None,
-        "updated_at": None,
-        "owner_id": "5de011f8-5aa9-4d5b-b991-f462c8dd6bb7",
-        "webpage_url": "http://tableauserver/#/workbooks/2/views",
-    },
-]
-
-
-class WorkbookTests(unittest.TestCase):
-    @mock.patch.dict(os.environ, KEYS)
-    def setUp(self):
-        self._client = ApiClient(test=True)
-        self._client._server = TSC.Server("http://test")
-
-        # Fake signin
-        self._client._server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
-        self._client._server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
-
-        self.baseurl = self._client._server.workbooks.baseurl
-
-    def test_fetch_workbooks(self):
-        with open(GET_XML, "rb") as f:
-            response_xml = f.read().decode("utf-8")
-        with requests_mock.mock() as m:
-            m.get(self.baseurl, text=response_xml)
-            results = self._client._fetch_workbooks()
-
-            self.assertEqual(results, EXPECTED)

castor_extractor/visualization/tableau/tests/unit/utils/__init__.py

@@ -1 +0,0 @@
-from .env_key import KEYS

castor_extractor/visualization/tableau/tests/unit/utils/env_key.py

@@ -1,6 +0,0 @@
-KEYS = {
-    "CASTOR_TABLEAU_USER": "Test",
-    "CASTOR_TABLEAU_SERVER_URL": "http://test",
-    "CASTOR_TABLEAU_PASSWORD": "Test",
-    "CASTOR_TABLEAU_SITE_ID": "dad65087-b08b-4603-af4e-2887b8aafc67",
-}

castor_extractor/visualization/tableau/tsc_fields.py

@@ -1,46 +0,0 @@
-# TSC for TableauServerClient: basic REST API to extracting core objects
-
-from .assets import TableauAsset
-
-# TSC fields extracted per assets
-TSC_FIELDS: dict[TableauAsset, set[str]] = {
-    TableauAsset.PROJECT: {
-        "id",
-        "name",
-        "description",
-        "parent_id",
-    },
-    TableauAsset.PUBLISHED_DATASOURCE: {
-        "id",
-        "name",
-        "description",
-        "tags",
-        "project_id",
-        "created_at",
-        "updated_at",
-        "owner_id",
-        "webpage_url",
-    },
-    TableauAsset.USAGE: {
-        "workbook_id",
-        "total_views",
-    },
-    TableauAsset.USER: {
-        "id",
-        "name",
-        "email",
-        "fullname",
-        "site_role",
-    },
-    TableauAsset.WORKBOOK: {
-        "id",
-        "name",
-        "description",
-        "tags",
-        "project_id",
-        "created_at",
-        "updated_at",
-        "owner_id",
-        "webpage_url",
-    },
-}

castor_extractor/visualization/tableau/types.py

@@ -1,11 +0,0 @@
-from typing import Union
-
-from tableauserverclient import ServerResponseError  # type: ignore
-from typing_extensions import Literal
-
-from .errors import TableauErrorCode
-
-PageReturn = Union[
-    tuple[list[dict], Literal[None]],
-    tuple[Literal[None], Union[TableauErrorCode, ServerResponseError]],
-]

castor_extractor/visualization/tableau/usage.py

@@ -1,14 +0,0 @@
-from ...utils import SerializedAsset, group_by
-
-
-def compute_usage_views(usages: SerializedAsset) -> SerializedAsset:
-    """Compute usages views with group by workbook_id"""
-    grouped = group_by("workbook_id", usages)
-    aggregated = [
-        {
-            "workbook_id": key,
-            "view_counts": sum(x["total_views"] for x in value),
-        }
-        for key, value in grouped.items()
-    ]
-    return aggregated