Самые ранние наброски к пьесе, которая сегодня известна под названием «Батум», на столе Булгакова появились 16 января 1939 года. Первая редакция пьесы называлась «Пастырь». В ее основу была положена история Батумской рабочей демонстрации 8–9 марта 1902 года, организованной Сталиным. Главным источником послужила книга «Батумская демонстрация 1902 года», выпущенная Партиздатом в марте 1937 года и содержавшая документы и воспоминания, призванные возвеличить первые шаги вождя по руководству революционным движением в Закавказье. В качестве вариантов заглавия, кроме «Пастырь», Булгаков рассматривал «Бессмертие», «Битва», «Рождение славы», «Аргонавты», «Геракл», «Кормчий», «Юность штурмана», «Так было», «Кондор», «Комета зажглась», «Штурман вел корабль», «Молния», «Вставший из снега», «Штурман вел по звездам», «Юность командора», «Юный штурман», «Юность рулевого», «Поход аргонавтов», «Штурман шел по звездам», «Море штормит», «Когда начинался шторм», «Шторм грохотал», «Будет буря», «Мастер», «Штурман вел аргонавтов», «Комета пришла», «Как начиналась слава», «У огня», «Дело было в Батуми» и, наконец, «Батум». Поскольку действие пьесы разворачивается в черноморском порту и на месте древнегреческой колонии, Булгаков пробовал названия, связанные с морем или героями древнегреческих мифов — аргонавтами, Гераклом. В ряде названий, начиная с «Пастыря», фигурирует образ молодого Сталина, выступающего в пьесе в качестве положительного культурного героя мифа (иная трактовка по условиям цензуры была в принципе невозможна).
15 июня Булгаков подписал договор с МХАТом. Первая редакция пьесы была закончена в середине июля, а беловая редакция — к 27 июля, когда Булгаков прочел «Батум» партийной группе МХАТа. Это чтение описано Е. С. Булгаковой: «Слушали замечательно, после чтения очень долго стоя аплодировали. Потом высказыванья. Все очень хорошо. Калишьян в последней речи сказал, что театр должен ее поставить к 21 декабря». От всех, включая главного режиссера МХАТа Немировича-Данченко, «Батум», несмотря на свои довольно скромные художественные достоинства, имел только благоприятные отзывы (других о произведении, посвященном Сталину, быть не могло). Уже намечались актеры на главные роли. В частности, Сталина должен был играть Н. П. Хмелев.
С «Батумом» Булгаков и Елена Сергеевна связывали большие надежды. Они надеялись, что постановка пьесы о Сталине позволит наконец Михаилу Афанасьевичу пробиться и на сцену, и в печать, а главное — опубликовать «Мастера и Маргариту». 11 июня 1939 года Елена Сергеевна записала в дневнике: «Пришла домой. Борис Эрдман сидит с Мишей, а потом подошел и Николай Робертович. Миша прочитал им три картины и рассказал всю пьесу („Батум“. —
Окрыленная Елена Сергеевна 11 августа писала матери:
«Мамочка, дорогая, давно уже собиралась написать тебе, но занята была безумно. Миша закончил и сдал МХАТу пьесу… Устал он дьявольски, работа была напряженная, надо было сдать ее к сроку. Но усталость хорошая — работа была страшно интересная. По общим отзывам, это большая удача. Было несколько чтений — два официальных и другие — у нас на квартире, и всегда большой успех».
В тот же день она записала в дневнике:
«Вечером звонок — завлит Воронежского театра — просит пьесу — „ее безумно расхваливал Афиногенов“. Сегодня встретила одного знакомого, то же самое — „слышал, что М. А. написал изумительную пьесу“. Слышал не в Москве, а где-то на юге. Забавный случай: Бюро заказов Елисеева. То же сообщение — Фани Николаевна. — А кто вам сказал? — Яков Данилыч (Розенталь. —
14 августа драматург с женой во главе бригады мхатовцев выехал в Грузию для сбора в Батуме и Тбилиси материалов (грузинский фольклор, пейзажные зарисовки для декораций и т. д.) для постановки пьесы. Через два часа после отбытия из Москвы в Серпухове на имя Булгакова принесли телеграмму директора театра Г. М. Калишьяна: «Надобность поездки отпала возвращайтесь в Москву». Булгаковы решили было ехать дальше — просто отдохнуть в Батуме, но потом поняли, что отдыха все равно не будет, сошли в Туле и на попутной машине вернулись в Москву. Тогда писатель впервые почувствовал резь в глазах — грозный признак начавшегося нефросклероза.
Елена Сергеевна так описала в дневнике возвращение в Москву: «Миша одной рукой закрывал глаза от солнца, а другой держался за меня и говорил: навстречу чему мы мчимся? может быть — смерти?