Commit2c81a97

authored

Switch to container auth token for pods client (#10)

1 parent296e483 commit2c81a97Copy full SHA for 2c81a97

File tree

+20

-7

lines changed

+20

-7

lines changed

Lines changed: 8 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,3 @@`
`1`		`-importos`
`2`		`-`
`3`	`1`	`fromlocalstack.sdk.api_clientimportApiClient`
`4`	`2`	`fromlocalstack.sdk.configurationimportConfiguration`
`5`	`3`
`@@ -12,7 +10,14 @@ class BaseClient:`
`12`	`10`	`auth_token:str\|None`
`13`	`11`
`14`	`12`	`def__init__(self,host:str\|None=None,auth_token:str\|None=None,**kwargs)->None:`
	`13`	`+"""`
	`14`	`+ Initialize a base client to interact with LocalStack developer endpoint.`
	`15`	`+ :param host: the host, http://localhost.localstack.cloud:4566 by default.`
	`16`	`+ :param auth_token: if provided, this token would be used for authentication against platform. It not, the`
	`17`	`+ LocalStack runtime will use the one used to start the container. The token used determines the Cloud`
	`18`	`+ Pods identity, i.e., which pods are available.`
	`19`	`+ """`
`15`	`20`	`_host=hostor"http://localhost.localstack.cloud:4566"`
`16`		`-self.auth_token=auth_tokenoros.getenv("LOCALSTACK_AUTH_TOKEN","").strip("'\" ")`
	`21`	`+self.auth_token=auth_token`
`17`	`22`	`self.configuration=Configuration(host=_host)`
`18`	`23`	`self._api_client=ApiClient(configuration=self.configuration)`

Lines changed: 6 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -24,10 +24,12 @@ class PodsClient(BaseClient):`
`24`	`24`	`def__init__(self,**args)->None:`
`25`	`25`	`super().__init__(**args)`
`26`	`26`	`self._client=PodsApi(self._api_client)`
`27`		`-# https://github.com/localstack/localstack-ext/pull/3469 could be avoided after this`
`28`		`-assertself.auth_token`
`29`		`-auth_header=get_platform_auth_header(self.auth_token)`
`30`		`-self._api_client.set_default_header("Authorization",auth_header["Authorization"])`
	`27`	`+ifself.auth_token:`
	`28`	`+# If an auth token is provided, it will be used to authenticate platform calls for Cloud Pods.`
	`29`	`+# Only the pods tied to this token will be visible. If not provided, the token will be fetched from the`
	`30`	`+# container. This allows to separate container identity for caller identity, if needed.`
	`31`	`+auth_header=get_platform_auth_header(self.auth_token)`
	`32`	`+self._api_client.set_default_header("Authorization",auth_header["Authorization"])`
`31`	`33`
`32`	`34`	`defsave_pod(self,pod_name:str)->None:`
`33`	`35`	`"""`

Lines changed: 6 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+importpytest`
	`2`	`+`
`1`	`3`	`fromlocalstack.sdk.podsimportPodsClient`
`2`	`4`
`3`	`5`	`POD_NAME="ls-sdk-integration"`
`@@ -14,3 +16,7 @@ def test_pod_crud(self):`
`14`	`16`	`self.client.save_pod(pod_name=POD_NAME)`
`15`	`17`	`self.client.load_pod(pod_name=POD_NAME)`
`16`	`18`	`self.client.delete_pod(pod_name=POD_NAME)`
	`19`	`+`
	`20`	`+deftest_not_existing_pod(self):`
	`21`	`+withpytest.raises(Exception):`
	`22`	`+self.client.load_pod(pod_name="i-do-not-exists")`

Comments

(0)