Prep firestore unit tests for generation from 'v1' protos.

firestore/tests/unit/v1beta1/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2017 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

firestore/tests/unit/test_client.py → firestore/tests/unit/v1beta1/test_client.py

@@ -25,9 +25,9 @@ class TestClient(unittest.TestCase):
 
     @staticmethod
     def _get_target_class():
-        from google.cloud import firestore
+        from google.cloud.firestore_v1beta1.client import Client
 
-        return firestore.Client
+        return Client
 
     def _make_one(self, *args, **kwargs):
         klass = self._get_target_class()

firestore/tests/unit/test_cross_language.py → firestore/tests/unit/v1beta1/test_cross_language.py

@@ -37,10 +37,10 @@ def _load_testproto(filename):
     return test_proto
 
 
-ALL_TESTPROTOS = [
-    _load_testproto(filename)
-    for filename in sorted(glob.glob("tests/unit/testdata/*.textproto"))
-]
+_here = os.path.dirname(__file__)
+_glob_expr = "{}/testdata/*.textproto".format(_here)
+_globs = glob.glob(_glob_expr)
+ALL_TESTPROTOS = [_load_testproto(filename) for filename in sorted(_globs)]
 
 _CREATE_TESTPROTOS = [
     test_proto