diff --git a/.docsettings.yml b/.docsettings.yml index 317e26b199cd..a8b9d07e4c40 100644 --- a/.docsettings.yml +++ b/.docsettings.yml @@ -9,7 +9,7 @@ omitted_paths: language: python root_check_enabled: True required_readme_sections: - - "Azure .+ client library for Python" + - ^Azure (.+ client library for Python|Smoke Test for Python) - ^Getting started$ - ^Key concepts$ - ^Examples$ diff --git a/samples/smoketest/Program.py b/samples/smoketest/Program.py new file mode 100644 index 000000000000..dd43a67f9287 --- /dev/null +++ b/samples/smoketest/Program.py @@ -0,0 +1,13 @@ +from key_vault_secrets import KeyVault +from storage_blob import StorageBlob +from event_hubs import EventHub +from cosmos_db import CosmosDB + +print("==========================================") +print(" AZURE TRACK 2 SDKs SMOKE TEST") +print("==========================================") + +KeyVault().Run() +StorageBlob().Run() +EventHub().Run() +CosmosDB().Run() diff --git a/samples/smoketest/README.md b/samples/smoketest/README.md new file mode 100644 index 000000000000..6e0a21c70235 --- /dev/null +++ b/samples/smoketest/README.md @@ -0,0 +1,116 @@ +# Azure Smoke Test for Python +This sample code is a smoke test to ensure that Azure Preview for Python work while loaded into the same process by performing 2 or more actions with them. + +Libraries tested: +* keyvault-secrets +* identity +* storage-blob +* event-hubs +* cosmos + +## Getting started +### Setup Azure resources +For this sample, it is necessary to create/have the following resources in the [Azure Portal](https://portal.azure.com/): +* **App registration**: Register a new app or use an existing one. + * Under _Certificates & secrets_ create a new **client secret** and store the value in a safe place. +* **Key Vaults**: Create a new Key Vault resource or use an existing one. + * Under _Access policies_, add the app registrated in the previous step. +* **Storage acounts**: Create a container in a new or existing storage account. The container in this sample is named "mycontainer", if you want to use other name you can change the value in `BlobStorage.ts` file: +`const containerName = "mycontainer";` +* **Event Hubs**: Create an event hub inside a new or existing Event Hubs Namespace. The container in this sample is named "myeventhub", if you want to use other name you can change the value in `EventHubsTest.ts` file: `let eventHubName = "myeventhub";` +* **Azure Cosmos DB**: Create a new account or use an existing one. + +### Azure credentials +The following environment variables are needed: +* From **App Registration**, in the _Overview_ section: + * AZURE_TENANT_ID: The directory tentant ID. + * AZURE_CLIENT_ID: The application ID. + * AZURE_CLIENT_SECRET: The client secret stored previusly when creating the _client secret_. + +* From **Key Vault**, in the _Overview_ section: + * AZURE_PROJECT_URL: The DNS Name + +* From **Event Hubs**, in _Shared access policies_ section: + * EVENT_HUBS_CONNECTION_STRING: Connection string from a policy + +* From **Storage Account**, in the _Access Keys_ section: + * STORAGE_CONNECTION_STRING : A connection strings. + +* From **Azure Cosmos DB**, in the _Keys_ section, select the _Read-Write Keys_ tab: + * COSMOS_ENDPOINT: URI. + * COSMOS_KEY: Primary or secondary key. + +``` +//Bash code to create the environment variables +export AZURE_CLIENT_ID="" +export AZURE_CLIENT_SECRET="" +export AZURE_TENANT_ID="" +export EVENT_HUBS_CONNECTION_STRING="" +export AZURE_PROJECT_URL="" +export STORAGE_CONNECTION_STRING="" +export COSMOS_ENDPOINT="" +export COSMOS_KEY="" +``` + +### Running the console app +[Python](https://www.python.org/downloads/) version 3.7.4 was used to run this sample. + +In the \SmokeTest\ directory, run Program.py +``` +python .\Program.py +``` + +## Key concepts + + +## Examples +All the classes in this sample has a `Run()` method as entry point, and do not depend on each other. + +It is possible to run them individually: +```python +from KeyVaultSecrets import KeyVault + +KeyVault().Run() +``` + +They can be included in other projects by moving the class in it: +```python +from KeyVaultSecrets import KeyVault + +... + +def myTests(): + console.log("Smoke Test imported from other project") + KeyVault().Run() + +myTests() +otherFunction() +... +``` + +The classes can be used as base code and be changed to satisfied specific needs. For example, the method `EventHub().SendAndReceiveEvents()` can be change to only send events from an array given from a parameter: +```python +def SendAndReceiveEvents(self, partitionID, events): + producer = self.client.create_producer(partition_id=partitionID) + producer.send(events) + producer.close() +``` + +**Note:** The methods in the classes are not necessary independent on each other, and the order matters. For example, in order to run `BlobStorage().DeleteBlob();`, the method `BlobStorage().UploadBLob();` must be run before, since in the other way it will fail because there is not going to be a blob to delete. + +## Troubleshooting + +### Authentication +Be sure to set the environment variables and credentials required before running the sample. + +## Next steps +Check the [Azure SDK for Python Repository](https://github.com/Azure/azure-sdk-for-python/tree/master/sdk) for more samples inside the sdk folder. + +## Contributing +This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.microsoft.com. + +When you submit a pull request, a CLA-bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA. + +If you'd like to contribute to this library, please read the contributing guide to learn more about how to build and test the code. + +This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments. \ No newline at end of file diff --git a/samples/smoketest/cosmos_db.py b/samples/smoketest/cosmos_db.py new file mode 100644 index 000000000000..b4e2a56aa025 --- /dev/null +++ b/samples/smoketest/cosmos_db.py @@ -0,0 +1,84 @@ +import os +from azure.cosmos import CosmosClient +from azure.cosmos.partition_key import PartitionKey + + +class CosmosDB: + def __init__(self): + URL = os.environ["COSMOS_ENDPOINT"] + KEY = os.environ["COSMOS_KEY"] + self.client = CosmosClient(URL, {"masterKey": KEY}) + self.dbName = "pySolarSystem" + + def CreateDatabase(self): + print("Creating '{0}' database...".format(self.dbName)) + return self.client.create_database(self.dbName) + + def CreateContainer(self, db): + collectionName = "Planets" + print("Creating '{0}' collection...".format(collectionName)) + partition_key = PartitionKey(path="/id", kind="Hash") + return db.create_container(id="Planets", partition_key=partition_key) + + def CreateDocuments(self, container): + # Cosmos will look for an 'id' field in the items, if the 'id' is not specify Cosmos is going to assing a random key. + planets = [ + { + "id": "Earth", + "HasRings": False, + "Radius": 3959, + "Moons": [{"Name": "Moon"}], + }, + { + "id": "Mars", + "HasRings": False, + "Radius": 2106, + "Moons": [{"Name": "Phobos"}, {"Name": "Deimos"}], + }, + ] + + print("Inserting items in the collection...") + for planet in planets: + container.create_item(planet) + print("\t'{0}' created".format(planet["id"])) + print("\tdone") + + def SimpleQuery(self, container): + print("Quering the container...") + items = list( + container.query_items( + query="SELECT c.id FROM c", enable_cross_partition_query=True + ) + ) + print("\tdone: {0}".format(items)) + + def DeleteDatabase(self): + print("Cleaning up the resource...") + self.client.delete_database(self.dbName) + print("\tdone") + + def Run(self): + print() + print("------------------------") + print("Cosmos DB") + print("------------------------") + print("1) Create a Database") + print("2) Create a Container in the database") + print("3) Insert Documents (items) into the Container") + print("4) Delete Database (Clean up the resource)") + print() + + # Ensure that the database does not exists + try: + self.DeleteDatabase() + except: + pass + + try: + db = self.CreateDatabase() + container = self.CreateContainer(db=db) + self.CreateDocuments(container=container) + self.SimpleQuery(container=container) + finally: + # if something goes wrong, the resource should be cleaned anyway + self.DeleteDatabase() diff --git a/samples/smoketest/event_hubs.py b/samples/smoketest/event_hubs.py new file mode 100644 index 000000000000..9a0ed400b5c5 --- /dev/null +++ b/samples/smoketest/event_hubs.py @@ -0,0 +1,66 @@ +import os +from datetime import datetime +from azure.eventhub import EventHubClient, EventData, EventPosition + + +class EventHub: + def __init__(self): + # This test requires a previusly created Event Hub. + # In this example the name is "myeventhub", but it could be change below + connectionString = os.environ["EVENT_HUBS_CONNECTION_STRING"] + eventHubName = "myeventhub" + self.client = EventHubClient.from_connection_string( + connectionString, eventHubName + ) + + def GetPartitionIds(self): + print("Getting partitions id...") + partition_ids = self.client.get_partition_ids() + print("\tdone") + return partition_ids + + def SendAndReceiveEvents(self, partitionID): + with self.client.create_consumer( + consumer_group="$default", + partition_id=partitionID, + event_position=EventPosition(datetime.utcnow()), + ) as consumer: + + print("Sending events...") + with self.client.create_producer(partition_id=partitionID) as producer: + event_list = [ + EventData(b"Test Event 1 in Python"), + EventData(b"Test Event 2 in Python"), + EventData(b"Test Event 3 in Python"), + ] + producer.send(event_list) + print("\tdone") + + print("Receiving events...") + received = consumer.receive(max_batch_size=len(event_list), timeout=2) + for event_data in received: + print("\tEvent Received: " + event_data.body_as_str()) + + print("\tdone") + + if len(received) != len(event_list): + raise Exception( + "Error, expecting {0} events, but {1} were received.".format( + str(len(event_list)), str(len(received)) + ) + ) + + def Run(self): + print() + print("------------------------") + print("Event Hubs") + print("------------------------") + print("1) Get partition ID") + print("2) Send Events") + print("3) Consume Events") + print() + + partitionID = self.GetPartitionIds() + # In this sample the same partition id is going to be used for the producer and consumer, + # It is the first one, but it could be any (is not relevant as long as it is the same in both producer and consumer) + self.SendAndReceiveEvents(partitionID[0]) diff --git a/samples/smoketest/key_vault_secrets.py b/samples/smoketest/key_vault_secrets.py new file mode 100644 index 000000000000..7f27b02460f1 --- /dev/null +++ b/samples/smoketest/key_vault_secrets.py @@ -0,0 +1,46 @@ +import os +from azure.identity import DefaultAzureCredential +from azure.keyvault.secrets import SecretClient + + +class KeyVault: + def __init__(self): + # DefaultAzureCredential() expects the following environment variables: + # * AZURE_CLIENT_ID + # * AZURE_CLIENT_SECRET + # * AZURE_TENANT_ID + credential = DefaultAzureCredential() + self.secret_client = SecretClient( + vault_url=os.environ["AZURE_PROJECT_URL"], credential=credential + ) + + def SetSecret(self): + print("Setting a secret...") + self.secret_client.set_secret("secret-name", "secret-value") + print("\tdone") + + def GetSecret(self): + print("Getting a secret...") + secret = self.secret_client.get_secret("secret-name") + print("\tdone: " + secret.name) + + def DeleteSecret(self): + print("Deleting a secret...") + deleted_secret = self.secret_client.delete_secret("secret-name") + print("\tdone: " + deleted_secret.name) + + def Run(self): + print() + print("------------------------") + print("Key Vault - Secrets\nIdentity - Credential") + print("------------------------") + print("1) Set a secret") + print("2) Get that secret") + print("3) Delete that secret (Clean up the resource)") + print() + + try: + self.SetSecret() + self.GetSecret() + finally: + self.DeleteSecret() diff --git a/samples/smoketest/requirements.txt b/samples/smoketest/requirements.txt new file mode 100644 index 000000000000..eaf49d15335d --- /dev/null +++ b/samples/smoketest/requirements.txt @@ -0,0 +1,8 @@ +azure-common==1.1.23 +azure-core==1.0.0b1 +azure-cosmos==4.0.0b1 +azure-eventhub==5.0.0b1 +azure-identity==1.0.0b1 +azure-keyvault-secrets==4.0.0b1 +azure-storage-blob==12.0.0b1 +azure-storage-common==2.0.0 \ No newline at end of file diff --git a/samples/smoketest/storage_blob.py b/samples/smoketest/storage_blob.py new file mode 100644 index 000000000000..7e0c394f9580 --- /dev/null +++ b/samples/smoketest/storage_blob.py @@ -0,0 +1,51 @@ +import os +from azure.storage.blob import BlobClient +from azure.core import exceptions + + +class StorageBlob: + def __init__(self): + connectionString = os.environ["STORAGE_CONNECTION_STRING"] + self.blob = BlobClient.from_connection_string( + connectionString, container="mycontainer", blob="pyTestBlob.txt" + ) + + def UploadBLob(self): + print("uploading blob...") + self.data = "This is a sample data for Python Test" + self.blob.upload_blob(self.data) + print("\tdone") + + def DownloadBlob(self): + print("downloading blob...") + with open("./downloadedBlob.txt", "wb+") as my_blob: + my_blob.writelines(self.blob.download_blob()) + + print("\tdone") + + def DeleteBlob(self): + print("Cleaning up the resource...") + self.blob.delete_blob() + print("\tdone") + + def Run(self): + print() + print("------------------------") + print("Storage - Blob") + print("------------------------") + print("1) Upload a Blob") + print("2) Download a Blob") + print("3) Delete that Blob (Clean up the resource)") + print() + + # Ensure that the blob does not exists before the tests + try: + self.DeleteBlob() + except exceptions.AzureError: + pass + + try: + self.UploadBLob() + self.DownloadBlob() + finally: + self.DeleteBlob()