File tree Expand file tree Collapse file tree 1 file changed +45
-0
lines changed Expand file tree Collapse file tree 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