En 2010, pressenti pour être candidat au Sénat[4], Rokita se présente finalement à la Chambre des représentants des États-Unis dans le 4e district de l'Indiana pour succéder au républicain Steve Buyer(en)[2]. Il remporte facilement la primaire républicaine[2] puis est élu en rassemblant 68,6 % des voix, devant le démocrate David Sanders (26,3 %) et le libertarien John Duncan (5,2 %)[5]. En 2011, la législature redessine les circonscriptions et met la maison de Rokita à 500 mètres en dehors de son district[4]. Il est réélu avec 62 % des suffrages en 2012 et 66,9 % en 2014[5].
Rokita abandonne son siège de représentant pour se présenter aux élections sénatoriales de 2018 face au démocrate Joe Donnelly. Lors des primaires républicaines, Rokita et son collègue Luke Messer passent plusieurs mois à dépenser des millions de dollars pour se critiquer l'un et l'autre[6]. Leur inimitié remonterait à leurs études au Wabash College, trente ans plus tôt[4]. Un troisième candidat, l'homme d'affaires Mike Braun les critique pour être des politiciens professionnels indifférenciables. Braun remporte finalement la primaire avec 41 % des suffrages tandis que Messer et Rokita finissent autour de 30 %[6].
Rokita souhaite devenir attorney general de l'Indiana lors de l'élection de novembre 2020. Il bat le sortant républicain Curtis Hill(en) lors de la primaire républicaine en juin 2020 (52 % contre 48 %) et le démocrate Jonathan Weinzapfel(en) en novembre (58 % contre 42 %)[7],[8],[9].