Первый ваш пример si on prenait des gâteaux au chocolat ? это случай употребления imparfait как предложение сделать что-то в вопросительном предложении после si. А не взять ли нам шоколадный торт?
Второй пример это не imparfait это futur immédiat dans le passé ( эта глагольная форма образуется так: imparfait глагола aller + инфинитив) здесь описывается действие которое должно совершиться непосредственно после другого действия в прошлом. Bonne chance!