Crystalnix / omaha-server

class SizeExceedTest(TestCase):

    maxDiff = None

    @is_private()

    def test_crashes(self):

        gpm['Crash__limit_size'] = 1

        crash_size = 10*1024*1023

        CrashFactory.create_batch(200, archive_size=crash_size, minidump_size=0)

        self.assertEqual(Crash.objects.all().count(), 200)

        del_count = 98

        deleted = list(Crash.objects.values_list('id', 'created', 'signature', 'userid', 'appid'))[:del_count]

        deleted = map(lambda x: dict(id=x[0], element_created=x[1].strftime("%d. %B %Y %I:%M%p"), signature=x[2],

                                      userid=x[3], appid=x[4]), deleted)

        result = delete_size_is_exceeded('crash', 'Crash')

        self.assertDictEqual(result, dict(count=del_count, size=del_count * crash_size, elements=deleted))

        self.assertEqual(Crash.objects.all().count(), 102)

    @is_private()

    def test_feedbacks(self):

from sparkle.factories import SparkleVersionFactory

class DeleteOldTest(TestCase):

    @is_private()

    def test_crashes(self):

        old_date = timezone.now() - timezone.timedelta(days=5)

        gpm['Crash__limit_storage_days'] = 2

        CrashFactory.create_batch(10, created=old_date)

        Crash.objects.update(created=old_date)

        self.assertEqual(Crash.objects.all().count(), 10)

        deleted = list(Crash.objects.values_list('id', 'created', 'signature', 'userid', 'appid'))

        deleted = map(lambda x: dict(id=x[0], element_created=x[1].strftime("%d. %B %Y %I:%M%p"), signature=x[2],

                                     userid=x[3], appid=x[4]), deleted)

        result = delete_older_than('crash', 'Crash')

        self.assertDictEqual(result, dict(count=10, size=0, elements=deleted))

        self.assertEqual(Crash.objects.all().count(), 0)

    @is_private()

    def test_feedbacks(self):