Prep firestore unit tests for generation from 'v1' protos. (#7437)

Author: tseaver

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__helpers.py …tore/tests/unit/v1beta1/test__helpers.pyfirestore/tests/unit/test__helpers.py renamed to firestore/tests/unit/v1beta1/test__helpers.py firestore/tests/unit/test__helpers.py renamed to firestore/tests/unit/v1beta1/test__helpers.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_batch.py …restore/tests/unit/v1beta1/test_batch.pyfirestore/tests/unit/test_batch.py renamed to firestore/tests/unit/v1beta1/test_batch.py firestore/tests/unit/test_batch.py renamed to firestore/tests/unit/v1beta1/test_batch.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