def setUp(self):
self._mock_migratable_db = Mock()
self._mock_ddbb = Mock()
self._mock_version_provider = Mock()
self._mock_upgrade_context = Mock()
self._installed_data = InstalledData(self._mock_migratable_db,
self._mock_ddbb,
self._mock_version_provider,
self._mock_upgrade_context)
def setUp(self):
self._mock_migratable_db = Mock()
self._mock_ddbb = Mock()
self._mock_version_provider = Mock()
self._mock_upgrade_context = Mock()
self._installed_data = InstalledData(self._mock_migratable_db, self._mock_ddbb, self._mock_version_provider, self._mock_upgrade_context)
class InstalledDataTest(unittest.TestCase):
def setUp(self):
self._mock_migratable_db = Mock()
self._mock_ddbb = Mock()
self._mock_version_provider = Mock()
self._mock_upgrade_context = Mock()
self._installed_data = InstalledData(self._mock_migratable_db,
self._mock_ddbb,
self._mock_version_provider,
self._mock_upgrade_context)
def test_get_version_should_return_migrate_version_when_available(self):
self._mock_migratable_db.get_version.return_value = 1
self.assertEqual(1, self._installed_data.get_version())
def test_get_version_should_return_legacy_version_when_available(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = 2
self.assertEqual(2, self._installed_data.get_version())
def test_get_version_should_return_none_when_no_existing_version(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = None
self.assertEqual(None, self._installed_data.get_version())
def test_get_state_should_return_current_when_data_version_equals_repository_version(
self):
self._mock_migratable_db.get_version.return_value = 1
self._mock_migratable_db.get_upgrade_version.return_value = 1
self.assertEqual(DataState.CURRENT, self._installed_data.get_state())
def test_get_state_should_return_fresh_when_data_version_unavailable(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = None
self.assertEqual(DataState.FRESH, self._installed_data.get_state())
def test_get_state_should_return_stale_when_data_version_less_than_repository_version(
self):
self._mock_migratable_db.get_version.return_value = 1
self._mock_migratable_db.get_upgrade_version.return_value = 2
self.assertEqual(DataState.STALE, self._installed_data.get_state())
def test_get_state_should_return_legacy_when_data_version_is_legacy(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = 1
self.assertEqual(DataState.LEGACY, self._installed_data.get_state())
def test_get_state_should_raise_error_when_version_too_large(self):
self._mock_migratable_db.is_versioned.return_value = True
self._mock_migratable_db.get_version.return_value = 2
self._mock_migratable_db.get_upgrade_version.return_value = 1
try:
self._installed_data.get_state()
except DataInitializationException:
pass
else:
self.fail()
def test_get_available_version_returns_migratable_db_upgrade_version(self):
self._mock_migratable_db.get_upgrade_version.return_value = 1
self.assertEqual(1, self._installed_data.get_available_version())
def test_is_versioned_returns_migratable_db_is_versioned(self):
self._mock_migratable_db.is_versioned.return_value = True
self.assertEqual(True, self._installed_data.is_versioned())
def test_initialize_version_versions_migratable_db(self):
self._installed_data.initialize_version(1)
self._mock_migratable_db.version.assert_called_with(1)
def test_initialize_creates_db_tables(self):
self._installed_data.initialize()
self.assertTrue(self._mock_ddbb.create_tables.called)
def test_upgrade_upgrades_migratable_db(self):
self._installed_data.upgrade()
self.assertTrue(self._mock_migratable_db.upgrade.called)
def test_upgrade_initializes_global_upgrade_context(self):
self._installed_data.upgrade()
self.assertEqual(self._mock_upgrade_context,
pytrainer.upgrade.context.UPGRADE_CONTEXT)
def test_update_to_current_delegates_to_data_state(self):
mock_state = Mock()
self._installed_data.get_state = Mock(return_value=mock_state)
self._installed_data.update_to_current()
self.assertTrue(mock_state.update_to_current.called)
class InstalledDataTest(unittest.TestCase):
def setUp(self):
self._mock_migratable_db = Mock()
self._mock_ddbb = Mock()
self._mock_version_provider = Mock()
self._mock_upgrade_context = Mock()
self._installed_data = InstalledData(self._mock_migratable_db, self._mock_ddbb, self._mock_version_provider, self._mock_upgrade_context)
def test_get_version_should_return_migrate_version_when_available(self):
self._mock_migratable_db.get_version.return_value = 1
self.assertEqual(1, self._installed_data.get_version())
def test_get_version_should_return_legacy_version_when_available(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = 2
self.assertEqual(2, self._installed_data.get_version())
def test_get_version_should_return_none_when_no_existing_version(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = None
self.assertEqual(None, self._installed_data.get_version())
def test_get_state_should_return_current_when_data_version_equals_repository_version(self):
self._mock_migratable_db.get_version.return_value = 1
self._mock_migratable_db.get_upgrade_version.return_value = 1
self.assertEqual(DataState.CURRENT, self._installed_data.get_state())
def test_get_state_should_return_fresh_when_data_version_unavailable(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = None
self.assertEqual(DataState.FRESH, self._installed_data.get_state())
def test_get_state_should_return_stale_when_data_version_less_than_repository_version(self):
self._mock_migratable_db.get_version.return_value = 1
self._mock_migratable_db.get_upgrade_version.return_value = 2
self.assertEqual(DataState.STALE, self._installed_data.get_state())
def test_get_state_should_return_legacy_when_data_version_is_legacy(self):
self._mock_migratable_db.is_versioned.return_value = False
self._mock_version_provider.get_legacy_version.return_value = 1
self.assertEqual(DataState.LEGACY, self._installed_data.get_state())
def test_get_state_should_raise_error_when_version_too_large(self):
self._mock_migratable_db.is_versioned.return_value = True
self._mock_migratable_db.get_version.return_value = 2
self._mock_migratable_db.get_upgrade_version.return_value = 1
try:
self._installed_data.get_state()
except DataInitializationException:
pass
else:
self.fail()
def test_get_available_version_returns_migratable_db_upgrade_version(self):
self._mock_migratable_db.get_upgrade_version.return_value = 1
self.assertEqual(1, self._installed_data.get_available_version())
def test_is_versioned_returns_migratable_db_is_versioned(self):
self._mock_migratable_db.is_versioned.return_value = True
self.assertEqual(True, self._installed_data.is_versioned())
def test_initialize_version_versions_migratable_db(self):
self._installed_data.initialize_version(1)
self._mock_migratable_db.version.assert_called_with(1)
def test_initialize_creates_db_tables(self):
self._installed_data.initialize()
self.assertTrue(self._mock_ddbb.create_tables.called)
def test_upgrade_upgrades_migratable_db(self):
self._installed_data.upgrade()
self.assertTrue(self._mock_migratable_db.upgrade.called)
def test_upgrade_initializes_global_upgrade_context(self):
self._installed_data.upgrade()
self.assertEqual(self._mock_upgrade_context, pytrainer.upgrade.context.UPGRADE_CONTEXT)
def test_update_to_current_delegates_to_data_state(self):
mock_state = Mock()
self._installed_data.get_state = Mock(return_value=mock_state)
self._installed_data.update_to_current()
self.assertTrue(mock_state.update_to_current.called)