Add test for setting primary selection