test: Add back state recovery test and refactor #5826

Job	Run time
test / Run E2E System Tests	1m 13s
test / Go Mod Tidy Check	20s
branch_name	2s
test / Build All Rollkit Binaries	2m 2s
test / Run Unit Tests	3m 11s
lint / golangci-lint	1m 8s
proto / buf-check	7s
lint / yamllint	7s
lint / goreleaser-check	10s
lint / markdown-lint	1m 9s
lint / hadolint / hadolint	7s
release	-1s
	9m 35s

Provide feedback