test/py/tests/test_bind.py

   1 # SPDX-License-Identifier: GPL-2.0
   2 # Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
   3
   4 import os.path
   5 import pytest
   6 import re
   7
   8 def in_tree(response, name, uclass, drv, depth, last_child):
   9         lines = [x.strip() for x in response.splitlines()]
  10         leaf = ' ' * 4 * depth;
  11         if not last_child:
  12                 leaf = leaf + '\|'
  13         else:
  14                 leaf = leaf + '`'
  15         leaf = leaf + '-- ' + name
  16         line = ' *{:10.10}  [0-9]*  \[ [ +] \]   {:20.20}  {}$'.format(uclass, drv, leaf)
  17         prog = re.compile(line)
  18         for l in lines:
  19                 if prog.match(l):
  20                         return True
  21         return False
  22
  23
  24 @pytest.mark.buildconfigspec('cmd_bind')
  25 def test_bind_unbind_with_node(u_boot_console):
  26
  27         #bind /bind-test. Device should come up as well as its children
  28         response = u_boot_console.run_command('bind  /bind-test generic_simple_bus')
  29         assert response == ''
  30         tree = u_boot_console.run_command('dm tree')
  31         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  32         assert in_tree(tree, 'bind-test-child1', 'phy', 'phy_sandbox', 1, False)
  33         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
  34
  35         #Unbind child #1. No error expected and all devices should be there except for bind-test-child1
  36         response = u_boot_console.run_command('unbind  /bind-test/bind-test-child1')
  37         assert response == ''
  38         tree = u_boot_console.run_command('dm tree')
  39         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  40         assert 'bind-test-child1' not in tree
  41         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
  42
  43         #bind child #1. No error expected and all devices should be there
  44         response = u_boot_console.run_command('bind  /bind-test/bind-test-child1 phy_sandbox')
  45         assert response == ''
  46         tree = u_boot_console.run_command('dm tree')
  47         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  48         assert in_tree(tree, 'bind-test-child1', 'phy', 'phy_sandbox', 1, True)
  49         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, False)
  50
  51         #Unbind child #2. No error expected and all devices should be there except for bind-test-child2
  52         response = u_boot_console.run_command('unbind  /bind-test/bind-test-child2')
  53         assert response == ''
  54         tree = u_boot_console.run_command('dm tree')
  55         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  56         assert in_tree(tree, 'bind-test-child1', 'phy', 'phy_sandbox', 1, True)
  57         assert 'bind-test-child2' not in tree
  58
  59
  60         #Bind child #2. No error expected and all devices should be there
  61         response = u_boot_console.run_command('bind /bind-test/bind-test-child2 generic_simple_bus')
  62         assert response == ''
  63         tree = u_boot_console.run_command('dm tree')
  64         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  65         assert in_tree(tree, 'bind-test-child1', 'phy', 'phy_sandbox', 1, False)
  66         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
  67
  68         #Unbind parent. No error expected. All devices should be removed and unbound
  69         response = u_boot_console.run_command('unbind  /bind-test')
  70         assert response == ''
  71         tree = u_boot_console.run_command('dm tree')
  72         assert 'bind-test' not in tree
  73         assert 'bind-test-child1' not in tree
  74         assert 'bind-test-child2' not in tree
  75
  76         #try binding invalid node with valid driver
  77         response = u_boot_console.run_command('bind  /not-a-valid-node generic_simple_bus')
  78         assert response != ''
  79         tree = u_boot_console.run_command('dm tree')
  80         assert 'not-a-valid-node' not in tree
  81
  82         #try binding valid node with invalid driver
  83         response = u_boot_console.run_command('bind  /bind-test not_a_driver')
  84         assert response != ''
  85         tree = u_boot_console.run_command('dm tree')
  86         assert 'bind-test' not in tree
  87
  88         #bind /bind-test. Device should come up as well as its children
  89         response = u_boot_console.run_command('bind  /bind-test generic_simple_bus')
  90         assert response == ''
  91         tree = u_boot_console.run_command('dm tree')
  92         assert in_tree(tree, 'bind-test', 'simple_bus', 'generic_simple_bus', 0, True)
  93         assert in_tree(tree, 'bind-test-child1', 'phy', 'phy_sandbox', 1, False)
  94         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
  95
  96         response = u_boot_console.run_command('unbind  /bind-test')
  97         assert response == ''
  98
  99 def get_next_line(tree, name):
 100         treelines = [x.strip() for x in tree.splitlines() if x.strip()]
 101         child_line = ''
 102         for idx, line in enumerate(treelines):
 103                 if ('-- ' + name) in line:
 104                         try:
 105                                 child_line = treelines[idx+1]
 106                         except:
 107                                 pass
 108                         break
 109         return child_line
 110
 111 @pytest.mark.buildconfigspec('cmd_bind')
 112 def test_bind_unbind_with_uclass(u_boot_console):
 113         #bind /bind-test
 114         response = u_boot_console.run_command('bind  /bind-test generic_simple_bus')
 115         assert response == ''
 116
 117         #make sure bind-test-child2 is there and get its uclass/index pair
 118         tree = u_boot_console.run_command('dm tree')
 119         child2_line = [x.strip() for x in tree.splitlines() if '-- bind-test-child2' in x]
 120         assert len(child2_line) == 1
 121
 122         child2_uclass = child2_line[0].split()[0]
 123         child2_index = int(child2_line[0].split()[1])
 124
 125         #bind generic_simple_bus as a child of bind-test-child2
 126         response = u_boot_console.run_command('bind  {} {} generic_simple_bus'.format(child2_uclass, child2_index, 'generic_simple_bus'))
 127
 128         #check that the child is there and its uclass/index pair is right
 129         tree = u_boot_console.run_command('dm tree')
 130
 131         child_of_child2_line = get_next_line(tree, 'bind-test-child2')
 132         assert child_of_child2_line
 133         child_of_child2_index = int(child_of_child2_line.split()[1])
 134         assert in_tree(tree, 'generic_simple_bus', 'simple_bus', 'generic_simple_bus', 2, True)
 135         assert child_of_child2_index == child2_index + 1
 136
 137         #unbind the child and check it has been removed
 138         response = u_boot_console.run_command('unbind  simple_bus {}'.format(child_of_child2_index))
 139         assert response == ''
 140         tree = u_boot_console.run_command('dm tree')
 141         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
 142         assert not in_tree(tree, 'generic_simple_bus', 'simple_bus', 'generic_simple_bus', 2, True)
 143         child_of_child2_line = get_next_line(tree, 'bind-test-child2')
 144         assert child_of_child2_line == ''
 145
 146         #bind generic_simple_bus as a child of bind-test-child2
 147         response = u_boot_console.run_command('bind  {} {} generic_simple_bus'.format(child2_uclass, child2_index, 'generic_simple_bus'))
 148
 149         #check that the child is there and its uclass/index pair is right
 150         tree = u_boot_console.run_command('dm tree')
 151         treelines = [x.strip() for x in tree.splitlines() if x.strip()]
 152
 153         child_of_child2_line = get_next_line(tree, 'bind-test-child2')
 154         assert child_of_child2_line
 155         child_of_child2_index = int(child_of_child2_line.split()[1])
 156         assert in_tree(tree, 'generic_simple_bus', 'simple_bus', 'generic_simple_bus', 2, True)
 157         assert child_of_child2_index == child2_index + 1
 158
 159         #unbind the child and check it has been removed
 160         response = u_boot_console.run_command('unbind  {} {} generic_simple_bus'.format(child2_uclass, child2_index, 'generic_simple_bus'))
 161         assert response == ''
 162
 163         tree = u_boot_console.run_command('dm tree')
 164         assert in_tree(tree, 'bind-test-child2', 'simple_bus', 'generic_simple_bus', 1, True)
 165
 166         child_of_child2_line = get_next_line(tree, 'bind-test-child2')
 167         assert child_of_child2_line == ''
 168
 169         #unbind the child again and check it doesn't change the tree
 170         tree_old = u_boot_console.run_command('dm tree')
 171         response = u_boot_console.run_command('unbind  {} {} generic_simple_bus'.format(child2_uclass, child2_index, 'generic_simple_bus'))
 172         tree_new = u_boot_console.run_command('dm tree')
 173
 174         assert response == ''
 175         assert tree_old == tree_new
 176
 177         response = u_boot_console.run_command('unbind  /bind-test')
 178         assert response == ''