Skip to content

Commit

Permalink
fix(datastore): Update Service calls to run_query and commit
Browse files Browse the repository at this point in the history
* Change positional params to keyword args

refs: #4461
pr:  #4482
  • Loading branch information
quartzmo authored Dec 5, 2019
1 parent 001da52 commit b36c30a
Show file tree
Hide file tree
Showing 10 changed files with 109 additions and 108 deletions.
17 changes: 9 additions & 8 deletions google-cloud-datastore/lib/google/cloud/datastore/service.rb
Original file line number Diff line number Diff line change
Expand Up @@ -107,12 +107,11 @@ def run_query query, namespace = nil, consistency: nil, transaction: nil
end

execute do
service.run_query project,
partition_id,
read_options: read_options,
query: query,
gql_query: gql_query,
options: default_options
service.run_query project, partition_id: partition_id,
read_options: read_options,
query: query,
gql_query: gql_query,
options: default_options
end
end

Expand Down Expand Up @@ -144,8 +143,10 @@ def begin_transaction read_only: nil, previous_transaction: nil
def commit mutations, transaction: nil
mode = transaction.nil? ? :NON_TRANSACTIONAL : :TRANSACTIONAL
execute do
service.commit project, mode, mutations, transaction: transaction,
options: default_options
service.commit project, mode: mode,
mutations: mutations,
transaction: transaction,
options: default_options
end
end

Expand Down
106 changes: 53 additions & 53 deletions google-cloud-datastore/support/doctest_helper.rb

Large diffs are not rendered by default.

56 changes: 28 additions & 28 deletions google-cloud-datastore/test/google/cloud/datastore/dataset_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@
end.to_grpc)
mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -134,7 +134,7 @@

mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test"
Expand All @@ -158,7 +158,7 @@

mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -180,7 +180,7 @@

mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test"
Expand All @@ -207,7 +207,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -238,7 +238,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -266,7 +266,7 @@

mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -288,7 +288,7 @@

mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test"
Expand Down Expand Up @@ -316,7 +316,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -348,7 +348,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -375,7 +375,7 @@
end.to_grpc)
mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -404,7 +404,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -437,7 +437,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand Down Expand Up @@ -627,7 +627,7 @@
)
mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -644,7 +644,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -665,7 +665,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -684,7 +684,7 @@
)
mode = :NON_TRANSACTIONAL
mutations = [mutation]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
dataset.delete key
Expand All @@ -698,7 +698,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -719,7 +719,7 @@
mode = :NON_TRANSACTIONAL
mutations = [mutation1, mutation2]

dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, multiple_commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity1 = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "thingie"
Expand All @@ -733,7 +733,7 @@
end

it "run will fulfill a query" do
dataset.service.mocked_service.expect :run_query, run_query_res, [project, nil, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: nil, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]

entities = dataset.run query
entities.count.must_equal 2
Expand Down Expand Up @@ -771,7 +771,7 @@
end.to_grpc), Google::Datastore::V1::Mutation.new(
delete: Google::Cloud::Datastore::Key.new("ds-test", "to-be-deleted").to_grpc)
]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode, mutations, transaction: nil, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: mode, mutations: mutations, transaction: nil, options: default_options]

entity_to_be_saved = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test", "to-be-saved"
Expand All @@ -791,7 +791,7 @@
end

it "run_query will fulfill a query" do
dataset.service.mocked_service.expect :run_query, run_query_res, [project, nil, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: nil, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]

entities = dataset.run_query query
entities.count.must_equal 2
Expand Down Expand Up @@ -821,7 +821,7 @@

it "run_query will fulfill a query with a namespace" do
partition_id = Google::Datastore::V1::PartitionId.new(namespace_id: "foobar")
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: partition_id, read_options: nil, query: query.to_grpc, gql_query: nil, options: default_options]

entities = dataset.run_query query, namespace: "foobar"
entities.count.must_equal 2
Expand Down Expand Up @@ -850,7 +850,7 @@
end

it "run will fulfill a gql query" do
dataset.service.mocked_service.expect :run_query, run_query_res, [project, nil, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: nil, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]

gql = dataset.gql "SELECT * FROM Task"
entities = dataset.run gql
Expand Down Expand Up @@ -882,7 +882,7 @@

it "run will fulfill a gql query with a namespace" do
partition_id = Google::Datastore::V1::PartitionId.new(namespace_id: "foobar")
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: partition_id, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]

