def process_block(self, block):
if block.type == block_pb2.BODY:
return conformance_validators.validate_node(
self.node, block.rows, 0, block.number, block.file_number)
else:
# For now we do not validate the header
return 0
def process_block(self, block):
if block.type == block_pb2.BODY:
return conformance_validators.validate_node(
self.node, block.rows, 0, block.number, block.file_number)
else:
# For now we do not validate the header
return 0
def test_validate_node_valid(self):
one_valid_row = [RowTest('c', 4)]
two_valid_rows = [RowTest('a', 1), RowTest('b', 2)]
three_valid_rows = [RowTest('a', 1), RowTest('b', 2), RowTest('c', 3)]
self.assertEquals(
conformance_validators.validate_node(
self.root, one_valid_row, index=0, block_number=4, file_number=3),
1)
self.assertEquals(
conformance_validators.validate_node(
self.root, two_valid_rows, 0, 3, 2),
2)
self.assertEquals(
conformance_validators.validate_node(
self.root, three_valid_rows, index=0, block_number=5,
file_number=1), 3)
def test_validate_node_invalid_row(self):
one_invalid_rows = [RowTest('a', 5)]
expected_error = textwrap.dedent(r"""
Block 4 starting on row 5 in file number 8 is non-conformant\.
First invalid row: 1 \(row 6 in input file\).
Expected structure:
\[Sequence \(a and b\) or c\]\+
Actual structure:
\['a'\]
""").lstrip()
with six.assertRaisesRegex(self, error.BlockConformanceFailure,
expected_error):
conformance_validators.validate_node(self.root, one_invalid_rows,
0, 4, 8)
def test_validate_node_valid(self):
one_valid_row = [RowTest('c', 4)]
two_valid_rows = [RowTest('a', 1), RowTest('b', 2)]
three_valid_rows = [RowTest('a', 1), RowTest('b', 2), RowTest('c', 3)]
self.assertEqual(
conformance_validators.validate_node(self.root,
one_valid_row,
index=0,
block_number=4,
file_number=3), 1)
self.assertEqual(
conformance_validators.validate_node(self.root, two_valid_rows, 0,
3, 2), 2)
self.assertEqual(
conformance_validators.validate_node(self.root,
three_valid_rows,
index=0,
block_number=5,
file_number=1), 3)