Python GuidBanlistの例

コード例 #1

ファイル: test_guid_banlist.py プロジェクト: wafflecopter/BigBrotherBot-For-UrT43

    def setUp(self):
        BanlistTestCase.setUp(self)

        console = Mock()
        self.banlist = GuidBanlist(
            console,
            ET.fromstring(r"""
            <guid_banlist>
                <name>Banlist_name</name>
                <file>F:\temp\banlist.txt</file>
                <force_ip_range>no</force_ip_range>
                <message>$name is BANNED</message>
            </guid_banlist>
        """))
        #        self.banlist.getModifiedTime = Mock(return_value=time.time())
        self.banlist.plugin.info = Mock(
            wraps=lambda x: sys.stdout.write(x + "\n"))

コード例 #2

ファイル: test_guid_banlist.py プロジェクト: HaoDrang/big-brother-bot

    def setUp(self):
        BanlistTestCase.setUp(self)

        console = Mock()
        self.banlist = GuidBanlist(console, ET.fromstring(r"""
            <guid_banlist>
                <name>Banlist_name</name>
                <file>F:\temp\banlist.txt</file>
                <force_ip_range>no</force_ip_range>
                <message>$name is BANNED</message>
            </guid_banlist>
        """))
#        self.banlist.getModifiedTime = Mock(return_value=time.time())
        self.banlist.plugin.info = Mock(wraps=lambda x: sys.stdout.write(x + "\n"))

コード例 #3

ファイル: test_guid_banlist.py プロジェクト: wafflecopter/BigBrotherBot-For-UrT43

class Test_GuidBanlist(BanlistTestCase):
    def setUp(self):
        BanlistTestCase.setUp(self)

        console = Mock()
        self.banlist = GuidBanlist(
            console,
            ET.fromstring(r"""
            <guid_banlist>
                <name>Banlist_name</name>
                <file>F:\temp\banlist.txt</file>
                <force_ip_range>no</force_ip_range>
                <message>$name is BANNED</message>
            </guid_banlist>
        """))
        #        self.banlist.getModifiedTime = Mock(return_value=time.time())
        self.banlist.plugin.info = Mock(
            wraps=lambda x: sys.stdout.write(x + "\n"))

    def isBanned(self, guid):
        client = Mock()
        client.guid = guid
        return self.banlist.isBanned(client)

    def assertBanned(self, guid, expected_debug=None):
        self.banlist.plugin.info.reset_mock()
        self.assertTrue(self.isBanned(guid))
        self.assertTrue(self.banlist.plugin.info.called)
        if expected_debug:
            self.banlist.plugin.info.assert_called_with(
                "guid '%s' matches banlist entry %r (Banlist_name 2000-01-01 00:00:00)"
                % (guid, expected_debug))
        else:
            args = self.banlist.plugin.info.call_args[0][0]
            self.assertTrue(
                args.startswith("guid '%s' matches banlist entry " % guid),
                args)

    def assertNotBanned(self, guid):
        self.banlist.plugin.info.reset_mock()
        self.assertFalse(self.isBanned(guid))
        self.banlist.plugin.verbose.assert_called_with(
            "guid '%s' not found in banlist (Banlist_name 2000-01-01 00:00:00)"
            % guid)

    ##############################################################################################

    def test_empty_banlist(self):
        self.file_content = ""
        self.assertNotBanned("STEAM:0:1:123456")

    def test_match(self):
        self.file_content = '''\
STEAM:0:1:111111
STEAM:0:1:222222
  STEAM:0:1:333333
STEAM:0:1:444444
64A8FC41E14548C2B8A0C50637FAF16E
690CD3D4975A4D4B83C1960A9CA0C060
'''
        self.assertBanned("STEAM:0:1:111111")
        self.assertNotBanned("STEAM:0:1:11111")
        self.assertBanned("STEAM:0:1:222222")
        self.assertNotBanned("STEAM:0:1:22222")
        self.assertBanned("STEAM:0:1:333333")
        self.assertNotBanned("STEAM:0:1:33333")
        self.assertBanned("STEAM:0:1:444444")
        self.assertNotBanned("STEAM:0:1:44444")
        self.assertBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertNotBanned("64A8FC41E14548C2B8A0C50637FAF16")
        self.assertBanned("690CD3D4975A4D4B83C1960A9CA0C060")
        self.assertNotBanned("690CD3D4975A4D4B83C1960A9CA0C06")

    def test_commented_entries(self):
        self.file_content = '''\
//STEAM:0:1:111111
STEAM:0:1:222222 // comment after guid
// STEAM:0:1:333333
#STEAM:0:1:444444
# 64A8FC41E14548C2B8A0C50637FAF16E
## 690CD3D4975A4D4B83C1960A9CA0C060
'''
        self.assertNotBanned("STEAM:0:1:111111")
        self.assertBanned("STEAM:0:1:222222",
                          "STEAM:0:1:222222 // comment after guid")
        self.assertNotBanned("STEAM:0:1:333333")
        self.assertNotBanned("STEAM:0:1:444444")
        self.assertNotBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertNotBanned("690CD3D4975A4D4B83C1960A9CA0C060")

    def test_case(self):
        self.file_content = '''\
steam:0:1:111111
STEAM:0:1:222222
STeam:0:1:333333
64A8FC41e14548c2b8a0c50637FAF16E
690cd3d4975a4d4b83c1960a9ca0c060
'''
        self.assertNotBanned("F00")
        self.assertBanned("STEAM:0:1:111111")
        self.assertBanned("STEAM:0:1:222222")
        self.assertBanned("STEAM:0:1:333333")
        self.assertBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertBanned("690CD3D4975A4D4B83C1960A9CA0C060")

