def test_wait_for_data_when_timeout_occurred(self, select_function_mock):
select_function_mock.return_value = [False]
wait_time = 3
status = Network.wait_for_data(self.socket_mock, wait_time)
select_function_mock.assert_called_once_with([self.socket_mock], [], [], wait_time)
assert status == Network.WaitStatus.TIMEOUT
def test_wait_for_data_when_new_data_is_available(self, select_function_mock):
select_function_mock.return_value = [True]
wait_time = 3
status = Network.wait_for_data(self.socket_mock, wait_time)
select_function_mock.assert_called_once_with([self.socket_mock], [], [], wait_time)
assert status == Network.WaitStatus.NEW_DATA
