from datetime import datetime

from django.test.testcases import TestCase

from django_evolution.models import Version


class VersionManagerTests(TestCase):
    """Unit tests for django_evolution.models.VersionManager."""

    def test_current_version_with_dup_timestamps(self):
        """Testing Version.current_version() with two entries with same timestamps"""
        timestamp = datetime(year=2015, month=12, day=10, hour=12, minute=13,
                             second=14)

        Version.objects.create(signature='abc123', when=timestamp)
        version = Version.objects.create(signature='abc123-def456',
                                         when=timestamp)

        latest_version = Version.objects.current_version()
        self.assertEqual(latest_version, version)

from datetime import datetime

from django.test.testcases import TestCase

from django_evolution.models import Version


class VersionManagerTests(TestCase):
    """Unit tests for django_evolution.models.VersionManager."""

    def test_current_version_with_dup_timestamps(self):
        """Testing Version.current_version() with two entries with same timestamps"""
        # Remove anything that may already exist.
        Version.objects.all().delete()

        timestamp = datetime(year=2015, month=12, day=10, hour=12, minute=13,
                             second=14)

        Version.objects.create(signature='abc123', when=timestamp)
        version = Version.objects.create(signature='abc123-def456',
                                         when=timestamp)

        latest_version = Version.objects.current_version()
        self.assertEqual(latest_version, version)

Fix unit tests for Version.objects.get_current().
Fix unit tests for Version.objects.get_current().

The unit tests assumed that no Version objects were present, but this
wasn't always the case. This clears them before the test runs.

  from datetime import datetime
  
  from django.test.testcases import TestCase
  
  from django_evolution.models import Version
  
  
  class VersionManagerTests(TestCase):
      """Unit tests for django_evolution.models.VersionManager."""
  
      def test_current_version_with_dup_timestamps(self):
          """Testing Version.current_version() with two entries with same timestamps"""
+         # Remove anything that may already exist.
+         Version.objects.all().delete()
+ 
          timestamp = datetime(year=2015, month=12, day=10, hour=12, minute=13,
                               second=14)
  
          Version.objects.create(signature='abc123', when=timestamp)
          version = Version.objects.create(signature='abc123-def456',
                                           when=timestamp)
  
          latest_version = Version.objects.current_version()
          self.assertEqual(latest_version, version)
  
Fix unit tests for Version.objects.get_current().
## Code Before:
from datetime import datetime

from django.test.testcases import TestCase

from django_evolution.models import Version


class VersionManagerTests(TestCase):
    """Unit tests for django_evolution.models.VersionManager."""

    def test_current_version_with_dup_timestamps(self):
        """Testing Version.current_version() with two entries with same timestamps"""
        timestamp = datetime(year=2015, month=12, day=10, hour=12, minute=13,
                             second=14)

        Version.objects.create(signature='abc123', when=timestamp)
        version = Version.objects.create(signature='abc123-def456',
                                         when=timestamp)

        latest_version = Version.objects.current_version()
        self.assertEqual(latest_version, version)

## Instruction:
Fix unit tests for Version.objects.get_current().
## Code After:
from datetime import datetime

from django.test.testcases import TestCase

from django_evolution.models import Version


class VersionManagerTests(TestCase):
    """Unit tests for django_evolution.models.VersionManager."""

    def test_current_version_with_dup_timestamps(self):
        """Testing Version.current_version() with two entries with same timestamps"""
        # Remove anything that may already exist.
        Version.objects.all().delete()

        timestamp = datetime(year=2015, month=12, day=10, hour=12, minute=13,
                             second=14)

        Version.objects.create(signature='abc123', when=timestamp)
        version = Version.objects.create(signature='abc123-def456',
                                         when=timestamp)

        latest_version = Version.objects.current_version()
        self.assertEqual(latest_version, version)
