Tweak extension code when it is rancher head

Signed-off-by: Julien Adamek <julien.adamek@suse.com>

index.ts

@@ -175,19 +175,24 @@ export function deleteUser(username) {
  * Enable the extension support
  * @remarks : Disable the Rancher Repo if you provide your own repo
  * @param withRancherRepo : Add the Rancher Extension Repository - boolean
+ * @param isRancherHead : Tell if the test run on rancher head - boolean
  */
-export function enableExtensionSupport(withRancherRepo) {
+export function enableExtensionSupport(withRancherRepo, isRancherHead) {
   cy.contains('Extensions')
     .click();
   // Make sure we are on the Extensions page
   cy.contains('.message-icon', 'Extension support is not enabled');
   cy.clickButton('Enable');
   cy.contains('Enable Extension Support?')
   if (!withRancherRepo) {
-    cy.contains('Add Official Rancher Extensions Repository')
-      .click();
-    cy.contains('Add Partners Extensions Repository')
-      .click();
+    if (isRancherHead) {
+      cy.contains('Add Partners Extensions Repository')
+        .click();
+    } else {
+      cy.contains('Add Official Rancher Extensions Repository')
+        .click();
+      cy.contains('Add Partners Extensions Repository')
+        .click();
   }
   cy.clickButton('OK');
   cy.get('.tabs', {timeout: 40000})