enhance dos_test.go