Skip to content

Commit 4198e48

Browse files
andrewsgandrewsg
committed
Add test info
1 parent a5b42c2 commit 4198e48

File tree

2 files changed

+16
-13
lines changed

2 files changed

+16
-13
lines changed

‎dlp/deid_test.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,11 @@
2323
HARMFUL_STRING = 'My SSN is 372819127'
2424
HARMLESS_STRING = 'My favorite color is blue'
2525
GCLOUD_PROJECT = os.getenv('GCLOUD_PROJECT')
26-
WRAPPED_KEY = os.getenv('DLP_DEID_WRAPPED_KEY')
27-
KEY_NAME = os.getenv('DLP_DEID_KEY_NAME')
26+
WRAPPED_KEY = ('CiQAaNd+NKZwUklWRkR/57xnFbkQX2YISRHDMpiOG4q92ISwuOkSQQASRgq4ht'
27+
'mOs+LXldmKxRvmQ+8MQz3o8xq7zSjG4N0rQbcMgPG7hONPp+PhyKVVbLNds5gM'
28+
'Kmx1jclPSTfQT+bH')
29+
KEY_NAME = ('projects/nodejs-docs-samples/locations/global/keyRings/'
30+
'integration-tests-dlp/cryptoKeys/test-key')
2831
SURROGATE_TYPE = 'SSN_TOKEN'
2932
CSV_FILE = os.path.join(os.path.dirname(__file__), 'resources/dates.csv')
3033
DATE_SHIFTED_AMOUNT = 30

‎dlp/risk_test.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import risk
2222

2323
GCLOUD_PROJECT = os.getenv('GCLOUD_PROJECT')
24+
TABLE_PROJECT = 'nodejs-docs-samples'
2425
TOPIC_ID = 'dlp-test'
2526
SUBSCRIPTION_ID = 'dlp-test-subscription'
2627
DATASET_ID = 'integration_tests_dlp'
@@ -66,7 +67,7 @@ def subscription_id(topic_id):
6667
def test_numerical_risk_analysis(topic_id, subscription_id, capsys):
6768
risk.numerical_risk_analysis(
6869
GCLOUD_PROJECT,
69-
GCLOUD_PROJECT,
70+
TABLE_PROJECT,
7071
DATASET_ID,
7172
'harmful',
7273
NUMERIC_FIELD,
@@ -81,7 +82,7 @@ def test_categorical_risk_analysis_on_string_field(
8182
topic_id, subscription_id, capsys):
8283
risk.categorical_risk_analysis(
8384
GCLOUD_PROJECT,
84-
GCLOUD_PROJECT,
85+
TABLE_PROJECT,
8586
DATASET_ID,
8687
'harmful',
8788
UNIQUE_FIELD,
@@ -94,10 +95,9 @@ def test_categorical_risk_analysis_on_string_field(
9495

9596
def test_categorical_risk_analysis_on_number_field(
9697
topic_id, subscription_id, capsys):
97-
9898
risk.categorical_risk_analysis(
9999
GCLOUD_PROJECT,
100-
GCLOUD_PROJECT,
100+
TABLE_PROJECT,
101101
DATASET_ID,
102102
'harmful',
103103
NUMERIC_FIELD,
@@ -111,7 +111,7 @@ def test_categorical_risk_analysis_on_number_field(
111111
def test_k_anonymity_analysis_single_field(topic_id, subscription_id, capsys):
112112
risk.k_anonymity_analysis(
113113
GCLOUD_PROJECT,
114-
GCLOUD_PROJECT,
114+
TABLE_PROJECT,
115115
DATASET_ID,
116116
'harmful',
117117
topic_id,
@@ -127,7 +127,7 @@ def test_k_anonymity_analysis_multiple_fields(topic_id, subscription_id,
127127
capsys):
128128
risk.k_anonymity_analysis(
129129
GCLOUD_PROJECT,
130-
GCLOUD_PROJECT,
130+
TABLE_PROJECT,
131131
DATASET_ID,
132132
'harmful',
133133
topic_id,
@@ -142,7 +142,7 @@ def test_k_anonymity_analysis_multiple_fields(topic_id, subscription_id,
142142
def test_l_diversity_analysis_single_field(topic_id, subscription_id, capsys):
143143
risk.l_diversity_analysis(
144144
GCLOUD_PROJECT,
145-
GCLOUD_PROJECT,
145+
TABLE_PROJECT,
146146
DATASET_ID,
147147
'harmful',
148148
topic_id,
@@ -160,7 +160,7 @@ def test_l_diversity_analysis_multiple_field(
160160
topic_id, subscription_id, capsys):
161161
risk.l_diversity_analysis(
162162
GCLOUD_PROJECT,
163-
GCLOUD_PROJECT,
163+
TABLE_PROJECT,
164164
DATASET_ID,
165165
'harmful',
166166
topic_id,
@@ -178,7 +178,7 @@ def test_k_map_estimate_analysis_single_field(
178178
topic_id, subscription_id, capsys):
179179
risk.k_map_estimate_analysis(
180180
GCLOUD_PROJECT,
181-
GCLOUD_PROJECT,
181+
TABLE_PROJECT,
182182
DATASET_ID,
183183
'harmful',
184184
topic_id,
@@ -196,7 +196,7 @@ def test_k_map_estimate_analysis_multiple_field(
196196
topic_id, subscription_id, capsys):
197197
risk.k_map_estimate_analysis(
198198
GCLOUD_PROJECT,
199-
GCLOUD_PROJECT,
199+
TABLE_PROJECT,
200200
DATASET_ID,
201201
'harmful',
202202
topic_id,
@@ -215,7 +215,7 @@ def test_k_map_estimate_analysis_quasi_ids_info_types_equal(
215215
with pytest.raises(ValueError):
216216
risk.k_map_estimate_analysis(
217217
GCLOUD_PROJECT,
218-
GCLOUD_PROJECT,
218+
TABLE_PROJECT,
219219
DATASET_ID,
220220
'harmful',
221221
topic_id,

0 commit comments

Comments
 (0)