1515import trustme
1616from pytest_httpserver import HTTPServer
1717from werkzeug import Response
18- from zepben .auth import ZepbenTokenFetcher
18+ from zepben .ewb . auth import ZepbenTokenFetcher
1919
2020from zepben .eas import EasClient , Study , SolveConfig
2121from zepben .eas import FeederConfig , ForecastConfig , FixedTimeLoadOverride
@@ -99,7 +99,7 @@ def test_get_request_headers_adds_access_token_in_auth_header():
9999 assert headers ["authorization" ] == f"Bearer { mock_access_token } "
100100
101101
102- @mock .patch ("zepben.auth.client.zepben_token_fetcher.ZepbenTokenFetcher.fetch_token" , return_value = "test_token3" )
102+ @mock .patch ("zepben.ewb. auth.client.zepben_token_fetcher.ZepbenTokenFetcher.fetch_token" , return_value = "test_token3" )
103103def test_get_request_headers_adds_token_from_token_fetcher_in_auth_header (_ ):
104104 eas_client = EasClient (
105105 mock_host ,
@@ -113,7 +113,7 @@ def test_get_request_headers_adds_token_from_token_fetcher_in_auth_header(_):
113113 assert headers ["authorization" ] == "test_token3"
114114
115115
116- @mock .patch ("zepben.auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
116+ @mock .patch ("zepben.ewb. auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
117117 {"authType" : "AUTH0" , "audience" : mock_audience , "issuer" : "test_issuer" }, 200 ))
118118def test_create_eas_client_with_password_success (_ ):
119119 eas_client = EasClient (
@@ -136,7 +136,7 @@ def test_create_eas_client_with_password_success(_):
136136 assert eas_client ._verify_certificate == mock_verify_certificate
137137
138138
139- @mock .patch ("zepben.auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
139+ @mock .patch ("zepben.ewb. auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
140140 {"authType" : "AUTH0" , "audience" : mock_audience , "issuer" : "test_issuer" }, 200 ))
141141def test_create_eas_client_with_client_secret_success (_ ):
142142 eas_client = EasClient (
@@ -521,7 +521,7 @@ def test_raises_error_if_token_fetcher_and_creds_configured(httpserver: HTTPServ
521521 )
522522
523523
524- @mock .patch ("zepben.auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
524+ @mock .patch ("zepben.ewb. auth.client.zepben_token_fetcher.requests.get" , side_effect = lambda * args , ** kwargs : MockResponse (
525525 {"authType" : "AUTH0" , "audience" : mock_audience , "issuer" : "test_issuer" }, 200 ))
526526def test_allows_secret_and_creds_configured (httpserver : HTTPServer ):
527527 eas_client = EasClient (
0 commit comments