aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc')
-rw-r--r--tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc81
1 files changed, 46 insertions, 35 deletions
diff --git a/tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc b/tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc
index bd362f7de5..d6b3964719 100644
--- a/tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc
+++ b/tensorflow/contrib/bigtable/kernels/test_kernels/bigtable_test_client_test.cc
@@ -21,34 +21,37 @@ namespace tensorflow {
namespace {
void WriteCell(const string& row, const string& family, const string& column,
- const string& value, ::bigtable::noex::Table* table) {
- ::bigtable::SingleRowMutation mut(row);
- mut.emplace_back(::bigtable::SetCell(family, column, value));
+ const string& value,
+ ::google::cloud::bigtable::noex::Table* table) {
+ ::google::cloud::bigtable::SingleRowMutation mut(row);
+ mut.emplace_back(::google::cloud::bigtable::SetCell(family, column, value));
table->Apply(std::move(mut));
}
TEST(BigtableTestClientTest, EmptyRowRead) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
- ::bigtable::RowSet rowset;
+ ::google::cloud::bigtable::RowSet rowset;
rowset.Append("r1");
- auto filter = ::bigtable::Filter::Chain(::bigtable::Filter::Latest(1));
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1));
auto rows = table.ReadRows(std::move(rowset), filter);
EXPECT_EQ(rows.begin(), rows.end()) << "Some rows were returned in response!";
EXPECT_TRUE(rows.Finish().ok()) << "Error reading rows.";
}
TEST(BigtableTestClientTest, SingleRowWriteAndRead) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
- ::bigtable::RowSet rowset("r1");
- auto filter = ::bigtable::Filter::Chain(::bigtable::Filter::Latest(1));
+ ::google::cloud::bigtable::RowSet rowset("r1");
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1));
auto rows = table.ReadRows(std::move(rowset), filter);
auto itr = rows.begin();
EXPECT_NE(itr, rows.end()) << "No rows were returned in response!";
@@ -64,16 +67,17 @@ TEST(BigtableTestClientTest, SingleRowWriteAndRead) {
}
TEST(BigtableTestClientTest, MultiRowWriteAndSingleRowRead) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
WriteCell("r2", "f1", "c1", "v2", &table);
WriteCell("r3", "f1", "c1", "v3", &table);
- ::bigtable::RowSet rowset("r1");
- auto filter = ::bigtable::Filter::Chain(::bigtable::Filter::Latest(1));
+ ::google::cloud::bigtable::RowSet rowset("r1");
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1));
auto rows = table.ReadRows(std::move(rowset), filter);
auto itr = rows.begin();
@@ -90,16 +94,17 @@ TEST(BigtableTestClientTest, MultiRowWriteAndSingleRowRead) {
}
TEST(BigtableTestClientTest, MultiRowWriteAndRead) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
WriteCell("r2", "f1", "c1", "v2", &table);
WriteCell("r3", "f1", "c1", "v3", &table);
- ::bigtable::RowSet rowset("r1", "r2", "r3");
- auto filter = ::bigtable::Filter::Chain(::bigtable::Filter::Latest(1));
+ ::google::cloud::bigtable::RowSet rowset("r1", "r2", "r3");
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1));
auto rows = table.ReadRows(std::move(rowset), filter);
auto itr = rows.begin();
@@ -134,16 +139,18 @@ TEST(BigtableTestClientTest, MultiRowWriteAndRead) {
}
TEST(BigtableTestClientTest, MultiRowWriteAndPrefixRead) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
WriteCell("r2", "f1", "c1", "v2", &table);
WriteCell("r3", "f1", "c1", "v3", &table);
- auto filter = ::bigtable::Filter::Chain(::bigtable::Filter::Latest(1));
- auto rows = table.ReadRows(::bigtable::RowRange::Prefix("r"), filter);
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1));
+ auto rows =
+ table.ReadRows(::google::cloud::bigtable::RowRange::Prefix("r"), filter);
auto itr = rows.begin();
EXPECT_NE(itr, rows.end()) << "Missing rows";
@@ -177,9 +184,9 @@ TEST(BigtableTestClientTest, MultiRowWriteAndPrefixRead) {
}
TEST(BigtableTestClientTest, ColumnFiltering) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
WriteCell("r2", "f1", "c1", "v2", &table);
@@ -190,10 +197,12 @@ TEST(BigtableTestClientTest, ColumnFiltering) {
WriteCell("r2", "f2", "c1", "v2", &table);
WriteCell("r3", "f1", "c2", "v3", &table);
- auto filter = ::bigtable::Filter::Chain(
- ::bigtable::Filter::Latest(1), ::bigtable::Filter::FamilyRegex("f1"),
- ::bigtable::Filter::ColumnRegex("c1"));
- auto rows = table.ReadRows(::bigtable::RowRange::Prefix("r"), filter);
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1),
+ ::google::cloud::bigtable::Filter::FamilyRegex("f1"),
+ ::google::cloud::bigtable::Filter::ColumnRegex("c1"));
+ auto rows =
+ table.ReadRows(::google::cloud::bigtable::RowRange::Prefix("r"), filter);
auto itr = rows.begin();
EXPECT_NE(itr, rows.end()) << "Missing rows";
@@ -227,9 +236,9 @@ TEST(BigtableTestClientTest, ColumnFiltering) {
}
TEST(BigtableTestClientTest, RowKeys) {
- std::shared_ptr<::bigtable::DataClient> client_ptr =
+ std::shared_ptr<::google::cloud::bigtable::DataClient> client_ptr =
std::make_shared<BigtableTestClient>();
- ::bigtable::noex::Table table(client_ptr, "test_table");
+ ::google::cloud::bigtable::noex::Table table(client_ptr, "test_table");
WriteCell("r1", "f1", "c1", "v1", &table);
WriteCell("r2", "f1", "c1", "v2", &table);
@@ -240,10 +249,12 @@ TEST(BigtableTestClientTest, RowKeys) {
WriteCell("r2", "f2", "c1", "v2", &table);
WriteCell("r3", "f1", "c2", "v3", &table);
- auto filter = ::bigtable::Filter::Chain(
- ::bigtable::Filter::Latest(1), ::bigtable::Filter::CellsRowLimit(1),
- ::bigtable::Filter::StripValueTransformer());
- auto rows = table.ReadRows(::bigtable::RowRange::Prefix("r"), filter);
+ auto filter = ::google::cloud::bigtable::Filter::Chain(
+ ::google::cloud::bigtable::Filter::Latest(1),
+ ::google::cloud::bigtable::Filter::CellsRowLimit(1),
+ ::google::cloud::bigtable::Filter::StripValueTransformer());
+ auto rows =
+ table.ReadRows(::google::cloud::bigtable::RowRange::Prefix("r"), filter);
auto itr = rows.begin();
EXPECT_NE(itr, rows.end()) << "Missing rows";
EXPECT_EQ(itr->row_key(), "r1");
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-18 13:27:18 +0000