Getting started with Gremlin’s Python SDK

Ana M Medina

Sr. Chaos Engineer

Last Updated: May 29, 2020

Categories: Chaos Engineering

Introduction

Using the Gremlin UI is an easy, visual way to run Chaos Experiments. But as your applications become more reliable and able to withstand attacks, automating your Chaos will prevent you from drifting back into failure. This tutorial will get you started using Gremlin's Python SDK for the Gremlin API.

Prerequisites

To successfully complete this course, you’ll need:

• A Repl.it Account - Repl.it is an online tool that allows you collaboratively write and run code. This will be provided by your Bootcamp Manager
• A Gremlin account. This will be provided by your Bootcamp Manager as well.
• A Datadog account. This will be provided by your Bootcamp Manager as well.

Overview

The Gremlin Python SDK makes it easy to interact with Gremin's API. You can use it to get information about your account, launch attacks and see their status, create scenarios, and more. In this tutorial we'll be using the SDK to connect to Gremlin, launch an attack, get its status and return the results when it's finished.

Step 0

Log in to Repl.it, Gremlin, and Datadog with the credentials provided.

Step 1

After you've logged into Repl.it, click the "My Repls" link in the left side navigation, then open the Gremlin Python SDK project.

The project has already loaded the gremlinapi python library and the code has imported the the modules that we'll be using for the tutorial.

To connect to the Gremlin API, you'll first need to set your Gremlin Team ID. In the Gremlin application, go to your "Team Settings", then visit "Configuration", and copy your Team ID and paste it below. For example:

1config.team_id = "648737d4-XXXX-XXXXX-XXXX-9e3ac67e287d"

Step 2

You'll also need an API key to connect to Gremlin. In the Gremlin application, go to your "Account Settings", then visit "API Keys", and create an API Key in the Gremlin application and paste it below.

Copy your API Key and paste it into the code. Note that you can easily copy the api key by clicking the eye icon next to the API Key name, then clicking the "Click to copy" message that appears.

1config.api_key = "8fdb3329XXXXXXXXXXXXXXXXXXXXXX4b13e6cc"

After you've copied your Team ID and API Key, click the "run" button.

Your application will user your credentials to connect to the Gremlin API to retrieve a list of your Gremlin organization and print the results.

1all_orgs = orgs.list_orgs()
2pprint(all_orgs)

Step 3

Next we'll run an attack by using the attacks.create_attack() method. We'll also use the GremlinAttackHelper() to easily format our attack parameters. Copy and paste the following code into your application. This will launch a memory attack of 100% of yours as defined by GremlinTargetHosts. The memory attack will run for a length of 240 seconds and inject an increase of 85%

1# Memory Attack - This attack will run a memory attack consuming
2# 85% of memory for 240 seconds. It will target all hosts (100%).
3
4attack_id = attacks.create_attack(
5    body=GremlinAttackHelper(
6        target=GremlinTargetHosts(percent=100),
7        command=GremlinMemoryAttack(length=240, amount=85, amount_type="%"),
8    ).api_model()
9)
10pprint("Your attack id is: {}".format(attack_id))

Run your application to launch the attack. Click "Attacks in the left side navigation in Gremlin to view your currently running attacks and you'll see the attack you just created via code!

Among your team discuss:

• Compared to my hypothesis, how is my application handling this failure?
• Are we able to notice anything on Datadog?
• What can we do to make our application more resilient to conditions like this?

Step 4

Although the Gremlin application is a great way to visually see what attacks are running, we need to be able to verify the attack via code and report its status.

To do this, we'll create a simple loop that polls the Gremlin attack status every 5 seconds by calling the attack.get_attack() method.

Copy and paste the following code into your application:

1# Loop every 5 seconds, get the attack information and report
2# the status. Break the loop when the attack is complete.
3
4while True:
5  time.sleep(5)
6  attack_details = attacks.get_attack(guid=attack_id)
7  pprint("Attack status: {}".format(attack_details["stage"]))
8  if attack_details["stage_lifecycle"] == "Complete" \
9    or attack_details["stage_lifecycle"] == "Error":
10    break
11
12# Print the final attack status details.
13pprint("Your attack details:")
14pprint(attack_details)

Run your application to launch the attack again and see the attack status via the application.

Example Code

If you would like to see it all working, this is an example implementation you can copy and paste into your application.

1from pprint import pprint
2import time
3from gremlinapi.config import GremlinAPIConfig as config
4from gremlinapi.orgs import GremlinAPIOrgs as orgs
5from gremlinapi.attacks import GremlinAPIAttacks as attacks
6from gremlinapi.attack_helpers import GremlinAttackHelper, GremlinTargetContainers, GremlinTargetHosts, GremlinCPUAttack, GremlinMemoryAttack
7
8# Step 1: Log into Gremlin and change to the right team
9# Step 2: # In the Gremlin application, go to your "Team Settings", then visit "Configuration", and copy your Team ID and paste it below.
10config.team_id = "648737d4-XXXX-XXXXX-XXXX-9e3ac67e287d"
11
12# Step 3: In the Gremlin application, go to your "Account Settings", then visit "API Keys", and create an API Key in the Gremlin application and paste it below.
13config.api_key = "8fdb3329XXXXXXXXXXXXXXXXXXXXXX4b13e6cc"
14
15# Step 4: Get a list of all the organizations in your Gremlin account.
16all_orgs = orgs.list_orgs()
17pprint(all_orgs)
18
19
20# Step 5: Uncomment the attack below (remove the """) and [MORE INSTRUCTIONS HERE ON WHAT TO DO]
21# For more info about attacks: https://github.com/gremlin/gremlin-python/blob/master/examples/attacks.md
22"""
23attack_id = attacks.create_attack(
24    body=GremlinAttackHelper(
25        target=GremlinTargetHosts(percent=100),
26        command=GremlinMemoryAttack(length=240, amount=85, amount_type="%"),
27    ).api_model()
28)
29pprint("Your attack id is: {}".format(attack_id))
30"""
31
32# Step 6: Now that you've run the attack, your CI/CD code will need to verify that the attack has successfully completed. Uncomment the block below to check the status of your attack.
33
34attack_details = {}
35while True:
36  time.sleep(5)
37  attack_details = attacks.get_attack(guid=attack_id)
38  pprint("Attack status: {}".format(attack_details["stage"]))
39  if attack_details["stage_lifecycle"] == "Complete":
40    break
41
42pprint(attack_details)

Next Steps

This tutorial provides a brief look at how you can run an attack and get the attack status using the Gremlin Python SDK. To take your learning further, try changing the attack parameters or launching a different attack. You can find more information in the Gremlin Python SDK repository README file and in the Examples directory.

Conclusion

By using the Gremlin Python SDK, you can programmatically interact with the Gremlin platform. This provides an easy way to automate your Chaos Engineering practices.