[FAB-3758]: Stop and Start containers

This allows for the implementation to stop/start
docker containers for the system behave tests.

Change-Id: I48ad73b8b7c2cc5eaaf51d494f4cb29a4ca316b5
Signed-off-by: Latitia M Haskins <[email protected]>

Author: lhaskins

test/feature/steps/basic_impl.py

@@ -15,6 +15,11 @@
 
 from behave import *
 import time
+import os
+import uuid
+import compose_util
+import config_util
+import endorser_util
 
 
 ORDERER_TYPES = ["solo",
@@ -33,21 +38,36 @@ def step_impl(context, seconds):
     time.sleep(float(seconds))
 
 @given(u'we compose "{composeYamlFile}"')
-def compose_impl(context, composeYamlFile):
-    pass
+def compose_impl(context, composeYamlFile, projectName=None, startContainers=True):
+    composition = compose_util.Composition(context, composeYamlFile,
+                                           projectName=projectName,
+                                           startContainers=startContainers)
+    context.compose_containers = composition.containerDataList
+    context.composition = composition
 
 @given(u'I have a bootstrapped fabric network')
 def step_impl(context):
     bootstrapped_impl(context, "solo")
 
 @given(u'I have a bootstrapped fabric network of type {networkType}')
 def bootstrapped_impl(context, networkType):
-    pass
+    assert networkType in ORDERER_TYPES, "Unknown network type '%s'" % networkType
+    curpath = os.path.realpath('.')
+    context.composeFile = "%s/docker-compose/docker-compose-%s.yml" % (curpath, networkType)
+    assert os.path.exists(context.composeFile), "The docker compose file does not exist: {0}".format(context.composeFile)
+    profile = PROFILE_TYPES.get(networkType, "SampleInsecureSolo")
+    channelID = endorser_util.TEST_CHANNEL_ID
+    projectName = str(uuid.uuid1()).replace('-','')
+    config_util.generateCrypto(projectName)
+    config_util.generateConfig(channelID, profile, projectName)
+    compose_impl(context, context.composeFile, projectName=projectName)
 
 @given(u'{component} is taken down')
 def step_impl(context, component):
-    pass
+    assert component in context.composition.collectServiceNames(), "Unknown component '{0}'".format(component)
+    context.composition.stop([component])
 
 @given(u'{component} comes back up')
 def step_impl(context, component):
-    pass
+    assert component in context.composition.collectServiceNames(), "Unknown component '{0}'".format(component)
+    context.composition.start([component])

test/feature/steps/compose_util.py

@@ -41,44 +41,51 @@ def getEnv(self, key):
 class Composition:
 
     def __init__(self, context, composeFilesYaml, projectName = None,
-                 force_recreate = True, components = []):
+                 force_recreate = True, components = [], startContainers=True):
         if not projectName:
             projectName = str(uuid.uuid1()).replace('-','')
         self.projectName = projectName
         self.context = context
         self.containerDataList = []
         self.composeFilesYaml = composeFilesYaml
-        self.serviceNames = []
-        self.serviceNames = self._collectServiceNames()
-        self.up(context, force_recreate, components)
+        if startContainers:
+            self.up(force_recreate, components)
 
-    def _collectServiceNames(self):
+    def collectServiceNames(self):
         'First collect the services names.'
         servicesList = [service for service in self.issueCommand(["config", "--services"]).splitlines() if "WARNING" not in service]
         return servicesList
 
-    def up(self, context, force_recreate=True, components=[]):
-        self.serviceNames = self._collectServiceNames()
+    def up(self, force_recreate=True, components=[]):
+        self.serviceNames = self.collectServiceNames()
         command = ["up", "-d"]
         if force_recreate:
             command += ["--force-recreate"]
         self.issueCommand(command + components)
 
-    def scale(self, context, serviceName, count=1):
-        self.serviceNames = self._collectServiceNames()
+    def scale(self, serviceName, count=1):
+        self.serviceNames = self.collectServiceNames()
         command = ["scale", "%s=%d" %(serviceName, count)]
         self.issueCommand(command)
 
-    def stop(self, context, components=[]):
-        self.serviceNames = self._collectServiceNames()
+    def stop(self, components=[]):
+        self.serviceNames = self.collectServiceNames()
         command = ["stop"]
         self.issueCommand(command, components)
 
-    def start(self, context, components=[]):
-        self.serviceNames = self._collectServiceNames()
+    def start(self, components=[]):
+        self.serviceNames = self.collectServiceNames()
         command = ["start"]
         self.issueCommand(command, components)
 
+    def docker_exec(self, command, components=[]):
+        results = {}
+        updatedCommand = " ".join(command)
+        for component in components:
+            execCommand = ["exec", component, updatedCommand]
+            results[component] = self.issueCommand(execCommand, [])
+        return results
+
     def parseComposeFilesArg(self, composeFileArgs):
         composeFileList = []
         for composeFile in composeFileArgs.split():
@@ -138,7 +145,8 @@ def issueCommand(self, command, components=[]):
             cmdArgs = command + componentList
             cmd = ["docker"] + cmdArgs
 
-        output = subprocess.check_output(cmd, env=self.getEnv())
+        process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.getEnv())
+        output, _error = process.communicate()
 
         # Don't rebuild if ps command
         if command[0] !="ps" and command[0] !="config":