28 lines
604 B
Go
28 lines
604 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestMainFunction(t *testing.T) {
|
|
// Сохраняем оригинальную функцию
|
|
originalExecute := execute
|
|
defer func() {
|
|
execute = originalExecute // Восстанавливаем после теста
|
|
}()
|
|
|
|
// Флаг для проверки вызова
|
|
called := false
|
|
execute = func() {
|
|
called = true // Подменяем функцию
|
|
}
|
|
|
|
// Запускаем main
|
|
main()
|
|
|
|
// Проверяем, что функция была вызвана
|
|
if !called {
|
|
t.Fatal("Expected cmd.Execute to be called, but it wasn't")
|
|
}
|
|
}
|