refactor build rule

Author: aakash070

test/cel_go_test.bzl

@@ -89,12 +89,6 @@ def cel_go_test(
 
     test_data_path = test_data_path.lstrip("/")
 
-    if cel_expr_format != ".cel" and cel_expr_format != ".celpolicy" and cel_expr_format != ".yaml" and not is_raw_expr:
-        data = data + [cel_expr]
-
-    if file_descriptor_set != "":
-        data = data + [file_descriptor_set]
-
     if test_suite != "":
         test_suite = test_data_path + "/" + test_suite
 
@@ -112,16 +106,16 @@ def cel_go_test(
         "--base_config_path=%s" % base_config,
     ]
 
-    _, cel_expr_format = paths.split_extension(cel_expr)
-
     if cel_expr_format == ".cel" or cel_expr_format == ".celpolicy" or cel_expr_format == ".yaml":
         args.append("--cel_expr=%s" % test_data_path + "/" + cel_expr)
     elif is_raw_expr:
+        data = data + [cel_expr]
         args.append("--cel_expr=%s" % cel_expr)
     else:
         args.append("--cel_expr=$(location {})".format(cel_expr))
 
     if file_descriptor_set != "":
+        data = data + [file_descriptor_set]
         args.append("--file_descriptor_set=$(location {})".format(file_descriptor_set))
 
     go_test(