def test__get_reference_failure(): from google.cloud.firestore_v1.base_client import _BAD_DOC_TEMPLATE from google.cloud.firestore_v1.base_client import _get_reference doc_path = "1/888/call-now" with pytest.raises(ValueError) as exc_info: _get_reference(doc_path, {}) err_msg = _BAD_DOC_TEMPLATE.format(doc_path) assert exc_info.value.args == (err_msg, )
def _call_fut(document_path, reference_map): from google.cloud.firestore_v1.base_client import _get_reference return _get_reference(document_path, reference_map)
def test__get_reference_success(): from google.cloud.firestore_v1.base_client import _get_reference doc_path = "a/b/c" reference_map = {doc_path: mock.sentinel.reference} assert _get_reference(doc_path, reference_map) is mock.sentinel.reference