Skip to content

Commit

Permalink
Optimize content format
Browse files Browse the repository at this point in the history
  • Loading branch information
doleyzi committed May 21, 2024
1 parent 3bdf06c commit 513846e
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 14 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,10 @@
import lombok.AllArgsConstructor;
import lombok.Data;

/**
* The Audit information description currently only supports Chinese and English.
* If other languages need to be added, it can be expanded.
*/
@Data
@AllArgsConstructor
public class AuditInformation {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@

public enum FlowType {

INPUT("received", "接收"),
OUTPUT("send", "发送");
INPUT("received", " 接收"),
OUTPUT("send", " 发送");
private final String nameInEnglish;
private final String nameInChinese;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,43 +44,43 @@ public void buildAuditInformation() {
AuditInformation auditInfo =
AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(AGENT_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("Agent接收成功", auditInfo.getNameInChinese());
assertEquals("Agent 接收成功", auditInfo.getNameInChinese());

auditType = "DataProxy";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(DATA_PROXY_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("DataProxy接收成功", auditInfo.getNameInChinese());
assertEquals("DataProxy 接收成功", auditInfo.getNameInChinese());

auditType = "Hive";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(SORT_HIVE_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("Hive接收成功", auditInfo.getNameInChinese());
assertEquals("Hive 接收成功", auditInfo.getNameInChinese());

auditType = "StarRocks";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(SORT_STARROCKS_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("StarRocks接收成功", auditInfo.getNameInChinese());
assertEquals("StarRocks 接收成功", auditInfo.getNameInChinese());

// Test the scenario of dataFlow case compatibility.
auditType = "agent";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(AGENT_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("Agent接收成功", auditInfo.getNameInChinese());
assertEquals("Agent 接收成功", auditInfo.getNameInChinese());

auditType = "dataProxy";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(DATA_PROXY_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("DataProxy接收成功", auditInfo.getNameInChinese());
assertEquals("DataProxy 接收成功", auditInfo.getNameInChinese());

auditType = "hive";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(SORT_HIVE_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("Hive接收成功", auditInfo.getNameInChinese());
assertEquals("Hive 接收成功", auditInfo.getNameInChinese());

auditType = "STARROCKS";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals(SORT_STARROCKS_INPUT.getValue(), auditInfo.getAuditId());
assertEquals("StarRocks接收成功", auditInfo.getNameInChinese());
assertEquals("StarRocks 接收成功", auditInfo.getNameInChinese());

// Test send failed audit items.
auditType = "Agent";
Expand All @@ -90,28 +90,30 @@ public void buildAuditInformation() {
discard = false;
retry = false;
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals("Agent发送失败", auditInfo.getNameInChinese());
assertEquals("Agent 发送失败", auditInfo.getNameInChinese());

auditType = "DataProxy";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals("DataProxy发送失败", auditInfo.getNameInChinese());
assertEquals("DataProxy 发送失败", auditInfo.getNameInChinese());

auditType = "Hive";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals("Hive发送失败", auditInfo.getNameInChinese());
assertEquals("Hive 发送失败", auditInfo.getNameInChinese());

auditType = "StarRocks";
auditInfo = AuditManagerUtils.buildAuditInformation(auditType, flowType, success, isRealtime, discard, retry);
assertEquals("StarRocks发送失败", auditInfo.getNameInChinese());
assertEquals("StarRocks 发送失败", auditInfo.getNameInChinese());
}

@Test
public void getAllAuditInformation() {
List<AuditInformation> auditInfoList = AuditManagerUtils.getAllAuditInformation();
System.out.println(auditInfoList);
assertTrue(auditInfoList.size() > 0);

auditInfoList.clear();
auditInfoList = AuditManagerUtils.getAllAuditInformation("Agent");

assertTrue(auditInfoList.size() > 0);
}

Expand Down

0 comments on commit 513846e

Please sign in to comment.