Fixing integration tests

Signed-off-by: Viktor Adam <rycus86@gmail.com>

tests/integration/api_network_test.py

@@ -118,6 +118,7 @@ class TestNetworks(BaseAPIIntegrationTest):
                 key
                 for net in network_list
                 for key in net['Containers'].keys()
+                if net['Id'] == net_id
             ),
             [container['Id']]
         )
@@ -156,6 +157,7 @@ class TestNetworks(BaseAPIIntegrationTest):
                 key
                 for net in network_list
                 for key in net['Containers'].keys()
+                if net['Id'] == net_id
             ),
             [container['Id']]
         )
@@ -169,6 +171,7 @@ class TestNetworks(BaseAPIIntegrationTest):
             key
             for net in network_list
             for key in net['Containers'].keys()
+            if net['Id'] == net_id
         ))
 
         with pytest.raises(docker.errors.APIError):
@@ -216,6 +219,7 @@ class TestNetworks(BaseAPIIntegrationTest):
                 key
                 for net in network_list
                 for key in net['Containers'].keys()
+                if net['Id'] == net_id
             ),
             [container['Id']]
         )

tests/integration/models_networks_test.py

@@ -61,7 +61,7 @@ class NetworkTest(BaseIntegrationTest):
         assert client.networks.get(network.id).containers == [container]
         network_containers = list(
             c
-            for net in client.networks.list(greedy=True)
+            for net in client.networks.list(ids=[network.id], greedy=True)
             for c in net.containers
         )
         assert network_containers == [container]