#!/usr/bin/python3
import unittest
import amulet
class TestDeploy(unittest.TestCase):
"""
Deployment test for Apache Kafka
"""
@classmethod
def setUpClass(cls):
cls.d = amulet.Deployment(series='trusty')
# Deploy Kafka Service
cls.d.add('kafka', charm='cs:~bigdata-dev/trusty/apache-kafka')
cls.d.add('zookeeper', charm='cs:~bigdata-dev/trusty/apache-zookeeper')
cls.d.relate('kafka:zookeeper', 'zookeeper:zookeeper')
cls.d.setup(timeout=1800)
cls.d.sentry.wait(timeout=1800)
cls.unit = cls.d.sentry['kafka'][0]
def test_deploy(self):
output, retcode = self.unit.run("pgrep -a java")
assert 'Kafka' in output, "Kafka daemon is not started"
if __name__ == '__main__':
unittest.main()