# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import logging

from devil.android.sdk import adb_wrapper

from pylib.local.device import local_device_environment


class LocalDeviceNetworkEnvironment(
    local_device_environment.LocalDeviceEnvironment):
  """A subclass of LocalDeviceEnvironment for devices connected over TCP/IP."""

  def __init__(self, args, output_manager, error_func):
    super().__init__(args, output_manager, error_func)
    logging.info('connecting to %s', args.test_devices[0])
    adb_wrapper.AdbWrapper.Connect(args.test_devices[0])