def test_add_command(self, test_com, expected_id):
act = Action()
act.addCommand(test_com)
assert act.commands == [test_com]
assert act.ids == [expected_id]
def local_action_space(self) -> Tuple[Action]:
wait_action = (Action(0, self.index, self.all_phases[0]), )
light_Actions = ()
for phase in self.all_phases[1:]:
light_Actions += (Action(phase.index, self.index,
self.all_phases[phase.index]), )
if self.phase_duration >= self.orange_phase_duration:
return light_Actions
return wait_action
def test_reset(self):
act = Action()
act.addCommand("Command_1")
act.addCommand("Command_2")
assert act.next() == "Command_1"
assert act.next() == "Command_2"
act.reset()
assert act.next() == "Command_1"
def __init__(self, device_id):
self.action = Action(device_id)
self.device_id = device_id
class Flow(object):
def __init__(self, device_id):
self.action = Action(device_id)
self.device_id = device_id
def test(self):
print(self.device_id, '开始测试流程')
self.action.screen_shot()
status_dict = self.action.get_home_data()
print(status_dict)
# flag = self.action.get_home_data()
# print(flag)
# rtn = get_int('1 1')
# print(rtn)
# print(home_data)
def skim_video(self):
print(self.device_id, '浏览视频')
start_time = int(time.time())
repeat_count = 0
while True:
# 运行3个小时，休息9个小时
cur_time = int(time.time())
if cur_time - start_time >= 2 * 3600:
self.action.to_home()
print(self.device_id, 'waiting......')
time.sleep(4 * 3600)
start_time = int(time.time())
self.action.open_douyin()
time.sleep(30)
self.action.next_video()
time.sleep(5)
self.action.screen_shot()
# 检查是否位于视频界面
flag = self.action.check_home()
if flag == False:
self.action.screen_shot()
# 5次以上重复，重启抖音
if repeat_count >= 5:
print(self.device_id, '超过5次已关注或已点赞，重启App！', status_dict)
self.action.restart_douyin()
repeat_count = 0
continue
# 判断是否已关注
status_dict = self.action.get_status()
if status_dict['followed'] or status_dict['liked']:
repeat_count += 1
print(self.device_id, '已关注或者已点赞', status_dict)
continue
else:
repeat_count = 0
# 获取首页数据
home_data = self.action.get_home_data()
if not home_data:
# print(self.device_id, '获取页面数据失败')
continue
else:
print(
self.device_id,
'点赞数：{}，评论数：{}'.format(home_data['like_num'],
home_data['reply_num']))
# 给少于1万点赞数的人关注和点赞
if home_data['like_num'] < 10000:
# 点赞
if not status_dict['liked']:
time.sleep(10)
self.action.like()
# 关注
if not status_dict['followed']:
time.sleep(10)
self.action.follow()
time.sleep(2)
# 评论
if home_data.__contains__('desc'):
question = home_data['desc']
print(self.device_id, '描述:', question)
answer = self.action.get_answer(question)
if answer:
time.sleep(10)
print(self.device_id, '评论:', answer)
self.action.reply(answer)
time.sleep(3)
else:
print(self.device_id, '获取评论失败')
else:
print(self.device_id, '点赞数超过一万:', home_data['like_num'])
class Flow(object):
def __init__(self, device_id):
self.action = Action(device_id)
self.device_id = device_id
def test(self):
print(self.device_id, '开始测试流程')
self.action.screen_shot()
status_dict = self.action.get_home_data()
print(status_dict)
# flag = self.action.get_home_data()
# print(flag)
# rtn = get_int('1 1')
# print(rtn)
# print(home_data)
def skim_video(self):
print(self.device_id, '浏览视频')
start_time = int(time.time())
while True:
# 运行3个小时，休息9个小时
cur_time = int(time.time())
if cur_time - start_time >= 3 * 3600:
self.adb.home()
time.sleep(9 * 3600)
start_time = int(time.time())
self.adb.open_douyin()
time.sleep(30)
self.action.next_video()
time.sleep(5)
self.action.screen_shot()
# 判断是否已关注
status_dict = self.action.get_status()
if status_dict['followed'] or status_dict['liked']:
print(self.device_id, '已关注或者已点赞', status_dict)
continue
# 获取首页数据
home_data = self.action.get_home_data()
if not home_data:
print(self.device_id, '获取页面数据失败')
continue
else:
print(
self.device_id,
'点赞数：{}，评论数：{}'.format(home_data['like_num'],
home_data['reply_num']))
time.sleep(5)
# 给少于1万点赞数的人关注和点赞
if home_data['like_num'] < 10000:
# 点赞
if not status_dict['liked']:
self.action.like()
time.sleep(2)
# 关注
if not status_dict['followed']:
self.action.follow()
time.sleep(2)
# 评论
if home_data.__contains__('desc'):
question = home_data['desc']
print(self.device_id, '描述:', question)
answer = self.action.get_answer(question)
if answer:
print(self.device_id, '评论:', answer)
self.action.reply(answer)
time.sleep(2)
else:
print(self.device_id, '获取评论失败')
else:
print(self.device_id, '点赞数超过一万:', home_data['like_num'])
def test_repr(self):
act = Action("abc")
act.addCommand(FUNCTIONS.select_point("select", [1, 2]))
assert act.__repr__() == "Action<group=abc, ids={}>".format(
[_Functions['select_point']])
def test_equal(self):
act1 = Action()
act2 = Action()
act3 = Action()
act1.addCommand(FUNCTIONS.select_point("select", [1, 2]))
act2.addCommand(FUNCTIONS.Train_SCV_quick("now"))
act3.addCommand(FUNCTIONS.select_point("select", [3, 4]))
assert act1 != act2
assert act1 == act3