gql = dataset.gql "SELECT * FROM Task"
entities = dataset.run gql, namespace: "foobar"
Expand Down Expand Up @@ -913,7 +913,7 @@
end

it "run_query will fulfill a gql query" do
dataset.service.mocked_service.expect :run_query, run_query_res, [project, nil, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: nil, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]

gql = dataset.gql "SELECT * FROM Task"
entities = dataset.run_query gql
Expand Down Expand Up @@ -945,7 +945,7 @@

it "run_query will fulfill a gql query with a namespace" do
partition_id = Google::Datastore::V1::PartitionId.new(namespace_id: "foobar")
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]
dataset.service.mocked_service.expect :run_query, run_query_res, [project, partition_id: partition_id, read_options: nil, query: nil, gql_query: gql_query_grpc, options: default_options]

gql = dataset.gql "SELECT * FROM Task"
entities = dataset.run_query gql, namespace: "foobar"
Expand Down Expand Up @@ -1181,7 +1181,7 @@
e["name"] = "thingamajig"
end.to_grpc)
dataset.service.mocked_service.expect :begin_transaction, begin_tx_res, [project, transaction_options: nil]
dataset.service.mocked_service.expect :commit, commit_res, [project, :TRANSACTIONAL, [mutation], transaction: tx_id, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: :TRANSACTIONAL, mutations: [mutation], transaction: tx_id, options: default_options]

entity = Google::Cloud::Datastore::Entity.new.tap do |e|
e.key = Google::Cloud::Datastore::Key.new "ds-test"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@
# )]
)
dataset.service.mocked_service.expect :begin_transaction, begin_tx_res, [project, transaction_options: nil]
dataset.service.mocked_service.expect :commit, commit_res, [project, :TRANSACTIONAL, [], transaction: tx_id, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: :TRANSACTIONAL, mutations: [], transaction: tx_id, options: default_options]
read_options = Google::Datastore::V1::ReadOptions.new(transaction: tx_id)
dataset.service.mocked_service.expect :lookup, first_lookup_res, [project, first_keys, read_options: read_options, options: default_options]
dataset.service.mocked_service.expect :lookup, second_lookup_res, [project, second_keys, read_options: read_options, options: default_options]
Expand Down Expand Up @@ -233,7 +233,7 @@
# )]
)
dataset.service.mocked_service.expect :begin_transaction, begin_tx_res, [project, transaction_options: nil]
dataset.service.mocked_service.expect :commit, commit_res, [project, :TRANSACTIONAL, [], transaction: tx_id, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: :TRANSACTIONAL, mutations: [], transaction: tx_id, options: default_options]
read_options = Google::Datastore::V1::ReadOptions.new(transaction: tx_id)
dataset.service.mocked_service.expect :lookup, first_lookup_res, [project, first_keys, read_options: read_options, options: default_options]
dataset.service.mocked_service.expect :lookup, second_lookup_res, [project, second_keys, read_options: read_options, options: default_options]
Expand Down Expand Up @@ -297,7 +297,7 @@
# )]
)
dataset.service.mocked_service.expect :begin_transaction, begin_tx_res, [project, transaction_options: nil]
dataset.service.mocked_service.expect :commit, commit_res, [project, :TRANSACTIONAL, [], transaction: tx_id, options: default_options]
dataset.service.mocked_service.expect :commit, commit_res, [project, mode: :TRANSACTIONAL, mutations: [], transaction: tx_id, options: default_options]
read_options = Google::Datastore::V1::ReadOptions.new(transaction: tx_id)
dataset.service.mocked_service.expect :lookup, first_lookup_res, [project, first_keys, read_options: read_options, options: default_options]
dataset.service.mocked_service.expect :lookup, second_lookup_res, [project, second_keys, read_options: read_options, options: default_options]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,8 @@

before do
dataset.service.mocked_service = Minitest::Mock.new
dataset.service.mocked_service.expect :run_query, first_run_query_res, [project, nil, read_options: nil, query: first_run_query, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, next_run_query_res, [project, nil, read_options: nil, query: next_run_query, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, first_run_query_res, [project, partition_id: nil, read_options: nil, query: first_run_query, gql_query: nil, options: default_options]
dataset.service.mocked_service.expect :run_query, next_run_query_res, [project, partition_id: nil, read_options: nil, query: next_run_query, gql_query: nil, options: default_options]
end

after do
Expand Down
Loading

0 comments on commit b36c30a

Please sign in to comment.