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_recv_request(build_state, file_state): 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.request = ('IDLE', None) assert slave_state.request == ('IDLE', None) assert slave_state.last_seen == now assert slave_state.build is None 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()] ]) assert slave_state.last_seen == now build_state._slave_id = slave_state.slave_id assert slave_state.build == build_state