Beispiel #1
0
    def test_sync_successful(self, mock_quota_create, mock_quota_update,
                             mock_quota_del, mock_quota_get_all,
                             mock_list_quotas, get_lock):
        quota_obj = resources.StorageQuotaTask(
            context, 'c5c91c98-91aa-40e6-85ac-37a1d3b32bda')
        quota_obj.sync()
        self.assertTrue(mock_list_quotas.called)
        self.assertTrue(mock_quota_get_all.called)
        self.assertTrue(get_lock.called)

        # collect the quotas from fake_storage
        fake_storage_obj = fake_storage.FakeStorageDriver()

        # add the quotas to DB
        mock_list_quotas.return_value =\
            fake_storage_obj.list_quotas(context)
        mock_quota_get_all.return_value = list()
        quota_obj.sync()
        self.assertTrue(mock_quota_create.called)

        # update the quotas to DB
        mock_list_quotas.return_value = quotas_list
        mock_quota_get_all.return_value = quotas_list
        quota_obj.sync()
        self.assertTrue(mock_quota_update.called)

        # delete the quotas to DB
        mock_list_quotas.return_value = list()
        mock_quota_get_all.return_value = quotas_list
        quota_obj.sync()
        self.assertTrue(mock_quota_del.called)
Beispiel #2
0
 def test_remove(self, mock_quota_del):
     quota_obj = resources.StorageQuotaTask(
         context, 'c5c91c98-91aa-40e6-85ac-37a1d3b32bda')
     quota_obj.remove()
     self.assertTrue(mock_quota_del.called)