Developing the Aurora Client

The client is written in Python, and uses the Pants build tool.

Building and Testing

Building and testing the client code are both done using Pants. The relevant targets to know about are:

  • Build a client executable: ./pants binary src/main/python/apache/aurora/client:aurora
  • Test client code: ./pants test src/test/python/apache/aurora/client/cli:cli

If you want to build a source distribution of the client, you need to run ./build-support/release/make-python-sdists.

Creating Custom Builds

There are situations where you may want to plug in custom logic to the Client that may not be applicable to the open source codebase. Rather than create a whole CLI from scratch, you can easily create your own custom, drop-in replacement aurora.pex using the pants build tool.

First, create an AuroraCommandLine implementation as an entry-point for registering customizations:

from apache.aurora.client.cli.client import AuroraCommandLine

class CustomAuroraCommandLine(AuroraCommandLine):
"""Custom AuroraCommandLine for your needs"""

@property
def name(self):
  return "your-company-aurora"

@classmethod
def get_description(cls):
  return 'Your Company internal Aurora client command line'

def __init__(self):
  super(CustomAuroraCommandLine, self).__init__()
  # Add custom plugins..
  self.register_plugin(YourCustomPlugin())

def register_nouns(self):
  super(CustomAuroraCommandLine, self).register_nouns()
  # You can even add new commands / sub-commands!
  self.register_noun(YourStartUpdateProxy())
  self.register_noun(YourDeployWorkflowCommand())

Secondly, create a main entry point:

def proxy_main():
  client = CustomAuroraCommandLine()
  if len(sys.argv) == 1:
    sys.argv.append("-h")
  sys.exit(client.execute(sys.argv[1:]))

Finally, you can wire everything up with a pants BUILD file in your project directory:

python_binary(
  name='aurora',
  entry_point='your_company.aurora.client:proxy_main',
  dependencies=[
    ':client_lib'
  ]
)

python_library(
  name='client_lib',
  sources = [
    'client.py',
    'custom_plugin.py',
    'custom_command.py',
  ],
  dependencies = [
    # The Apache Aurora client
    # Any other dependencies for your custom code
  ],
)

Using the same commands to build the client as above (but obviously pointing to this BUILD file instead), you will have a drop-in replacement aurora.pex file with your customizations.

Running/Debugging

For manually testing client changes against a cluster, we use Vagrant. To start a virtual cluster, you need to install Vagrant, and then run vagrant up for the root of the aurora workspace. This will create a vagrant host named “devcluster”, with a Mesos master, a set of Mesos agents, and an Aurora scheduler.

If you have a change you would like to test in your local cluster, you’ll rebuild the client:

vagrant ssh -c 'aurorabuild client'

Once this completes, the aurora command will reflect your changes.

Running/Debugging in PyCharm

It’s possible to use PyCharm to run and debug both the client and client tests in an IDE. In order to do this, first run:

build-support/python/make-pycharm-virtualenv

This script will configure a virtualenv with all of our Python requirements. Once the script completes it will emit instructions for configuring PyCharm:

Your PyCharm environment is now set up.  You can open the project root
directory with PyCharm.

Once the project is loaded:
  - open project settings
  - click 'Project Interpreter'
  - click the cog in the upper-right corner
  - click 'Add Local'
  - select 'build-support/python/pycharm.venv/bin/python'
  - click 'OK'

Running/Debugging Tests

After following these instructions, you should now be able to run/debug tests directly from the IDE by right-clicking on a test (or test class) and choosing to run or debug:

Debug Client Test

If you’ve set a breakpoint, you can see the run will now stop and let you debug:

Debugging Client Test

Running/Debugging the Client

Actually running and debugging the client is unfortunately a bit more complex. You’ll need to create a Run configuration:

  • Go to Run → Edit Configurations
  • Click the + icon to add a new configuration.
  • Choose python and name the configuration ‘client’.
  • Set the script path to /your/path/to/aurora/src/main/python/apache/aurora/client/cli/client.py
  • Set the script parameters to the command you want to run (e.g. job status <job key>)
  • Expand the Environment section and click the ellipsis to add a new environment variable
  • Click the + at the bottom to add a new variable named AURORACONFIGROOT whose value is the path where the your cluster configuration can be found. For example, to talk to the scheduler running in the vagrant image, it would be set to /your/path/to/aurora/examples/vagrant (this is the directory where our example clusters.json is found).
  • You should now be able to run and debug this configuration!