Skip to content

Commit

Permalink
Add to flake8 in workflow and fix python files (#25251)
Browse files Browse the repository at this point in the history
  • Loading branch information
DamMicSzm committed Mar 6, 2023
1 parent 1d8d73f commit d5ebdab
Show file tree
Hide file tree
Showing 8 changed files with 53 additions and 62 deletions.
1 change: 0 additions & 1 deletion .flake8
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@ exclude = third_party
# temporarily scan only directories with fixed files
# TODO: Remove the paths below when all bugs are fixed
src/tools/chip-cert/*
src/test_driver/openiotsdk/*
src/test_driver/mbed/*
src/test_driver/linux-cirque/*
build/chip/java/tests/*
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,6 @@ def wait_for_output(self, search: str, timeout: float = 10, assert_timeout: bool
if line:
lines.append(line)
if search in line:
end = time()
return lines

except queue.Empty:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,11 @@
import os
import pathlib
import shutil
from time import sleep

import chip.CertificateAuthority
import chip.native
import pytest
from chip import ChipDeviceCtrl, exceptions
from chip.ChipStack import *
from chip import exceptions

from .fvp_device import FvpDevice
from .telnet_connection import TelnetConnection
Expand Down Expand Up @@ -113,7 +111,7 @@ def controller(vendor_id, fabric_id, node_id):
except exceptions.ChipStackException as ex:
log.error("Controller initialization failed {}".format(ex))
return None
except:
except Exception:
log.error("Controller initialization failed")
return None

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
#

import logging
import os
import subprocess
import threading
from time import sleep
Expand Down Expand Up @@ -45,7 +44,7 @@ def __init__(self, fvp, fvp_config, binary_file, connection_channel, network_int
'-C', 'mps3_board.telnetterminal0.start_port={}'.format(self.connection_channel.get_port())
]

if network_interface != None:
if network_interface is not None:
self.fvp_cmd.extend(['-C', 'mps3_board.hostbridge.interfaceName={}'.format(network_interface), ])
else:
self.fvp_cmd.extend(['-C', 'mps3_board.hostbridge.userNetworking=1'])
Expand Down
11 changes: 4 additions & 7 deletions src/test_driver/openiotsdk/integration-tests/common/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,10 @@
#

import asyncio
import ctypes
import logging
import os
import random
import re
import shlex
from time import sleep

from chip import discovery, exceptions
from chip.clusters import Objects as GeneratedObjects
Expand All @@ -40,7 +37,7 @@ def get_setup_payload(device):
:return: setup payload or None
"""
ret = device.wait_for_output("SetupQRCode")
if ret == None or len(ret) < 2:
if ret is None or len(ret) < 2:
return None

qr_code = re.sub(
Expand Down Expand Up @@ -85,7 +82,7 @@ def connect_device(setupPayload, commissionableDevice, nodeId=None):
:param nodeId: device node ID
:return: node ID if connection successful or None if failed
"""
if nodeId == None:
if nodeId is None:
nodeId = random.randint(1, 1000000)

pincode = int(setupPayload.attributes['SetUpPINCode'])
Expand Down Expand Up @@ -191,7 +188,7 @@ def send_zcl_command(devCtrl, line, requestTimeoutMs: int = None):
raise exceptions.UnknownCluster(cluster)
cmdArgsWithType = allCommands.get(cluster).get(command, None)
# When command takes no arguments, (not command) is True
if command == None:
if command is None:
raise exceptions.UnknownCommand(cluster, command)

args = FormatZCLArguments(cluster, cmdArgsLine, cmdArgsWithType)
Expand Down Expand Up @@ -232,7 +229,7 @@ def read_zcl_attribute(devCtrl, line):
raise exceptions.UnknownCluster(cluster)

attrDetails = allAttrs.get(cluster).get(attribute, None)
if attrDetails == None:
if attrDetails is None:
raise exceptions.UnknownAttribute(cluster, attribute)

res = devCtrl.ZCLReadAttribute(cluster, attribute, int(
Expand Down
2 changes: 0 additions & 2 deletions src/test_driver/openiotsdk/integration-tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,6 @@
# limitations under the License.
#

import pytest

pytest_plugins = ['common.fixtures']


Expand Down
45 changes: 24 additions & 21 deletions src/test_driver/openiotsdk/integration-tests/lock-app/test_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,11 @@
#

import logging
from time import sleep
import os

import pytest
from chip.clusters.Objects import DoorLock
from common.utils import *
from common.utils import connect_device, disconnect_device, discover_device, get_setup_payload, read_zcl_attribute, send_zcl_command

log = logging.getLogger(__name__)

Expand All @@ -36,32 +36,32 @@ def binaryPath(request, rootDir):
@pytest.mark.smoketest
def test_smoke_test(device):
ret = device.wait_for_output("Open IoT SDK lock-app example application start")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0
ret = device.wait_for_output("Open IoT SDK lock-app example application run")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0


@pytest.mark.commissioningtest
def test_commissioning(device, controller):
assert controller != None
assert controller is not None
devCtrl = controller

setupPayload = get_setup_payload(device)
assert setupPayload != None
assert setupPayload is not None

commissionable_device = discover_device(devCtrl, setupPayload)
assert commissionable_device != None
assert commissionable_device is not None

assert commissionable_device.vendorId == int(setupPayload.attributes['VendorID'])
assert commissionable_device.productId == int(setupPayload.attributes['ProductID'])
assert commissionable_device.addresses[0] != None
assert commissionable_device.addresses[0] is not None

nodeId = connect_device(setupPayload, commissionable_device)
assert nodeId != None
assert nodeId is not None
log.info("Device {} connected".format(commissionable_device.addresses[0]))

ret = device.wait_for_output("Commissioning completed successfully", timeout=30)
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

assert disconnect_device(devCtrl, nodeId)

Expand All @@ -75,20 +75,20 @@ def test_commissioning(device, controller):

@pytest.mark.ctrltest
def test_lock_ctrl(device, controller):
assert controller != None
assert controller is not None
devCtrl = controller

setupPayload = get_setup_payload(device)
assert setupPayload != None
assert setupPayload is not None

commissionable_device = discover_device(devCtrl, setupPayload)
assert commissionable_device != None
assert commissionable_device is not None

nodeId = connect_device(setupPayload, commissionable_device)
assert nodeId != None
assert nodeId is not None

ret = device.wait_for_output("Commissioning completed successfully", timeout=30)
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

err, res = send_zcl_command(
devCtrl, "DoorLock SetUser {} {} operationType={} userIndex={} userName={} userUniqueId={} "
Expand All @@ -105,7 +105,7 @@ def test_lock_ctrl(device, controller):
ret = device.wait_for_output("Successfully set the user [mEndpointId={},index={},adjustedIndex=0]".format(
LOCK_CTRL_TEST_ENDPOINT_ID,
LOCK_CTRL_TEST_USER_INDEX))
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

err, res = send_zcl_command(
devCtrl, "DoorLock GetUser {} {} userIndex={}".format(nodeId, LOCK_CTRL_TEST_ENDPOINT_ID,
Expand Down Expand Up @@ -133,9 +133,12 @@ def test_lock_ctrl(device, controller):
assert res.status == DoorLock.Enums.DlStatus.kSuccess

ret = device.wait_for_output("Successfully set the credential [mEndpointId={},index={},"
"credentialType={},creator=1,modifier=1]".format(LOCK_CTRL_TEST_ENDPOINT_ID,
LOCK_CTRL_TEST_USER_INDEX, DoorLock.Enums.DlCredentialType.kPin))
assert ret != None and len(ret) > 0
"credentialType={},creator=1,modifier=1]".format(
LOCK_CTRL_TEST_ENDPOINT_ID,
LOCK_CTRL_TEST_USER_INDEX,
DoorLock.Enums.DlCredentialType.kPin
))
assert ret is not None and len(ret) > 0

err, res = send_zcl_command(
devCtrl, "DoorLock GetCredentialStatus {} {} credential=struct:DlCredential(credentialType={},"
Expand All @@ -152,7 +155,7 @@ def test_lock_ctrl(device, controller):
assert err == 0

ret = device.wait_for_output("Setting door lock state to \"Locked\" [endpointId={}]".format(LOCK_CTRL_TEST_ENDPOINT_ID))
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

err, res = read_zcl_attribute(
devCtrl, "DoorLock LockState {} {}".format(nodeId, LOCK_CTRL_TEST_ENDPOINT_ID))
Expand All @@ -165,7 +168,7 @@ def test_lock_ctrl(device, controller):
assert err == 0

ret = device.wait_for_output("Setting door lock state to \"Unlocked\" [endpointId={}]".format(LOCK_CTRL_TEST_ENDPOINT_ID))
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

err, res = read_zcl_attribute(
devCtrl, "DoorLock LockState {} {}".format(nodeId, LOCK_CTRL_TEST_ENDPOINT_ID))
Expand Down
46 changes: 22 additions & 24 deletions src/test_driver/openiotsdk/integration-tests/shell/test_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,12 @@
# limitations under the License.

import logging
import re
from time import sleep
import os

import chip.native
import pytest
from chip import exceptions
from chip.setup_payload import SetupPayload
from common.utils import *
from packaging import version

log = logging.getLogger(__name__)
Expand Down Expand Up @@ -72,9 +70,9 @@ def parse_boarding_codes_response(response):
@pytest.mark.smoketest
def test_smoke_test(device):
ret = device.wait_for_output("Open IoT SDK shell example application start")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0
ret = device.wait_for_output("Open IoT SDK shell example application run")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0


@pytest.mark.ctrltest
Expand All @@ -84,64 +82,64 @@ def test_command_check(device):
except exceptions.ChipStackException as ex:
log.error("CHIP initialization failed {}".format(ex))
assert False
except:
except Exception:
log.error("CHIP initialization failed")
assert False

ret = device.wait_for_output("Open IoT SDK shell example application start")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0
ret = device.wait_for_output("Open IoT SDK shell example application run")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

# Help
ret = device.send(command="help", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
shell_commands = get_shell_command(ret[1:-1])
assert set(SHELL_COMMAND_NAME) == set(shell_commands)

# Echo
ret = device.send(command="echo Hello", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert "Hello" in ret[-2]

# Log
ret = device.send(command="log Hello", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert "[INF] [TOO] Hello" in ret[-2]

# Rand
ret = device.send(command="rand", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert ret[-2].rstrip().isdigit()

# Base64
hex_string = "1234"
ret = device.send(command="base64 encode {}".format(
hex_string), expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
base64code = ret[-2]
ret = device.send(command="base64 decode {}".format(
base64code), expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert ret[-2].rstrip() == hex_string

# Version
ret = device.send(command="version", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert "CHIP" in ret[-2].split()[0]
app_version = ret[-2].split()[1]
assert isinstance(version.parse(app_version), version.Version)

# Config
ret = device.send(command="config", expected_output="Done")
assert ret != None and len(ret) > 2
assert ret is not None and len(ret) > 2

config = parse_config_response(ret[1:-1])

for param_name, value in config.items():
ret = device.send(command="config {}".format(
param_name), expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
if "discriminator" in param_name:
assert int(ret[-2].split()[0], 16) == value
else:
Expand All @@ -150,23 +148,23 @@ def test_command_check(device):
new_value = int(config['discriminator']) + 1
ret = device.send(command="config discriminator {}".format(
new_value), expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert "Setup discriminator set to: {}".format(new_value) in ret[-2]

ret = device.send(command="config discriminator", expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert int(ret[-2].split()[0], 16) == new_value

# Onboardingcodes
ret = device.send(command="onboardingcodes none", expected_output="Done")
assert ret != None and len(ret) > 2
assert ret is not None and len(ret) > 2

boarding_codes = parse_boarding_codes_response(ret[1:-1])

for param, value in boarding_codes.items():
ret = device.send(command="onboardingcodes none {}".format(
param), expected_output="Done")
assert ret != None and len(ret) > 1
assert ret is not None and len(ret) > 1
assert value == ret[-2].strip()

try:
Expand All @@ -175,16 +173,16 @@ def test_command_check(device):
except exceptions.ChipStackError as ex:
log.error(ex.msg)
assert False
assert device_details != None and len(device_details) != 0
assert device_details is not None and len(device_details) != 0

try:
device_details = dict(SetupPayload().ParseManualPairingCode(
boarding_codes['manualpairingcode']).attributes)
except exceptions.ChipStackError as ex:
log.error(ex.msg)
assert False
assert device_details != None and len(device_details) != 0
assert device_details is not None and len(device_details) != 0

# Exit - should be the last check
ret = device.send(command="exit", expected_output="Goodbye")
assert ret != None and len(ret) > 0
assert ret is not None and len(ret) > 0

0 comments on commit d5ebdab

Please sign in to comment.