FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
testcontainers-python/modules/google/example_basic.py at main · guenhter/testcontainers-python · GitHub
guenhter
/
testcontainers-python
Public
forked from
testcontainers/testcontainers-python
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
testcontainers-python
/
modules
/
google
/
example_basic.py
Copy path
More file actions
More file actions
Latest commit
History
History
History
127 lines (103 loc) · 4.39 KB
Breadcrumbs
testcontainers-python
/
modules
/
google
/
example_basic.py
Copy path
File metadata and controls
127 lines (103 loc) · 4.39 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import
json
from
datetime
import
datetime
from
google
.
cloud
import
bigquery
,
datastore
,
pubsub
,
storage
from
testcontainers
.
google
import
GoogleContainer
def
basic_example
():
with
GoogleContainer
()
as
google
:
# Get connection parameters
project_id
=
google
.
project_id
# Initialize clients
storage_client
=
storage
.
Client
(
project
=
project_id
)
pubsub_client
=
pubsub
.
PublisherClient
()
bigquery_client
=
bigquery
.
Client
(
project
=
project_id
)
datastore_client
=
datastore
.
Client
(
project
=
project_id
)
print
(
"Connected to Google Cloud services"
)
# Test Cloud Storage
bucket_name
=
f"test-bucket-
{
datetime
.
utcnow
().
strftime
(
'%Y%m%d-%H%M%S'
)
}
"
bucket
=
storage_client
.
create_bucket
(
bucket_name
)
print
(
f"
\n
Created bucket:
{
bucket_name
}
"
)
# Upload a file
blob
=
bucket
.
blob
(
"test.txt"
)
blob
.
upload_from_string
(
"Hello, Google Cloud Storage!"
)
print
(
"Uploaded test file"
)
# List files
blobs
=
list
(
bucket
.
list_blobs
())
print
(
"
\n
Files in bucket:"
)
for
blob
in
blobs
:
print
(
f"-
{
blob
.
name
}
"
)
# Test Pub/Sub
topic_name
=
f"projects/
{
project_id
}
/topics/test-topic"
pubsub_client
.
create_topic
(
name
=
topic_name
)
print
(
f"
\n
Created topic:
{
topic_name
}
"
)
# Create subscription
subscription_name
=
f"projects/
{
project_id
}
/subscriptions/test-subscription"
pubsub_client
.
create_subscription
(
name
=
subscription_name
,
topic
=
topic_name
)
print
(
f"Created subscription:
{
subscription_name
}
"
)
# Publish message
message
=
"Hello, Pub/Sub!"
future
=
pubsub_client
.
publish
(
topic_name
,
message
.
encode
())
message_id
=
future
.
result
()
print
(
f"Published message:
{
message_id
}
"
)
# Test BigQuery
dataset_id
=
"test_dataset"
bigquery_client
.
create_dataset
(
dataset_id
)
print
(
f"
\n
Created dataset:
{
dataset_id
}
"
)
# Create table
table_id
=
f"
{
project_id
}
.
{
dataset_id
}
.test_table"
schema
=
[
bigquery
.
SchemaField
(
"name"
,
"STRING"
),
bigquery
.
SchemaField
(
"age"
,
"INTEGER"
),
bigquery
.
SchemaField
(
"city"
,
"STRING"
),
]
table
=
bigquery_client
.
create_table
(
bigquery
.
Table
(
table_id
,
schema
=
schema
))
print
(
f"Created table:
{
table_id
}
"
)
# Insert data
rows_to_insert
=
[
{
"name"
:
"John"
,
"age"
:
30
,
"city"
:
"New York"
},
{
"name"
:
"Jane"
,
"age"
:
25
,
"city"
:
"Los Angeles"
},
{
"name"
:
"Bob"
,
"age"
:
35
,
"city"
:
"Chicago"
},
]
errors
=
bigquery_client
.
insert_rows_json
(
table
,
rows_to_insert
)
if
not
errors
:
print
(
"Inserted test data"
)
else
:
print
(
f"Encountered errors:
{
errors
}
"
)
# Query data
query
=
f"SELECT * FROM `
{
table_id
}
` WHERE age > 30"
query_job
=
bigquery_client
.
query
(
query
)
results
=
query_job
.
result
()
print
(
"
\n
Query results:"
)
for
row
in
results
:
print
(
json
.
dumps
(
dict
(
row
),
indent
=
2
))
# Test Datastore
kind
=
"test_entity"
key
=
datastore_client
.
key
(
kind
)
entity
=
datastore
.
Entity
(
key
=
key
)
entity
.
update
({
"name"
:
"Test Entity"
,
"value"
:
42
,
"timestamp"
:
datetime
.
utcnow
()})
datastore_client
.
put
(
entity
)
print
(
f"
\n
Created
{
kind
}
entity"
)
# Query entities
query
=
datastore_client
.
query
(
kind
=
kind
)
results
=
list
(
query
.
fetch
())
print
(
"
\n
Datastore entities:"
)
for
entity
in
results
:
print
(
json
.
dumps
(
dict
(
entity
),
indent
=
2
))
# Clean up
# Delete bucket and its contents
bucket
.
delete
(
force
=
True
)
print
(
"
\n
Deleted bucket"
)
# Delete topic and subscription
pubsub_client
.
delete_subscription
(
subscription_name
)
pubsub_client
.
delete_topic
(
topic_name
)
print
(
"Deleted Pub/Sub topic and subscription"
)
# Delete BigQuery dataset and table
bigquery_client
.
delete_table
(
table_id
)
bigquery_client
.
delete_dataset
(
dataset_id
,
delete_contents
=
True
)
print
(
"Deleted BigQuery dataset and table"
)
# Delete Datastore entities
query
=
datastore_client
.
query
(
kind
=
kind
)
keys
=
[
entity
.
key
for
entity
in
query
.
fetch
()]
datastore_client
.
delete_multi
(
keys
)
print
(
"Deleted Datastore entities"
)
if
__name__
==
"__main__"
:
basic_example
()
Back
|
FazBrowse Home
|
New Git URL