Add dependencies (#56666)

scripts/drone/pipelines/test_backend.star

@@ -23,7 +23,7 @@ def test_backend(trigger, ver_mode, edition="oss"):
     init_steps.extend([
         identify_runner_step(),
         compile_build_cmd(edition),
-        verify_gen_cue_step(edition="oss"),
+        verify_gen_cue_step(edition),
         wire_install_step(),
     ])
     test_steps = [

scripts/drone/pipelines/test_frontend.star

@@ -1,6 +1,8 @@
 load(
     'scripts/drone/steps/lib.star',
     'identify_runner_step',
+    'clone_enterprise_step',
+    'init_enterprise_step',
     'download_grabpl_step',
     'yarn_install_step',
     'betterer_frontend_step',
@@ -14,13 +16,16 @@ load(
 
 def test_frontend(trigger, ver_mode, edition="oss"):
     environment = {'EDITION': edition}
-    init_steps = [
+    init_steps = []
+    if edition != 'oss':
+        init_steps.extend([clone_enterprise_step(ver_mode), init_enterprise_step(ver_mode),])
+    init_steps.extend([
         identify_runner_step(),
         download_grabpl_step(),
         yarn_install_step(),
-    ]
+    ])
     test_steps = [
-        betterer_frontend_step(),
+        betterer_frontend_step(edition),
         test_frontend_step(),
     ]
     pipeline_name = '{}-test-frontend'.format(ver_mode)

scripts/drone/steps/lib.star

@@ -514,13 +514,15 @@ def test_backend_integration_step(edition):
         ],
     }
 
-def betterer_frontend_step():
+def betterer_frontend_step(edition="oss"):
+    deps = []
+    if edition == "enterprise":
+        deps.extend(['init-enterprise'])
+    deps.extend(['yarn-install'])
     return {
         'name': 'betterer-frontend',
         'image': build_image,
-        'depends_on': [
-            'yarn-install',
-        ],
+        'depends_on': deps,
         'commands': [
             'yarn betterer ci',
         ],
@@ -529,16 +531,18 @@ def betterer_frontend_step():
 
 
 
-def test_frontend_step():
+def test_frontend_step(edition="oss"):
+    deps = []
+    if edition == "enterprise":
+        deps.extend(['init-enterprise'])
+    deps.extend(['yarn-install'])
     return {
         'name': 'test-frontend',
         'image': build_image,
         'environment': {
             'TEST_MAX_WORKERS': '50%',
         },
-        'depends_on': [
-            'yarn-install',
-        ],
+        'depends_on': deps,
         'commands': [
             'yarn run ci:test-frontend',
         ],
@@ -1253,7 +1257,7 @@ def end_to_end_tests_deps(edition):
 
 def compile_build_cmd(edition='oss'):
     dependencies = []
-    if edition == 'enterprise':
+    if edition in ('enterprise', 'enterprise2'):
           dependencies = ['init-enterprise',]
     return {
         'name': 'compile-build-cmd',