Exemple #1
0
class LoadIbmSeedDataTest(TestCase):

    ANY_CONNECTION_IDENTITY = "0772334455"
    ANY_TEXT = "IBM Classifier Dummy"
    INCOMING = "I"

    categories_loaded = False
    categories_with_messages_inserted = []

    def setUp(self):
        print "setting up"
        self.load_ibm_seed_data = Command()

    def tearDown(self):
        Backend.objects.all().delete()
        Connection.objects.all().delete()
        IbmCategory.objects.all().delete()
        IbmMsgCategory.objects.all().delete()

    def test_should_insert_a_message_into_http_router_messages_table(self):
        self.assertEquals(1, 1)

    def test_should_insert_backend(self):
        self.create_backend()

        backend = Backend.objects.get(name=self.ANY_TEXT)

        self.assertIsNotNone(backend)

    def test_should_insert_connection_given_a_backend(self):
        connection = self.create_connection()

        connection = Connection.objects.get(pk=connection.id)

        self.assertIsNotNone(connection)

    def test_should_insert_message_given_a_connection(self):
        connection = self.create_connection()
        self.create_message(connection)

        message = Message.objects.get(connection=connection,
                                      direction=self.INCOMING)

        self.assertIsNotNone(message)

    def test_should_insert_ibm_message_category(self):
        connection = self.create_connection()
        message, created = self.create_message(connection)

        self.load_ibm_seed_data.load_categories()
        category = IbmCategory.objects.all()[0]

        self.load_ibm_seed_data.create_ibm_message_category(message=message,
                                                            category=category,
                                                            score=0.0123456789)
        expected_category = IbmMsgCategory.objects.get(msg=message)

        self.assertIsNotNone(expected_category)

    def load_classifier_categories(self):
        self.categories_loaded = True

    def test_should_load_categories_on_handle(self):
        self.load_ibm_seed_data.load_categories = self.load_classifier_categories

        self.load_ibm_seed_data.handle()

        self.assertTrue(self.categories_loaded)

    def test_should_load_twelve_categories_if_no_categories_exist(self):
        num_categories_before = len(IbmCategory.objects.all())

        self.load_ibm_seed_data.load_categories()
        num_categories_after = len(IbmCategory.objects.all())

        self.assertEquals(num_categories_before, 0)
        self.assertEquals(num_categories_after, 12)

    def test_should_only_load_categories_once(self):
        self.load_ibm_seed_data.load_categories()
        self.load_ibm_seed_data.load_categories()
        self.assertEquals(len(IbmCategory.objects.all()), 12)

    def test_should_insert_a_classifier_message_for_each_category(self):
        categories = self.create_dummy_categories()

        self.load_ibm_seed_data.create_ibm_message_category = self.create_ibm_message_category_mock
        self.load_ibm_seed_data.create_backend = lambda name: Backend(name=name
                                                                      )
        self.load_ibm_seed_data.create_connection = lambda backend, identity: Connection(
            backend=backend, identity=identity)
        self.load_ibm_seed_data.create_message = lambda connection, text, direction, status: (
            Message(connection=connection,
                    text=text,
                    direction=direction,
                    status=status), True)

        self.load_ibm_seed_data.create_seed_data(categories)

        self.assertEquals(self.categories_with_messages_inserted, categories)

    create_seed_data_called = False

    def test_should_call_create_seed_data_on_handle(self):
        self.load_ibm_seed_data.create_seed_data = self.create_seed_data_mock

        self.load_ibm_seed_data.handle()

        self.assertTrue(self.create_seed_data_called)

    def create_seed_data_mock(self, categories):
        self.create_seed_data_called = True

    def create_backend(self):
        return self.load_ibm_seed_data.create_backend(name=self.ANY_TEXT)

    def create_connection(self):
        backend = self.create_backend()
        connection = self.load_ibm_seed_data.create_connection(
            backend=backend, identity=self.ANY_CONNECTION_IDENTITY)

        return connection

    def create_message(self, connection):
        return self.load_ibm_seed_data.create_message(connection=connection,
                                                      text=self.ANY_TEXT,
                                                      direction=self.INCOMING,
                                                      status="H")

    def create_dummy_categories(self):
        dummy_categories = []
        for i in range(0, 12):
            dummy_categories.append(
                IbmCategory(pk=i, name="Category %d" % (i + 1)))
        return dummy_categories

    def create_ibm_message_category_mock(self, message, category, score):
        self.categories_with_messages_inserted.append(category)
