File tree 1 file changed +45
-0
lines changed
1 file changed +45
-0
lines changed Original file line number Diff line number Diff line change
1
+ # Copyright 2016 Google Inc. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ from google .cloud import bigquery
16
+ import pytest
17
+
18
+ import quickstart
19
+
20
+
21
+ # Must match the dataset listed in quickstart.py (there's no easy way to
22
+ # extract this).
23
+ DATASET_ID = 'my_new_dataset'
24
+
25
+
26
+ @pytest .fixture
27
+ def temporary_dataset ():
28
+ """Fixture that ensures the test dataset does not exist before or
29
+ after a test."""
30
+ bigquery_client = bigquery .Client ()
31
+ dataset = bigquery_client .dataset (DATASET_ID )
32
+
33
+ if dataset .exists ():
34
+ dataset .delete ()
35
+
36
+ yield
37
+
38
+ if dataset .exists ():
39
+ dataset .delete ()
40
+
41
+
42
+ def test_quickstart (capsys , temporary_dataset ):
43
+ quickstart .run_quickstart ()
44
+ out , _ = capsys .readouterr ()
45
+ assert DATASET_ID in out
You can’t perform that action at this time.
0 commit comments