#!/usr/bin/env python3

from __future__ import print_function

import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

import unittest

from vndk_definition_tool import BA_WARN, BannedLibDict


class BannedLibDictTest(unittest.TestCase):
    def test_add(self):
        d = BannedLibDict()
        d.add('libfoo.so', 'test', BA_WARN)
        x = d.get('libfoo.so')

        self.assertIsNotNone(x)
        self.assertEqual('libfoo.so', x.name)
        self.assertEqual('test', x.reason)
        self.assertEqual(BA_WARN, x.action)

    def test_get(self):
        d = BannedLibDict.create_default()
        self.assertIsNotNone(d.get('libbinder.so'))
        self.assertIsNotNone(d.get('libselinux.so'))
        self.assertIsNone(d.get('libc.so'))

    def test_is_banned(self):
        d = BannedLibDict.create_default()
        self.assertTrue(d.is_banned('/system/lib/libbinder.so'))
        self.assertTrue(d.is_banned('/system/lib/libselinux.so'))
        self.assertTrue(d.is_banned('/system/lib64/libbinder.so'))
        self.assertTrue(d.is_banned('/system/lib64/libselinux.so'))
        self.assertFalse(d.is_banned('/system/lib64/libc.so'))


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