Esempio n. 1
0
class ShowModelsCommandTestCase(TestCase):
    fixtures = ['initial_data.json']

    def setUp(self):
        self.cmd = Command()
        self.options = {'no_color': False}

    def test_if_command_exists(self):
        self.assertIsInstance(self.cmd, BaseCommand, "Command class must be instanciated from BaseCommand class")
        self.assertTrue(callable(self.cmd.handle))

    def test_command_output_is_valid(self):
        output = self.cmd._collect_data(**self.options)

        self.assertTrue(len(output), 9)
        self.assertTrue(any('Contact' in _m for _m in output))

        for line in output:
            model_name, count = line.split('\t')
            self.assertIsInstance(model_name, types.StringType)
            self.assertTrue(count.isdigit())

    def test_correct_number_of_objects_count(self):
        output = self.cmd._collect_data(**self.options)
        contact_row = filter(lambda x: 'Contact' in x, output)[0]
        model_name, count = contact_row.split('\t')
        self.assertTrue('Contact' in model_name)
        self.assertTrue('1' in count, "Incorrect number of objects for Count model %s != 0" % count)
Esempio n. 2
0
 def setUp(self):
     self.cmd = Command()
     self.options = {'no_color': False}