Fix test to make sure the initial mode is replicated

Signed-off-by: Felipe Ruhland <[email protected]>

Author: feliperuhland

docker/models/services.py

@@ -116,7 +116,7 @@ def scale(self, replicas):
             ``True``if successful.
         """
 
-        if not self.attrs['Spec']['Mode'].get('Global'):
+        if 'Global' in self.attrs['Spec']['Mode'].keys():
             raise InvalidArgument('Cannot scale a global container')
 
         service_mode = ServiceMode('replicated', replicas)

tests/integration/models_services_test.py

@@ -5,7 +5,7 @@
 from .. import helpers
 from .base import TEST_API_VERSION
 from docker.errors import InvalidArgument
-from docker.models.services import ServiceMode
+from docker.types.services import ServiceMode
 
 
 class ServiceTest(unittest.TestCase):
@@ -212,7 +212,7 @@ def test_scale_method_service(self):
             name=helpers.random_name(),
             # ContainerSpec arguments
             image="alpine",
-            command="sleep 300"
+            command="sleep 300",
         )
         tasks = []
         while len(tasks) == 0: