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