def test_schema():
json = """
{
"name" : "schema_name",
"share" : "share_name"
}
"""
schema = Schema.from_json(json)
assert schema == Schema("schema_name", "share_name")
def test_list_tables(sharing_client: SharingClient):
tables = sharing_client.list_tables(Schema(name="default", share="share1"))
assert tables == [
Table(name="table1", share="share1", schema="default"),
Table(name="table3", share="share1", schema="default"),
Table(name="table7", share="share1", schema="default"),
]
tables = sharing_client.list_tables(Schema(name="default", share="share2"))
assert tables == [Table(name="table2", share="share2", schema="default")]
def test_list_tables_with_pagination(rest_client: DataSharingRestClient):
response = rest_client.list_tables(Schema(name="default", share="share1"),
max_results=1)
assert response.tables == [
Table(name="table1", share="share1", schema="default"),
]
response = rest_client.list_tables(Schema(name="default", share="share1"),
page_token=response.next_page_token)
assert response.tables == [
Table(name="table3", share="share1", schema="default"),
Table(name="table7", share="share1", schema="default"),
]
def test_list_tables(rest_client: DataSharingRestClient):
response = rest_client.list_tables(Schema(name="default", share="share1"))
assert response.tables == [
Table(name="table1", share="share1", schema="default"),
Table(name="table3", share="share1", schema="default"),
Table(name="table7", share="share1", schema="default"),
]
response = rest_client.list_tables(Schema(name="default", share="share2"))
assert response.tables == [
Table(name="table2", share="share2", schema="default")
]
def list_schemas(self,
share: Share,
*,
max_results: Optional[int] = None,
page_token: Optional[str] = None) -> ListSchemasResponse:
data: Dict = {}
if max_results is not None:
data["maxResults"] = max_results
if page_token is not None:
data["pageToken"] = page_token
with self._get_internal(f"/shares/{share.name}/schemas",
data) as lines:
schemas_json = json.loads(next(lines))
return ListSchemasResponse(
schemas=[
Schema.from_json(schema_json)
for schema_json in schemas_json.get("items", [])
],
next_page_token=schemas_json.get("nextPageToken", None),
)
def test_list_schemas(sharing_client: SharingClient):
schemas = sharing_client.list_schemas(Share(name="share1"))
assert schemas == [Schema(name="default", share="share1")]
schemas = sharing_client.list_schemas(Share(name="share2"))
assert schemas == [Schema(name="default", share="share2")]
def test_list_schemas(rest_client: DataSharingRestClient):
response = rest_client.list_schemas(Share(name="share1"))
assert response.schemas == [Schema(name="default", share="share1")]
response = rest_client.list_schemas(Share(name="share2"))
assert response.schemas == [Schema(name="default", share="share2")]