-
Notifications
You must be signed in to change notification settings - Fork 1.6k
Add system test for Database convenience methods: #3385
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add system test for Database convenience methods: #3385
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks mostly fine
spanner/tests/system.py
Outdated
@@ -228,16 +247,49 @@ def test_update_database_ddl(self): | ||
|
||
self.assertEqual(len(temp_db.ddl_statements), len(DDL_STATEMENTS)) | ||
|
||
def test_db_batch_insert_then_db_snapshot_read_and_db_read(self): | ||
from google.cloud.spanner import KeySet |
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
spanner/tests/system.py
Outdated
@@ -228,16 +247,49 @@ def test_update_database_ddl(self): | ||
|
||
self.assertEqual(len(temp_db.ddl_statements), len(DDL_STATEMENTS)) | ||
|
||
def test_db_batch_insert_then_db_snapshot_read_and_db_read(self): | ||
from google.cloud.spanner import KeySet | ||
keyset = KeySet(all_=True) |
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
spanner/tests/system.py
Outdated
retry(self._db.reload)() | ||
|
||
with self._db.batch() as batch: | ||
batch.delete(self.TABLE, keyset) |
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
spanner/tests/system.py
Outdated
batch.insert(self.TABLE, self.COLUMNS, self.ROW_DATA) | ||
|
||
with self._db.snapshot(read_timestamp=batch.committed) as snapshot: | ||
from_snap = list(snapshot.read( self.TABLE, self.COLUMNS, keyset)) |
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
spanner/tests/system.py
Outdated
with self._db.batch() as batch: | ||
batch.delete(self.TABLE, keyset) | ||
|
||
def _unit_of_work(transaction, test): |
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
This comment was marked as spam.
This comment was marked as spam.
Sorry, something went wrong.
Should I follow suit and push the updated branch to the origin to get a build with system tests? We could close this PR and re-open there. |
- batch - snapshot - read - run_with_transaction - execute_sql Closes #3015.
@tseaver Yes I think you should push this branch to the GCP origin to get system tests running. You don't need to close this PR though, the tests will "just work" (i.e. be associated here). |
Pushed to |
@dhermes New system tests pass on CI. Any further issues? |
LGTM |
…on-docs-samples#3385) * video: use flaky on flaky timeouts * Update beta_snippets_test.py * remove api result check * Update beta_snippets_test.py
batch
snapshot
read
run_with_transaction
execute_sql
Closes #3015.