Skip to content

Commit

Permalink
Refactor acl_consul fixture
Browse files Browse the repository at this point in the history
This change is made in preparation for future releases, anticipating the likelihood of deploying multiple instances via Docker.
  • Loading branch information
cpaillet committed May 14, 2024
1 parent 232dd9a commit 46ba105
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 21 deletions.
21 changes: 7 additions & 14 deletions tests/api/test_acl.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,7 @@

class TestConsulAcl:
def test_acl_permission_denied(self, acl_consul):
port, _master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, _master_token, _consul_version = acl_consul

# No token
pytest.raises(consul.ACLPermissionDenied, c.acl.list)
Expand Down Expand Up @@ -46,8 +45,7 @@ def test_acl_permission_denied(self, acl_consul):
)

def test_acl_list(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

# Make sure both master and anonymous tokens are created
acls = c.acl.list(token=master_token)
Expand All @@ -65,8 +63,7 @@ def test_acl_list(self, acl_consul):
assert find_recursive(acls, anonymous_token_repr)

def test_acl_read(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

# Unknown token
pytest.raises(consul.ConsulException, c.acl.read, accessor_id="unknown", token=master_token)
Expand All @@ -79,8 +76,7 @@ def test_acl_read(self, acl_consul):
assert find_recursive(acl, anonymous_token_repr)

def test_acl_create(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

c.acl.create(accessor_id="00000000-DEAD-BEEF-0000-000000000000", token=master_token)
c.acl.create(secret_id="DEADBEEF-0000-0000-0000-000000000000", token=master_token)
Expand Down Expand Up @@ -113,8 +109,7 @@ def test_acl_create(self, acl_consul):
assert find_recursive(acl, expected)

def test_acl_clone(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

assert len(c.acl.list(token=master_token)) == 2

Expand All @@ -137,8 +132,7 @@ def test_acl_clone(self, acl_consul):
assert find_recursive(acl, expected)

def test_acl_update(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

# Unknown token
pytest.raises(consul.ConsulException, c.acl.update, accessor_id="unknown", token=master_token)
Expand All @@ -157,8 +151,7 @@ def test_acl_update(self, acl_consul):
assert find_recursive(acl, expected)

def test_acl_delete(self, acl_consul):
port, master_token, _consul_version = acl_consul
c = consul.Consul(port=port)
c, master_token, _consul_version = acl_consul

assert len(c.acl.list(token=master_token)) == 2
c.acl.create(accessor_id="00000000-DEAD-BEEF-0000-000000000000", token=master_token)
Expand Down
7 changes: 4 additions & 3 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,8 @@
"1.17.3": "consul-1.17.3",
}

ACLConsul = collections.namedtuple("ACLConsul", ["instance", "token", "version"])


def get_free_ports(num, host=None):
if not host:
Expand Down Expand Up @@ -136,9 +138,8 @@ def consul_port(consul_instance):

@pytest.fixture()
def acl_consul(acl_consul_instance):
ACLConsul = collections.namedtuple("ACLConsul", ["port", "token", "version"])
port, token, version = acl_consul_instance
return ACLConsul(port, token, version)
consul_port, token, version = acl_consul_instance
return ACLConsul(Consul(port=consul_port), token, version)


@pytest.fixture()
Expand Down
8 changes: 4 additions & 4 deletions tests/test_aio.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,10 @@ async def consul_obj(consul_port):

@pytest.fixture()
async def consul_acl_obj(acl_consul):
consul_port, token, consul_version = acl_consul
c = consul.aio.Consul(port=consul_port, token=token)
yield c, consul_version
await c.close()
consul, token, consul_version = acl_consul
consul.token = token
yield consul, consul_version
await consul.close()


class TestAsyncioConsul:
Expand Down

0 comments on commit 46ba105

Please sign in to comment.