fix: handle __main__ and no package gracefully

bazel-contrib · Feb 25, 2023 · f1394d8 · f1394d8
1 parent 0da3922
commit f1394d8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/oci/private/structure_test.bzl b/oci/private/structure_test.bzl
@@ -24,9 +24,12 @@ exec "{st_path}" test {fixed_args} "$@"
 def _structure_test_impl(ctx):
     st_info = ctx.toolchains["@contrib_rules_oci//oci:st_toolchain_type"].st_info
 
-    default_image_tag = "{workspace}.local/{package}/{name}:latest".format(
-        workspace = ctx.workspace_name,
-        package = ctx.label.package.replace("/", "_"),
+    workspace = "default_workspace" if ctx.workspace_name == "__main__" else ctx.workspace_name
+    package = "{}/".format(ctx.label.package.replace("/", "_")) if ctx.label.package else ""
+
+    default_image_tag = "{workspace}.local/{optional_package}{name}:latest".format(
+        workspace = workspace,
+        optional_package = package,
         name = ctx.label.name,
     )