Puustinen edusti Suomea MM-kilpailuissa vuonna 2021, jolloin hän voitti hopeaa. Puustinen pelasi turnauksessa kuitenkin vain kaksi ottelua. Yhteensä hän on pelannut seitsemän A-maaottelua tehopistein 1+1=2. Puustinen teki A-maajoukkuedebyyttinsä Channel One Cupin 12. joulukuuta 2019 Plzeňissä pelatussa avausottelussa Tšekkiä vastaan. Hänet merkittiin Suomen kolmosketjun oikeaan laitaan Otto Paajasen ja Eemeli Suomen viereen. Ensimmäisen A-maajoukkuemaalinsa Puustinen teki seuraavassa ottelussa 14. joulukuuta 2019 Ruotsia vastaan.