class LoadIbmSeedDataTest(TestCase):

    ANY_CONNECTION_IDENTITY = "0772334455"
    ANY_TEXT = "IBM Classifier Dummy"
    INCOMING = "I"

    categories_loaded = False
    categories_with_messages_inserted = []

    def setUp(self):
        print "setting up"
        self.load_ibm_seed_data = Command()

    def tearDown(self):
        Backend.objects.all().delete()
        Connection.objects.all().delete()
        IbmCategory.objects.all().delete()
        IbmMsgCategory.objects.all().delete()

    def test_should_insert_a_message_into_http_router_messages_table(self):
        self.assertEquals(1, 1)

    def test_should_insert_backend(self):
        self.create_backend()

        backend = Backend.objects.get(name=self.ANY_TEXT)

        self.assertIsNotNone(backend)

    def test_should_insert_connection_given_a_backend(self):
        connection = self.create_connection()

        connection = Connection.objects.get(pk=connection.id)

        self.assertIsNotNone(connection)

    def test_should_insert_message_given_a_connection(self):
        connection = self.create_connection()
        self.create_message(connection)

        message = Message.objects.get(connection=connection, direction=self.INCOMING)

        self.assertIsNotNone(message)

    def test_should_insert_ibm_message_category(self):
        connection = self.create_connection()
        message, created = self.create_message(connection)

        self.load_ibm_seed_data.load_categories()
        category = IbmCategory.objects.all()[0]

        self.load_ibm_seed_data.create_ibm_message_category(message=message, category=category, score=0.0123456789)
        expected_category = IbmMsgCategory.objects.get(msg=message)

        self.assertIsNotNone(expected_category)

    def load_classifier_categories(self):
        self.categories_loaded = True

    def test_should_load_categories_on_handle(self):
        self.load_ibm_seed_data.load_categories = self.load_classifier_categories

        self.load_ibm_seed_data.handle()

        self.assertTrue(self.categories_loaded)

    def test_should_load_twelve_categories_if_no_categories_exist(self):
        num_categories_before = len(IbmCategory.objects.all())

        self.load_ibm_seed_data.load_categories()
        num_categories_after = len(IbmCategory.objects.all())

        self.assertEquals(num_categories_before, 0)
        self.assertEquals(num_categories_after, 12)

    def test_should_only_load_categories_once(self):
        self.load_ibm_seed_data.load_categories()
        self.load_ibm_seed_data.load_categories()
        self.assertEquals(len(IbmCategory.objects.all()), 12)

    def test_should_insert_a_classifier_message_for_each_category(self):
        categories = self.create_dummy_categories()

        self.load_ibm_seed_data.create_ibm_message_category = self.create_ibm_message_category_mock
        self.load_ibm_seed_data.create_backend = lambda name: Backend(name=name)
        self.load_ibm_seed_data.create_connection = lambda backend, identity: Connection(backend=backend,
                                                                                         identity=identity)
        self.load_ibm_seed_data.create_message = lambda connection, text, direction, status: (Message(
            connection=connection, text=text, direction=direction, status=status), True)

        self.load_ibm_seed_data.create_seed_data(categories)

        self.assertEquals(self.categories_with_messages_inserted, categories)

    create_seed_data_called = False

    def test_should_call_create_seed_data_on_handle(self):
        self.load_ibm_seed_data.create_seed_data = self.create_seed_data_mock

        self.load_ibm_seed_data.handle()

        self.assertTrue(self.create_seed_data_called)

    def create_seed_data_mock(self, categories):
        self.create_seed_data_called = True

    def create_backend(self):
        return self.load_ibm_seed_data.create_backend(name=self.ANY_TEXT)

    def create_connection(self):
        backend = self.create_backend()
        connection = self.load_ibm_seed_data.create_connection(backend=backend, identity=self.ANY_CONNECTION_IDENTITY)

        return connection

    def create_message(self, connection):
        return self.load_ibm_seed_data.create_message(connection=connection, text=self.ANY_TEXT,
                                                      direction=self.INCOMING, status="H")

    def create_dummy_categories(self):
        dummy_categories = []
        for i in range(0, 12):
            dummy_categories.append(IbmCategory(pk=i, name="Category %d" % (i + 1)))
        return dummy_categories

    def create_ibm_message_category_mock(self, message, category, score):
        self.categories_with_messages_inserted.append(category)
Exemple #3
0
 def setUp(self):
     print "setting up"
     self.load_ibm_seed_data = Command()
 def setUp(self):
     print "setting up"
     self.load_ibm_seed_data = Command()