Skip to content

Commit

Permalink
TST: Always make total_rows=0 in MockEmptyResult
Browse files Browse the repository at this point in the history
  • Loading branch information
cortadocodes committed Apr 23, 2024
1 parent 76a4867 commit 6d04c97
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tests/cloud/pub_sub/test_bigquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,8 @@
class MockEmptyResult:
"""A mock empty query result."""

def __init__(self, total_rows=0):
self.total_rows = total_rows

def result(self):
return MagicMock(total_rows=self.total_rows)
return MagicMock(total_rows=0)


class MockEmptyBigQueryClient:
Expand Down

0 comments on commit 6d04c97

Please sign in to comment.