def test_command_call(self): commands_dict = get_commands() self.assertTrue(self.COMMAND in commands_dict) saved_streams = sys.stdout, sys.stderr sys.stdout = self.Output() sys.stderr = self.Output() prefix = "stderr:" has_errors = False try: call_command(self.COMMAND, tee=1, prefix=prefix) except: has_errors = True self.assertEqual(has_errors, False) valid_stdout = "" valid_stderr = "" for k, v in count_instances().items(): row = "%s\t%d" % (k, v) valid_stdout += "%s\n" % row valid_stderr += "%s %s\n" % (prefix, row) self.assertEquals(sys.stdout.text, valid_stdout) self.assertEquals(sys.stderr.text, valid_stderr) sys.stdout, sys.stderr = saved_streams
def test_instance_num(self): """ Calculate number of instances using models.get_models() and compare """ result = {} for m in models.get_models(): model_name = "%s.%s" % (m.__module__, m.__name__) result[model_name] = m._default_manager.count() self.assertEquals(result, count_instances())