1#!/usr/bin/env python3
2#
3#  Copyright (c) 2018, The OpenThread Authors.
4#  All rights reserved.
5#
6#  Redistribution and use in source and binary forms, with or without
7#  modification, are permitted provided that the following conditions are met:
8#  1. Redistributions of source code must retain the above copyright
9#     notice, this list of conditions and the following disclaimer.
10#  2. Redistributions in binary form must reproduce the above copyright
11#     notice, this list of conditions and the following disclaimer in the
12#     documentation and/or other materials provided with the distribution.
13#  3. Neither the name of the copyright holder nor the
14#     names of its contributors may be used to endorse or promote products
15#     derived from this software without specific prior written permission.
16#
17#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18#  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19#  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20#  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
21#  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22#  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23#  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24#  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25#  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26#  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27#  POSSIBILITY OF SUCH DAMAGE.
28
29import wpan
30from wpan import verify
31
32# -----------------------------------------------------------------------------------------------------------------------
33# Test description: Child-table and child recovery
34#
35
36test_name = __file__[:-3] if __file__.endswith('.py') else __file__
37print('-' * 120)
38print('Starting \'{}\''.format(test_name))
39
40# -----------------------------------------------------------------------------------------------------------------------
41# Creating `wpan.Nodes` instances
42
43speedup = 4
44wpan.Node.set_time_speedup_factor(speedup)
45
46NUM_CHILDREN = 7
47
48router = wpan.Node()
49
50children = []
51for i in range(NUM_CHILDREN):
52    children.append(wpan.Node())
53
54all_nodes = [router] + children
55
56# -----------------------------------------------------------------------------------------------------------------------
57# Init all nodes
58
59wpan.Node.init_all_nodes()
60
61# -----------------------------------------------------------------------------------------------------------------------
62# Build network topology
63
64router.form('child-table')
65for child in children:
66    child.join_node(router, node_type=wpan.JOIN_TYPE_SLEEPY_END_DEVICE)
67    child.set(wpan.WPAN_POLL_INTERVAL, '1000')
68
69# -----------------------------------------------------------------------------------------------------------------------
70# Test implementation
71
72# Get the child table and verify all children are in the table.
73
74child_table = wpan.parse_child_table_result(router.get(wpan.WPAN_THREAD_CHILD_TABLE))
75
76verify(len(child_table) == len(children))
77
78for child in children:
79    ext_addr = child.get(wpan.WPAN_EXT_ADDRESS)[1:-1]
80    for entry in child_table:
81        if entry.ext_address == ext_addr:
82            break
83    else:
84        print('Failed to find a child entry for extended address {} in table'.format(ext_addr))
85        exit(1)
86
87    verify(int(entry.rloc16, 16) == int(child.get(wpan.WPAN_THREAD_RLOC16), 16))
88    verify(int(entry.timeout, 0) == 120)
89    verify(entry.is_rx_on_when_idle() is False)
90    verify(entry.is_ftd() is False)
91
92# -----------------------------------------------------------------------------------------------------------------------
93# Test finished
94
95wpan.Node.finalize_all_nodes()
96
97print('\'{}\' passed.'.format(test_name))
98