# -*- Encoding: utf-8 -*-
###
# Copyright (c) 2006 Dennis Kaarsemaker
# Copyright (c) 2010 Elián Hanisch
# Copyright (c) 2018 Krytarik Raido
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
###

from supybot.test import *
import supybot.conf as conf

Econf = conf.supybot.plugins.Encyclopedia
Econf.prefixchar.set('@')

class EncyclopediaTestCase(ChannelPluginTestCase):
    plugins = ('Encyclopedia',)

    def setUp(self):
        super(EncyclopediaTestCase, self).setUp()
        conf.supybot.reply.whenNotCommand.setValue(False)
        self.createDB()

    def createDB(self):
        import sqlite3, os
        dbfile = os.path.join(Econf.datadir(), '%s.db' % Econf.database())
        try:
            os.remove(dbfile)
        except:
            pass
        db = sqlite3.connect(dbfile)
        cur = db.cursor()
        cur.execute("""CREATE TABLE facts (
            id INTEGER PRIMARY KEY,
            name TEXT NOT NULL,
            value TEXT NOT NULL,
            author TEXT NOT NULL,
            added TEXT NOT NULL,
            editor TEXT,
            edited TEXT,
            popularity INTEGER NOT NULL DEFAULT 0
        )""")
        cur.execute("""CREATE TABLE log (
            id INTEGER PRIMARY KEY,
            type TEXT NOT NULL,
            name TEXT NOT NULL,
            value TEXT NOT NULL,
            author TEXT NOT NULL,
            added TEXT NOT NULL
        )""")
        cur.execute("""CREATE TABLE requests (
            id INTEGER PRIMARY KEY,
            type TEXT NOT NULL,
            name TEXT NOT NULL,
            value TEXT NOT NULL,
            requester TEXT NOT NULL,
            requested TEXT NOT NULL
        )""")
        db.commit()
        db.close()
        self.getCallback().databases = {}

    def getCallback(self):
        for cb in self.irc.callbacks:
            if cb.name() == 'Encyclopedia':
                break
        return cb

    def testAdd(self):
        self.assertNotError('foo is bar')
        self.assertResponse('foo', 'foo is bar')

    def testEdit(self):
        self.assertNotError('foo is bar')
        self.assertNotError('no, foo is bar1')
        self.assertResponse('foo', 'foo is bar1')

    def testKeyword(self):
        self.assertNotError('hello is <reply> Hi, welcome to $chan!')
        self.assertResponse('hello', 'Hi, welcome to #test!')

    def testRequests(self):
        world.testing = False
        self.prefix = 'user!user@home.com'
        try:
            self.assertResponse('test-#ubuntu-se is <reply> blah',
                                'Your edit request has been forwarded to #ubuntu-ops.  Thank you ' \
                                'for your attention to detail')
            self.assertEqual(self.irc.takeMsg().args[1],
                             'In #test, user said: @test-#ubuntu-se is <reply> blah')
            # test in private, it shouldn't use the prefix char.
            self.assertResponse('test-#ubuntu-se is <reply> blah',
                                'Your edit request has been forwarded to #ubuntu-ops.  Thank you ' \
                                'for your attention to detail', private=True, usePrefixChar=False)
            self.assertEqual(self.irc.takeMsg().args[1],
                             'In test, user said: test-#ubuntu-se is <reply> blah')
        finally:
            world.testing = True