def test_set_cache_value(self, mock_cache, mock_get_cache_key): value = "some_value" attribute = 'progress' model_name = 'ExportTaskRecord' uid = 'test_uid' etr = ExportTaskRecord(uid=uid) expiration = 1 cache_key = "{}.{}.{}".format('ExportTaskRecord', uid, attribute) mock_get_cache_key.return_value = cache_key set_cache_value(obj=etr, attribute=attribute, uid=uid, model_name=model_name, value=value, expiration=1) mock_get_cache_key.assert_called_once_with(obj=etr, attribute=attribute, uid=uid, model_name=model_name) mock_cache.set.assert_called_once_with(cache_key, value, timeout=1)
def test_set_cache_value(self, mock_cache, mock_get_cache_key): value = "some_value" attribute = "progress" model_name = "ExportTaskRecord" uid = "test_uid" etr = ExportTaskRecord(uid=uid) cache_key = "{}.{}.{}".format("ExportTaskRecord", uid, attribute) mock_get_cache_key.return_value = cache_key set_cache_value(obj=etr, attribute=attribute, uid=uid, model_name=model_name, value=value, expiration=1) mock_get_cache_key.assert_called_once_with(obj=etr, attribute=attribute, uid=uid, model_name=model_name) mock_cache.set.assert_called_once_with(cache_key, value, timeout=1)
def estimated_finish(self, value, expiration=DEFAULT_CACHE_EXPIRTATION): return set_cache_value(obj=self, attribute="estimated_finish", value=value, expiration=expiration)
def progress(self, value, expiration=DEFAULT_CACHE_EXPIRTATION): return set_cache_value(obj=self, attribute="progress", value=value, expiration=expiration)
def status(self, value, expiration=DEFAULT_CACHE_EXPIRATION): return set_cache_value(obj=self, attribute="status", value=value, expiration=expiration)
def message(self, value, expiration=DEFAULT_CACHE_EXPIRATION): return set_cache_value(obj=self, attribute="message", value=value, expiration=expiration)