コード例 #4

ファイル: test_guid_banlist.py プロジェクト: HaoDrang/big-brother-bot

class Test_GuidBanlist(BanlistTestCase):

    def setUp(self):
        BanlistTestCase.setUp(self)

        console = Mock()
        self.banlist = GuidBanlist(console, ET.fromstring(r"""
            <guid_banlist>
                <name>Banlist_name</name>
                <file>F:\temp\banlist.txt</file>
                <force_ip_range>no</force_ip_range>
                <message>$name is BANNED</message>
            </guid_banlist>
        """))
#        self.banlist.getModifiedTime = Mock(return_value=time.time())
        self.banlist.plugin.info = Mock(wraps=lambda x: sys.stdout.write(x + "\n"))



    def isBanned(self, guid):
        client = Mock()
        client.guid = guid
        return self.banlist.isBanned(client)


    def assertBanned(self, guid, expected_debug=None):
        self.banlist.plugin.info.reset_mock()
        self.assertTrue(self.isBanned(guid))
        self.assertTrue(self.banlist.plugin.info.called)
        if expected_debug:
            self.banlist.plugin.info.assert_called_with("guid '%s' matches banlist entry %r (Banlist_name 2000-01-01 00:00:00)" % (guid, expected_debug))
        else:
            args = self.banlist.plugin.info.call_args[0][0]
            self.assertTrue(args.startswith("guid '%s' matches banlist entry " % guid), args)

    def assertNotBanned(self, guid):
        self.banlist.plugin.info.reset_mock()
        self.assertFalse(self.isBanned(guid))
        self.banlist.plugin.verbose.assert_called_with("guid '%s' not found in banlist (Banlist_name 2000-01-01 00:00:00)" % guid)


    ##############################################################################################

    def test_empty_banlist(self):
        self.file_content = ""
        self.assertNotBanned("STEAM:0:1:123456")


    def test_match(self):
        self.file_content = '''\
STEAM:0:1:111111
STEAM:0:1:222222
  STEAM:0:1:333333
STEAM:0:1:444444
64A8FC41E14548C2B8A0C50637FAF16E
690CD3D4975A4D4B83C1960A9CA0C060
'''
        self.assertBanned("STEAM:0:1:111111")
        self.assertNotBanned("STEAM:0:1:11111")
        self.assertBanned("STEAM:0:1:222222")
        self.assertNotBanned("STEAM:0:1:22222")
        self.assertBanned("STEAM:0:1:333333")
        self.assertNotBanned("STEAM:0:1:33333")
        self.assertBanned("STEAM:0:1:444444")
        self.assertNotBanned("STEAM:0:1:44444")
        self.assertBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertNotBanned("64A8FC41E14548C2B8A0C50637FAF16")
        self.assertBanned("690CD3D4975A4D4B83C1960A9CA0C060")
        self.assertNotBanned("690CD3D4975A4D4B83C1960A9CA0C06")


    def test_commented_entries(self):
        self.file_content = '''\
//STEAM:0:1:111111
STEAM:0:1:222222 // comment after guid
// STEAM:0:1:333333
#STEAM:0:1:444444
# 64A8FC41E14548C2B8A0C50637FAF16E
## 690CD3D4975A4D4B83C1960A9CA0C060
'''
        self.assertNotBanned("STEAM:0:1:111111")
        self.assertBanned("STEAM:0:1:222222", "STEAM:0:1:222222 // comment after guid")
        self.assertNotBanned("STEAM:0:1:333333")
        self.assertNotBanned("STEAM:0:1:444444")
        self.assertNotBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertNotBanned("690CD3D4975A4D4B83C1960A9CA0C060")


    def test_case(self):
        self.file_content = '''\
steam:0:1:111111
STEAM:0:1:222222
STeam:0:1:333333
64A8FC41e14548c2b8a0c50637FAF16E
690cd3d4975a4d4b83c1960a9ca0c060
'''
        self.assertNotBanned("F00")
        self.assertBanned("STEAM:0:1:111111")
        self.assertBanned("STEAM:0:1:222222")
        self.assertBanned("STEAM:0:1:333333")
        self.assertBanned("64A8FC41E14548C2B8A0C50637FAF16E")
        self.assertBanned("690CD3D4975A4D4B83C1960A9CA0C060")