Skip to content

Commit d17b9d5

Browse files
committed
Fix mocking in test_online_store_rest_client.py
1 parent 227a4e1 commit d17b9d5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/tests/core/test_online_store_rest_client.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515
#
16-
import hsfs
16+
import hopsworks
1717
import pytest
1818
from furl import furl
1919
from hsfs.client import auth, exceptions, online_store_rest_client
@@ -50,7 +50,7 @@ def test_setup_rest_client_external(self, mocker, monkeypatch):
5050
def client_get_instance():
5151
return MockExternalClient()
5252

53-
monkeypatch.setattr(hsfs.client, "get_instance", client_get_instance)
53+
monkeypatch.setattr(hopsworks.client, "get_instance", client_get_instance)
5454
variable_api_mock = mocker.patch(
5555
"hsfs.core.variable_api.VariableApi.get_loadbalancer_external_domain",
5656
return_value="app.hopsworks.ai",
@@ -86,7 +86,7 @@ def test_setup_online_store_rest_client_internal(self, mocker, monkeypatch):
8686
def client_get_instance():
8787
return MockInternalClient()
8888

89-
monkeypatch.setattr(hsfs.client, "get_instance", client_get_instance)
89+
monkeypatch.setattr(hopsworks.client, "get_instance", client_get_instance)
9090
variable_api_mock = mocker.patch(
9191
"hsfs.core.variable_api.VariableApi.get_service_discovery_domain",
9292
return_value="consul",

0 commit comments

Comments
 (0)