Exemple #1
0
 def compress_string(self, s):
     """Gzip a given string."""
     zbuf = StringIO()
     zfile = gzip.GzipFile(mode='wb', compresslevel=6, fileobj=zbuf)
     zfile.write(s)
     zfile.close()
     return zbuf.getvalue()
 def compress_string(self, s):
     """Gzip a given string."""
     zbuf = StringIO()
     zfile = gzip.GzipFile(mode='wb', compresslevel=6, fileobj=zbuf)
     zfile.write(s)
     zfile.close()
     return zbuf.getvalue()
 def test_some_output(self):
     out = StringIO()
     call_command('show_template_tags', stdout=out)
     output = out.getvalue()
     # Once django_extension is installed during tests it should appear with
     # its templatetags
     self.assertIn('django_extensions', output)
     # let's check at least one
     self.assertIn('truncate_letters', output)
 def test_some_output(self):
     out = StringIO()
     call_command('show_template_tags', stdout=out)
     output = out.getvalue()
     # Once django_extension is installed during tests it should appear with
     # its templatetags
     self.assertIn('django_extensions', output)
     # let's check at least one
     self.assertIn('truncate_letters', output)
    def test_works(self):
        from django.db import models

        class PermModel(models.Model):
            class Meta:
                app_label = 'django_extensions'
                permissions = (('test_permission', 'test_permission'), )

        original_stdout = sys.stdout
        out = sys.stdout = StringIO()
        call_command('update_permissions', stdout=out, verbosity=3)
        sys.stdout = original_stdout
        self.assertIn("Can change perm model", out.getvalue())
    def test_works(self):
        from django_extensions.management.signals import post_command, \
            pre_command
        from django_extensions.management.commands.show_template_tags import \
            Command

        def pre(sender, **kwargs):
            CommandSignalTests.pre = dict(**kwargs)

        def post(sender, **kwargs):
            CommandSignalTests.post = dict(**kwargs)

        pre_command.connect(pre, Command)
        post_command.connect(post, Command)

        out = StringIO()
        call_command('show_template_tags', stdout=out)

        self.assertIn('args', CommandSignalTests.pre)
        self.assertIn('kwargs', CommandSignalTests.pre)

        self.assertIn('args', CommandSignalTests.post)
        self.assertIn('kwargs', CommandSignalTests.post)
        self.assertIn('outcome', CommandSignalTests.post)