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_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
def test_slave_state_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 = ('ACK', [slave_state.slave_id, const.PYPI_XMLRPC])
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, 'ACK',
[slave_state.slave_id, 'https://pypi.org/pypi']
])