def test_context_double_jeopardy():
with patch_credentials("testing"):
client = client_module.Client()
with client.context():
with pytest.raises(RuntimeError):
client.context().__enter__()
def test_context():
with patch_credentials("testing"):
client = client_module.Client()
with client.context():
context = context_module.get_context()
assert context.client is client
def test_constructor_no_args():
with patch_credentials("testing"):
client = client_module.Client()
assert client.SCOPE == ("https://www.googleapis.com/auth/datastore", )
assert client.namespace is None
assert client.host == _http.DATASTORE_API_HOST
assert client.project == "testing"
def test_client_info():
with patch_credentials("testing"):
client = client_module.Client()
agent = client.client_info.to_user_agent()
assert "google-cloud-ndb" in agent
version = agent.split("/")[1]
assert version[0].isdigit()
assert "." in version
def test_constructor_all_args():
with patch_credentials("testing") as creds:
client = client_module.Client(
project="test-project",
namespace="test-namespace",
credentials=creds,
)
assert client.namespace == "test-namespace"
assert client.project == "test-project"
def test_constructor_no_args():
patch_environ = mock.patch.dict(
"google.cloud.ndb.client.os.environ", {}, clear=True
)
with patch_environ:
with patch_credentials("testing"):
client = client_module.Client()
assert client.SCOPE == ("https://www.googleapis.com/auth/datastore",)
assert client.namespace is None
assert client.host == _http.DATASTORE_API_HOST
assert client.project == "testing"
assert client.secure is True
def test_constructor_no_args_emulator():
patch_environ = mock.patch.dict(
"google.cloud.ndb.client.os.environ",
{"DATASTORE_EMULATOR_HOST": "foo"},
)
with patch_environ:
with patch_credentials("testing"):
client = client_module.Client()
assert client.SCOPE == ("https://www.googleapis.com/auth/datastore",)
assert client.namespace is None
assert client.host == "foo"
assert client.project == "testing"
assert client.secure is False
def test_context_unfinished_business():
"""Regression test for #213.
Make sure the eventloop is exhausted inside the context.
https://github.com/googleapis/python-ndb/issues/213
"""
with patch_credentials("testing"):
client = client_module.Client()
def finish_up():
context = context_module.get_context()
assert context.client is client
with client.context():
_eventloop.call_soon(finish_up)
def test__http():
with patch_credentials("testing"):
client = client_module.Client()
with pytest.raises(NotImplementedError):
client._http
def test__determine_default():
with patch_credentials("testing"):
client = client_module.Client()
assert client._determine_default("this") == "this"
def test_constructor_get_project_from_environ(environ):
environ[environment_vars.GCD_DATASET] = "gcd-project"
with patch_credentials(None):
client = client_module.Client()
assert client.project == "gcd-project"
