tools/json_schema_compiler/compiler_test.py

#!/usr/bin/env python3
# Copyright 2012 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import compiler
import json_schema
import unittest


class CompilerUnittest(unittest.TestCase):

  def testNocompile(self):
    compiled = [{
        "namespace": "compile",
        "description": "The compile API.",
        "functions": [],
        "types": {}
    }, {
        "namespace": "functions",
        "description": "The functions API.",
        "functions": [{
            "id": "two"
        }, {
            "id": "four"
        }],
        "types": {
            "one": {
                "key": "value"
            }
        }
    }, {
        "namespace": "types",
        "description": "The types API.",
        "functions": [{
            "id": "one"
        }],
        "types": {
            "two": {
                "key": "value"
            },
            "four": {
                "key": "value"
            }
        }
    }, {
        "namespace": "nested",
        "description": "The nested API.",
        "properties": {
            "sync": {
                "functions": [{
                    "id": "two"
                }, {
                    "id": "four"
                }],
                "types": {
                    "two": {
                        "key": "value"
                    },
                    "four": {
                        "key": "value"
                    }
                }
            }
        }
    }]

    schema = json_schema.CachedLoad('test/compiler_test.json')
    self.assertEqual(compiled, compiler.DeleteNodes(schema, 'nocompile'))

    def should_delete(value):
      return isinstance(value, dict) and not value.get('valid', True)

    expected = [{'one': {'test': 'test'}}, {'valid': True}, {}]
    given = [{
        'one': {
            'test': 'test'
        },
        'two': {
            'valid': False
        }
    }, {
        'valid': True
    }, {}, {
        'valid': False
    }]
    self.assertEqual(expected, compiler.DeleteNodes(given,
                                                    matcher=should_delete))


if __name__ == '__main__':
  unittest.main()