diff --git a/tensorflow/security/fuzzing/cc/BUILD b/tensorflow/security/fuzzing/cc/BUILD index c32a54ab..f9557657 100644 --- a/tensorflow/security/fuzzing/cc/BUILD +++ b/tensorflow/security/fuzzing/cc/BUILD @@ -8,19 +8,24 @@ load( "//tensorflow/security/fuzzing:tf_fuzzing.bzl", "tf_cc_fuzz_test", ) +load( + "//tensorflow:tensorflow.bzl", + "tf_cc_test", +) package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], licenses = ["notice"], ) -tf_cc_fuzz_test( +tf_cc_test( name = "status_fuzz", srcs = ["status_fuzz.cc"], - tags = ["no_oss"], deps = [ ":fuzz_domains", "//tensorflow/core/platform:status", + "@com_google_fuzztest//fuzztest", + "@com_google_fuzztest//fuzztest:fuzztest_gtest_main", ], ) @@ -34,32 +39,34 @@ tf_cc_fuzz_test( ], ) -tf_cc_fuzz_test( +tf_cc_test( name = "base64_fuzz", srcs = ["base64_fuzz.cc"], - tags = ["no_oss"], deps = [ + ":fuzz_domains", "//tensorflow/core/platform:base64", "//tensorflow/core/platform:status", "//tensorflow/core/platform:stringpiece", + "@com_google_fuzztest//fuzztest", + "@com_google_fuzztest//fuzztest:fuzztest_gtest_main", ], ) -tf_cc_fuzz_test( +tf_cc_test( name = "bfloat16_fuzz", srcs = ["bfloat16_fuzz.cc"], - tags = ["no_oss"], # b/175698644 deps = [ "//tensorflow/core:test", "//tensorflow/core/framework:bfloat16", + "@com_google_fuzztest//fuzztest", + "@com_google_fuzztest//fuzztest:fuzztest_gtest_main", ], ) -tf_cc_fuzz_test( +tf_cc_test( name = "checkpoint_reader_fuzz", srcs = ["checkpoint_reader_fuzz.cc"], data = glob(["checkpoint_reader_testdata/*"]), - tags = ["no_oss"], deps = [ ":checkpoint_reader_fuzz_input_proto_cc", "//tensorflow/c:checkpoint_reader", @@ -70,6 +77,8 @@ tf_cc_fuzz_test( "//tensorflow/core/framework:types_proto_cc", "//tensorflow/core/platform:resource_loader", "//tensorflow/core/platform:status", + "@com_google_fuzztest//fuzztest", + "@com_google_fuzztest//fuzztest:fuzztest_gtest_main", ], ) @@ -83,13 +92,14 @@ tf_proto_library( ], ) -tf_cc_fuzz_test( +tf_cc_test( name = "cleanpath_fuzz", srcs = ["cleanpath_fuzz.cc"], - tags = ["no_oss"], deps = [ "//tensorflow/core/platform:path", "@com_google_absl//absl/strings", + "@com_google_fuzztest//fuzztest", + "@com_google_fuzztest//fuzztest:fuzztest_gtest_main", ], )