Make use of GetDataDependencyFilepath and JoinPath to build paths which will

work across operating systems.

The previous implementation doesn't work correctly on Windows.

PiperOrigin-RevId: 296251590
Change-Id: Iac5ad8dbf78d06969a51b9476f66e0b8affdaaa4
This commit is contained in:
Brian Atkinson 2020-02-20 11:04:07 -08:00 committed by TensorFlower Gardener
parent 95436d6125
commit 1fa03ff291
2 changed files with 9 additions and 7 deletions

View File

@ -428,6 +428,7 @@ tf_cc_test(
"//tensorflow/core:test_main",
"//tensorflow/core/platform:base64",
"//tensorflow/core/platform:path",
"//tensorflow/core/platform:resource_loader",
"//tensorflow/core/platform:scanner",
"@boringssl//:crypto",
],

View File

@ -25,13 +25,16 @@ limitations under the License.
#include "tensorflow/core/platform/cloud/http_request_fake.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/path.h"
#include "tensorflow/core/platform/resource_loader.h"
#include "tensorflow/core/platform/scanner.h"
#include "tensorflow/core/platform/test.h"
namespace tensorflow {
namespace {
constexpr char kTestData[] = "core/platform/cloud/testdata/";
string TestData() {
return io::JoinPath("tensorflow", "core", "platform", "cloud", "testdata");
}
constexpr char kTokenJson[] = R"(
{
@ -92,9 +95,8 @@ TEST(OAuthClientTest, GetTokenFromRefreshTokenJson) {
}
TEST(OAuthClientTest, GetTokenFromServiceAccountJson) {
std::ifstream credentials(
io::JoinPath(io::JoinPath(testing::TensorFlowSrcRoot(), kTestData),
"service_account_credentials.json"));
std::ifstream credentials(GetDataDependencyFilepath(
io::JoinPath(TestData(), "service_account_credentials.json")));
ASSERT_TRUE(credentials.is_open());
Json::Value json;
Json::Reader reader;
@ -135,9 +137,8 @@ TEST(OAuthClientTest, GetTokenFromServiceAccountJson) {
// Check that 'signature' signs 'header_dot_claim'.
// Read the serialized public key.
std::ifstream public_key_stream(
io::JoinPath(io::JoinPath(testing::TensorFlowSrcRoot(), kTestData),
"service_account_public_key.txt"));
std::ifstream public_key_stream(GetDataDependencyFilepath(
io::JoinPath(TestData(), "service_account_public_key.txt")));
string public_key_serialized(
(std::istreambuf_iterator<char>(public_key_stream)),
(std::istreambuf_iterator<char>()));