@@ -1913,9 +1913,6 @@ def setUp(self):
1913
1913
self .client = client .BigQueryClient (self .mock_bq_service , self .project )
1914
1914
self .body = {
1915
1915
'schema' : {'fields' : self .schema },
1916
- 'tableReference' : {
1917
- 'tableId' : self .table , 'projectId' : self .project ,
1918
- 'datasetId' : self .dataset }
1919
1916
}
1920
1917
self .expiration_time = 1437513693000
1921
1918
@@ -1941,7 +1938,8 @@ def test_table_patch_failed(self):
1941
1938
self .client .swallow_results = True
1942
1939
1943
1940
self .mock_tables .patch .assert_called_with (
1944
- projectId = self .project , datasetId = self .dataset , body = self .body )
1941
+ projectId = self .project , datasetId = self .dataset ,
1942
+ tableId = self .table , body = self .body )
1945
1943
1946
1944
self .mock_tables .patch .return_value .execute . \
1947
1945
assert_called_with (num_retries = 0 )
@@ -1968,7 +1966,8 @@ def test_table_patch_success(self):
1968
1966
self .client .swallow_results = True
1969
1967
1970
1968
self .mock_tables .patch .assert_called_with (
1971
- projectId = self .project , datasetId = self .dataset , body = self .body )
1969
+ projectId = self .project , datasetId = self .dataset ,
1970
+ tableId = self .table , body = self .body )
1972
1971
1973
1972
self .mock_tables .patch .return_value .execute . \
1974
1973
assert_called_with (num_retries = 0 )
0 commit comments