Fix the ee-files-location-check job

Signed-off-by: Rémy Coutable <remy@rymai.me>

scripts/ee-files-location-check

@@ -13,11 +13,21 @@ WHITELIST = %w[
 `git remote add canonical-ee https://gitlab.com/gitlab-org/gitlab-ee.git`
 `git remote add canonical-ce https://gitlab.com/gitlab-org/gitlab-ce.git`
 `git fetch canonical-ee master --quiet`
-`git fetch canonical-ce master --quiet`
+
 new_files_in_this_branch_not_at_the_ee_top_level =
   `git diff canonical-ee/master...HEAD --name-status --diff-filter=A -- ./ ':!ee' | cut -f2`.lines.map(&:strip)
+
+current_branch = ENV.fetch('CI_COMMIT_REF_NAME', `git rev-parse --abbrev-ref HEAD`).strip
+ce_branch_name = current_branch.sub(/(\Aee\-|\-ee\z)/, '')
+
+`git fetch canonical-ce #{ce_branch_name} --quiet`
+
+unless $?.success?
+  `git fetch canonical-ce 'master' --quiet`
+end
+
 ee_specific_files_in_ce_master_not_at_the_ee_top_level =
-  `git diff canonical-ce/master...HEAD --name-status --diff-filter=A -- ./ ':!ee' | cut -f2`.lines.map(&:strip)
+  `git diff FETCH_HEAD..HEAD --name-status --diff-filter=A -- ./ ':!ee' | cut -f2`.lines.map(&:strip)
 new_ee_specific_files_not_at_the_ee_top_level =
   new_files_in_this_branch_not_at_the_ee_top_level & ee_specific_files_in_ce_master_not_at_the_ee_top_level