Bartosz Kapustka zadebiutował w pierwszej reprezentacji Polski 7 września 2015 roku w zwycięskim meczu z Gibraltarem (8:1) w którym strzelił bramkę. W listopadzie ubiegłego roku wystąpił w kolejnych dwóch wygranych meczach: z Islandią (4:2) w którym ponownie strzelił gola i zaliczył asystę, oraz z Czechami (3:1) w którym asystował przy bramce. W marcu tego roku zagrał w dwóch wygranych meczach towarzyskich z Serbią (1:0) i Finlandią (5:0). Łącznie Bartosz Kapustka w 5 meczach pierwszej reprezentacji strzelił 2 bramki i miał dwie asysty.

Wcześniej Bartosz Kapustka grał w młodzieżowych reprezentacjach Polski: U-20 (1 mecz), U-19 (7 meczów, 1 bramka), U-18 (5 meczy), U-17 (1 mecz).