run_tests.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python

# This Source Code Form is subject to the terms of the Mozilla Public

# License, v. 2.0. If a copy of the MPL was not distributed with this

# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from lxml import etree

import sys

import unittest

from scrape_plugins import Overlay

class TestOverlays(unittest.TestCase):

    def setUp(self):

        self.plugin = etree.parse("Tests/Base/testplugin.xml")

    def testAppend(self):

        overlay = Overlay("Tests/Overlays/append.xml")

        overlay.apply(self.plugin)

        self.assertEqualsExpectedXML(plugin=self.plugin, expectedPath="Tests/Expected/append.xml")

    def testReplace(self):

        overlay = Overlay("Tests/Overlays/replace.xml")

        overlay.apply(self.plugin)

        self.assertEqualsExpectedXML(plugin=self.plugin, expectedPath="Tests/Expected/replace.xml")

    def assertEqualsExpectedXML(self, plugin, expectedPath):

        actual = etree.tostring(plugin, pretty_print=True)

        with open(expectedPath, "r") as file:

            expected = file.read()

            self.assertEqual(actual, expected, "\nExpected:\n%s\n\nActual:\n%s" % (expected, actual))

if __name__ == '__main__':

    unittest.main()