Add test for granting focus on flex split launch

* Inject IActivityTaskManager instead of getting it from
binder

Bug: 396026394
Flag: com.android.wm.shell.enable_flexible_two_app_split
Change-Id: I2bc438af7fb150aee199429288801e61d4498c61