def test_slave_recv_bad_built(build_state, file_state, slave_queue): slave_state = SlaveState('10.0.0.2', timedelta(hours=3), '34', 'cp34m', 'linux_armv7l', 'piwheels2') with mock.patch('piwheels.states.datetime') as dt: now = datetime.now(tz=UTC) dt.now.return_value = now slave_state._reply = ('BUILD', ['foo', '0.1']) slave_state.request = ('BUILT', None) assert slave_state.build is None slave_state.reply = ('DONE', None) assert slave_state.build is None
def test_slave_state_recv_hello(master_status_queue, slave_queue): with mock.patch('piwheels.states.datetime') as dt: now = datetime.now(tz=UTC) dt.now.return_value = now slave_state = SlaveState('10.0.0.2', timedelta(hours=3), '34', 'cp34m', 'linux_armv7l', 'piwheels2') slave_state._reply = ('IDLE', None) slave_state.hello() assert master_status_queue.recv_msg() == ('SLAVE', [ slave_state.slave_id, now, 'HELLO', [timedelta(hours=3), '34', 'cp34m', 'linux_armv7l', 'piwheels2'] ]) assert master_status_queue.recv_msg() == ('SLAVE', [ slave_state.slave_id, now, 'IDLE', None ])
def test_slave_recv_reply(build_state, file_state, slave_queue): slave_state = SlaveState('10.0.0.2', timedelta(hours=3), '34', 'cp34m', 'linux_armv7l', 'piwheels2') with mock.patch('piwheels.states.datetime') as dt: now = datetime.now(tz=UTC) dt.now.return_value = now slave_state._reply = ('BUILD', ['foo', '0.1']) slave_state.request = ('BUILT', [ build_state.status, build_state.duration, build_state.output, [file_state.as_message()] ]) build_state._slave_id = slave_state.slave_id assert slave_state.build == build_state slave_state.reply = ('DONE', None) assert slave_state.build is None