import unittest
import os

from programytest.aiml_tests.client import TestClient
from programy.config.sections.brain.file import BrainFileConfiguration

class SraiSetTestClient(TestClient):

    def __init__(self):
        TestClient.__init__(self)

    def load_configuration(self, arguments):
        super(SraiSetTestClient, self).load_configuration(arguments)
        self.configuration.brain_configuration.files.aiml_files._files = [os.path.dirname(__file__)]

class SraiAIMLTests(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        SraiAIMLTests.test_client = SraiSetTestClient()

    def test_srai__set_response(self):
        response = SraiAIMLTests.test_client.bot.ask_question("test", "TEST SRAI SET")
        self.assertIsNotNone(response)
        self.assertEqual(response, 'BLANK RESPONSE')

import unittest
import os

from programytest.aiml_tests.client import TestClient
from programy.config.sections.brain.file import BrainFileConfiguration

class SraiSetTestClient(TestClient):

    def __init__(self):
        TestClient.__init__(self)

    def load_configuration(self, arguments):
        super(SraiSetTestClient, self).load_configuration(arguments)
        self.configuration.brain_configuration.files.aiml_files._files = [os.path.dirname(__file__)]

class SraiAIMLTests(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        SraiAIMLTests.test_client = SraiSetTestClient()

    def test_srai__set_response(self):
        response = SraiAIMLTests.test_client.bot.ask_question("test", "TEST SRAI SET")
        self.assertIsNotNone(response)
        self.assertEqual(response, 'BLANK RESPONSE')
