mirror of https://github.com/halo-dev/halo
chore: cleanup subscription integration test (#6671)
#### What type of PR is this? /kind cleanup /area core /milestone 2.20.x #### What this PR does / why we need it: 删除 SubscriptionServiceIntegrationTest 类。因为当前测试类不经常性出错，暂时无法排查原因。 #### Does this PR introduce a user-facing change? ```release-note None ```pull/6679/head
guqing
GitHub
parent
3fda9e6db4
commit
749c80cb96
|
|
@ -1,171 +0,0 @@
|
|||
package run.halo.app.notification;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.Mockito.atLeast;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static run.halo.app.extension.index.query.QueryFactory.isNull;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Nested;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.boot.test.mock.mockito.SpyBean;
|
||||
import org.springframework.test.annotation.DirtiesContext;
|
||||
import reactor.core.publisher.Flux;
|
||||
import reactor.core.publisher.Mono;
|
||||
import reactor.test.StepVerifier;
|
||||
import run.halo.app.core.extension.notification.Subscription;
|
||||
import run.halo.app.extension.Extension;
|
||||
import run.halo.app.extension.ExtensionStoreUtil;
|
||||
import run.halo.app.extension.ListOptions;
|
||||
import run.halo.app.extension.PageRequestImpl;
|
||||
import run.halo.app.extension.ReactiveExtensionClient;
|
||||
import run.halo.app.extension.SchemeManager;
|
||||
import run.halo.app.extension.index.IndexerFactory;
|
||||
import run.halo.app.extension.router.selector.FieldSelector;
|
||||
import run.halo.app.extension.store.ReactiveExtensionStoreClient;
|
||||
import run.halo.app.infra.utils.JsonUtils;
|
||||
|
||||
/**
|
||||
* Integration tests for {@link SubscriptionService}.
|
||||
*
|
||||
* @author guqing
|
||||
* @since 2.15.0
|
||||
*/
|
||||
@DirtiesContext
|
||||
@SpringBootTest
|
||||
class SubscriptionServiceIntegrationTest {
|
||||
|
||||
@Autowired
|
||||
private SchemeManager schemeManager;
|
||||
|
||||
@SpyBean
|
||||
private ReactiveExtensionClient client;
|
||||
|
||||
@Autowired
|
||||
private ReactiveExtensionStoreClient storeClient;
|
||||
|
||||
@Autowired
|
||||
private IndexerFactory indexerFactory;
|
||||
|
||||
Mono<Extension> deleteImmediately(Extension extension) {
|
||||
var name = extension.getMetadata().getName();
|
||||
var scheme = schemeManager.get(extension.getClass());
|
||||
// un-index
|
||||
var indexer = indexerFactory.getIndexer(extension.groupVersionKind());
|
||||
indexer.unIndexRecord(extension.getMetadata().getName());
|
||||
|
||||
// delete from db
|
||||
var storeName = ExtensionStoreUtil.buildStoreName(scheme, name);
|
||||
return storeClient.delete(storeName, extension.getMetadata().getVersion())
|
||||
.thenReturn(extension);
|
||||
}
|
||||
|
||||
@Nested
|
||||
class RemoveInitialBatchTest {
|
||||
static int size = 310;
|
||||
private final List<Subscription> storedSubscriptions = subscriptionsForStore();
|
||||
|
||||
@Autowired
|
||||
private SubscriptionService subscriptionService;
|
||||
|
||||
@BeforeEach
|
||||
void setUp() {
|
||||
Flux.fromIterable(storedSubscriptions)
|
||||
.flatMap(comment -> client.create(comment))
|
||||
.as(StepVerifier::create)
|
||||
.expectNextCount(storedSubscriptions.size())
|
||||
.verifyComplete();
|
||||
}
|
||||
|
||||
@AfterEach
|
||||
void tearDown() {
|
||||
Flux.fromIterable(storedSubscriptions)
|
||||
.flatMap(SubscriptionServiceIntegrationTest.this::deleteImmediately)
|
||||
.as(StepVerifier::create)
|
||||
.expectNextCount(storedSubscriptions.size())
|
||||
.verifyComplete();
|
||||
}
|
||||
|
||||
private List<Subscription> subscriptionsForStore() {
|
||||
List<Subscription> subscriptions = new ArrayList<>(size);
|
||||
for (int i = 0; i < size; i++) {
|
||||
var subscription = createSubscription();
|
||||
subscription.getMetadata().setName("subscription-" + i);
|
||||
subscriptions.add(subscription);
|
||||
}
|
||||
return subscriptions;
|
||||
}
|
||||
|
||||
@Test
|
||||
void removeTest() {
|
||||
var subscriber = new Subscription.Subscriber();
|
||||
subscriber.setName("admin");
|
||||
var interestReason = new Subscription.InterestReason();
|
||||
interestReason.setReasonType("new-comment-on-post");
|
||||
var subject = new Subscription.ReasonSubject();
|
||||
subject.setApiVersion("content.halo.run/v1alpha1");
|
||||
subject.setKind("Post");
|
||||
interestReason.setSubject(subject);
|
||||
|
||||
subscriptionService.remove(subscriber, interestReason).block();
|
||||
|
||||
verify(client, atLeast(size)).delete(any(Subscription.class));
|
||||
assertCleanedUp();
|
||||
}
|
||||
|
||||
@Test
|
||||
void removeBySubscriberTest() {
|
||||
var subscriber = new Subscription.Subscriber();
|
||||
subscriber.setName("admin");
|
||||
|
||||
subscriptionService.remove(subscriber).block();
|
||||
verify(client, atLeast(size)).delete(any(Subscription.class));
|
||||
assertCleanedUp();
|
||||
}
|
||||
|
||||
private void assertCleanedUp() {
|
||||
var listOptions = new ListOptions();
|
||||
listOptions.setFieldSelector(FieldSelector.of(isNull("metadata.deletionTimestamp")));
|
||||
client.listBy(Subscription.class, listOptions, PageRequestImpl.ofSize(1))
|
||||
.as(StepVerifier::create)
|
||||
.consumeNextWith(result -> {
|
||||
assertThat(result.getTotal()).isEqualTo(0);
|
||||
assertThat(result.getItems()).isEmpty();
|
||||
})
|
||||
.verifyComplete();
|
||||
}
|
||||
}
|
||||
|
||||
Subscription createSubscription() {
|
||||
return JsonUtils.jsonToObject("""
|
||||
{
|
||||
"spec": {
|
||||
"subscriber": {
|
||||
"name": "admin"
|
||||
},
|
||||
"unsubscribeToken": "423530c9-bec7-446e-b73b-dd98ac00ba2b",
|
||||
"reason": {
|
||||
"reasonType": "new-comment-on-post",
|
||||
"subject": {
|
||||
"name": "5152aea5-c2e8-4717-8bba-2263d46e19d5",
|
||||
"apiVersion": "content.halo.run/v1alpha1",
|
||||
"kind": "Post"
|
||||
}
|
||||
},
|
||||
"disabled": false
|
||||
},
|
||||
"apiVersion": "notification.halo.run/v1alpha1",
|
||||
"kind": "Subscription",
|
||||
"metadata": {
|
||||
"generateName": "subscription-"
|
||||
}
|
||||
}
|
||||
""", Subscription.class);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue