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")]