bazel-skylib/tests/native_binary/assertdata.cc

62 lines
1.8 KiB
C++

// Copyright 2019 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <stdio.h>
#include <memory>
#include <string>
#include "tools/cpp/runfiles/runfiles.h"
using bazel::tools::cpp::runfiles::Runfiles;
int main(int argc, char **argv) {
std::string error;
std::unique_ptr<Runfiles> runfiles(Runfiles::Create(argv[0], &error));
if (runfiles == nullptr) {
fprintf(stderr, "ERROR(" __FILE__ ":%d): Could not init runfiles\n",
__LINE__);
return 1;
}
char* workspace = getenv("TEST_WORKSPACE");
if (workspace == nullptr) {
fprintf(stderr, "ERROR(" __FILE__ ":%d): envvar TEST_WORKSPACE is undefined\n",
__LINE__);
return 1;
}
// This should have runfiles, either from the binary itself or from the
// native_test.
std::string path =
runfiles->Rlocation(std::string(workspace) + "/tests/native_binary/testdata.txt");
FILE *f = fopen(path.c_str(), "rt");
if (!f) {
fprintf(stderr, "ERROR(" __FILE__ ":%d): Could not find runfile '%s'\n",
__LINE__, path.c_str());
}
char buf[6];
size_t s = fread(buf, 1, 5, f);
fclose(f);
buf[5] = 0;
if (s != 5 || std::string("hello") != std::string(buf, 5)) {
fprintf(stderr, "ERROR(" __FILE__ ":%d): bad runfile contents (%s)\n",
__LINE__, buf);
return 1;
}
return 0;
}