fix: add sms provider

This commit is contained in:
TakahashiNg
2026-04-19 15:30:20 +00:00
parent 5274b408d4
commit e381397491
9 changed files with 236 additions and 21 deletions
+2
View File
@@ -5,5 +5,7 @@
"tabWidth": 2,
"trailingComma": "all",
"useTabs": true,
"xmlSortAttributesByKey": true,
"xmlQuoteAttributes": "double",
"plugins": ["prettier-plugin-java", "@prettier/plugin-xml"]
}
+5
View File
@@ -49,6 +49,11 @@
<artifactId>lib</artifactId>
<version>${project.version}</version>
</dependency>
<dependency>
<groupId>io.quarkus</groupId>
<artifactId>quarkus-junit5-mockito</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>io.quarkus</groupId>
<artifactId>quarkus-jdbc-h2</artifactId>
@@ -22,10 +22,7 @@ public class AccountService {
.entity("InvalidPhoneNumber")
.build();
String otp = otpService.generateAndSaveOtp(input.phone);
// Ở đây bạn sẽ gọi Kafka để gửi SMS thực tế
System.out.println("OTP cho " + input.phone + " là: " + otp);
otpService.generateAndSaveOtp(input.phone);
return Response.accepted().build();
}
@@ -1,9 +1,11 @@
package com.drinkool.services;
import com.drinkool.utils.SpeedSMSAPI;
import io.quarkus.redis.datasource.RedisDataSource;
import io.quarkus.redis.datasource.value.ValueCommands;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
import java.io.IOException;
import java.util.Random;
@ApplicationScoped
@@ -11,18 +13,25 @@ public class OtpService {
private final ValueCommands<String, String> countCommands;
@Inject
SpeedSMSAPI smsApi;
@Inject
public OtpService(RedisDataSource ds) {
this.countCommands = ds.value(String.class);
}
public String generateAndSaveOtp(String phone) {
public void generateAndSaveOtp(String phone) {
String otp = String.format("%06d", new Random().nextInt(1000000));
String key = "otp:" + phone;
countCommands.setex(key, 300, otp);
return otp;
try {
smsApi.sendSMS(phone, "Mã OTP drinkool của bạn là " + otp, 2, "Drinkool");
} catch (IOException e) {
e.printStackTrace();
}
}
public boolean verifyOtp(String phone, String inputOtp) {
@@ -0,0 +1,110 @@
package com.drinkool.utils;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.xml.bind.DatatypeConverter;
import java.io.*;
import java.net.*;
import org.eclipse.microprofile.config.inject.ConfigProperty;
@ApplicationScoped
public class SpeedSMSAPI {
public static final String API_URL = "https://api.speedsms.vn/index.php";
protected String mAccessToken;
public SpeedSMSAPI(
@ConfigProperty(name = "sms.api.token") String accessToken
) {
this.mAccessToken = accessToken;
}
/**
* Get user information
* @param: none
* @return: json string
* */
public String getUserInfo() throws IOException {
URI uri = URI.create(API_URL + "/user/info");
HttpURLConnection conn = (HttpURLConnection) uri.toURL().openConnection();
conn.setRequestMethod("GET");
String userCredentials = mAccessToken + ":x";
String basicAuth =
"Basic " +
DatatypeConverter.printBase64Binary(userCredentials.getBytes());
conn.setRequestProperty("Authorization", basicAuth);
BufferedReader in = new BufferedReader(
new InputStreamReader(conn.getInputStream())
);
String inputLine = "";
StringBuffer buffer = new StringBuffer();
while ((inputLine = in.readLine()) != null) {
buffer.append(inputLine);
}
in.close();
return buffer.toString();
}
/**
* Send SMS
* @param
* @return: json string
* */
public String sendSMS(String to, String content, int type, String sender)
throws IOException {
String json =
"{\"to\": [\"" +
to +
"\"], \"content\": \"" +
EncodeNonAsciiCharacters(content) +
"\", \"type\":" +
type +
", \"brandname\":\"" +
sender +
"\"}";
URI uri = URI.create(API_URL + "/sms/send");
HttpURLConnection conn = (HttpURLConnection) uri.toURL().openConnection();
conn.setRequestMethod("POST");
String userCredentials = mAccessToken + ":x";
String basicAuth =
"Basic " +
DatatypeConverter.printBase64Binary(userCredentials.getBytes());
conn.setRequestProperty("Authorization", basicAuth);
conn.setRequestProperty("Content-Type", "application/json");
conn.setDoOutput(true);
DataOutputStream wr = new DataOutputStream(conn.getOutputStream());
wr.writeBytes(json);
wr.flush();
wr.close();
BufferedReader in = new BufferedReader(
new InputStreamReader(conn.getInputStream())
);
String inputLine = "";
StringBuffer buffer = new StringBuffer();
while ((inputLine = in.readLine()) != null) {
buffer.append(inputLine);
}
in.close();
return buffer.toString();
}
private String EncodeNonAsciiCharacters(String value) {
StringBuffer sb = new StringBuffer();
for (int i = 0; i < value.length(); i++) {
char c = value.charAt(i);
int unit = (int) c;
if (unit > 127) {
String hex = String.format("%04x", (int) unit);
String encodedValue = "\\u" + hex;
sb.append(encodedValue);
} else {
sb.append(c);
}
}
return sb.toString();
}
}
@@ -1,16 +1,19 @@
package com.drinkool.services;
import static io.restassured.RestAssured.*;
import static org.hamcrest.CoreMatchers.*;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;
import com.drinkool.*;
import com.drinkool.dtos.*;
import com.drinkool.entities.CustomerEntity;
import io.quarkus.test.junit.QuarkusTest;
import com.drinkool.utils.SpeedSMSAPI;
import io.quarkus.test.junit.*;
import io.restassured.http.ContentType;
import jakarta.inject.Inject;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import org.junit.jupiter.api.*;
import org.mockito.ArgumentCaptor;
@QuarkusTest
public class CustomerServiceTest {
@@ -18,6 +21,15 @@ public class CustomerServiceTest {
@Inject
OtpService otpService;
SpeedSMSAPI smsApiMock;
@BeforeEach
public void setup() {
SpeedSMSAPI mock = mock(SpeedSMSAPI.class);
QuarkusMock.installMockForType(mock, SpeedSMSAPI.class);
this.smsApiMock = mock;
}
@Test
void testSignup() {
String phone = Utils.generateRandomPhone();
@@ -114,7 +126,7 @@ public class CustomerServiceTest {
}
@Test
public void testQuickLogin_Success() {
public void testQuickLogin_Success() throws IOException {
String phone = Utils.generateRandomPhone();
QuickSignup input = new QuickSignup();
@@ -129,7 +141,23 @@ public class CustomerServiceTest {
QuickLogin payload = new QuickLogin();
payload.phone = phone;
payload.otp = otpService.generateAndSaveOtp(phone);
otpService.generateAndSaveOtp(phone);
ArgumentCaptor<String> contentCaptor = ArgumentCaptor.forClass(
String.class
);
verify(smsApiMock).sendSMS(
eq(phone),
contentCaptor.capture(),
anyInt(),
anyString()
);
// 4. Trích xuất mã OTP từ nội dung tin nhắn đã bắt được
// Giả sử tin nhắn có định dạng: "... là 123456"
String sentMessage = contentCaptor.getValue();
payload.otp = sentMessage.substring(sentMessage.length() - 6);
given()
.contentType(ContentType.JSON)
@@ -1,11 +1,17 @@
package com.drinkool.services;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.*;
import io.quarkus.test.junit.QuarkusTest;
import com.drinkool.utils.SpeedSMSAPI;
import io.quarkus.redis.datasource.RedisDataSource;
import io.quarkus.test.junit.*;
import jakarta.inject.Inject;
import java.io.IOException;
import java.util.Random;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.*;
import org.mockito.ArgumentCaptor;
@QuarkusTest
public class OtpServiceTest {
@@ -13,24 +19,58 @@ public class OtpServiceTest {
@Inject
OtpService otpService;
@Inject
RedisDataSource redisDataSource;
SpeedSMSAPI smsApiMock;
@BeforeEach
public void setup() {
SpeedSMSAPI mock = mock(SpeedSMSAPI.class);
QuarkusMock.installMockForType(mock, SpeedSMSAPI.class);
this.smsApiMock = mock;
}
// 1. Test case: Tạo và lưu OTP thành công
@Test
public void testGenerateAndSaveOtp() {
String phone =
"+8477" + String.format("%07d", new Random().nextInt(10000000));
String otp = otpService.generateAndSaveOtp(phone);
String redisKey = "otp:" + phone;
assertNotNull(otp, "OTP sinh ra không được null");
assertEquals(6, otp.length(), "OTP phải có độ dài đúng 6 ký tự");
assertTrue(otp.matches("\\d{6}"), "OTP chỉ được chứa các chữ số");
otpService.generateAndSaveOtp(phone);
String otpFromRedis = redisDataSource.value(String.class).get(redisKey);
assertNotNull(otpFromRedis, "Mã OTP phải tồn tại trong Redis");
assertEquals(6, otpFromRedis.length(), "Mã OTP phải có 6 chữ số");
long ttl = redisDataSource.key().ttl(redisKey);
assertTrue(ttl > 0 && ttl <= 300, "TTL của OTP phải hợp lệ");
}
// 2. Test case: Xác thực OTP thành công
@Test
public void testVerifyOtpSuccess() {
public void testVerifyOtpSuccess() throws IOException {
String phone =
"+8477" + String.format("%07d", new Random().nextInt(10000000));
String generatedOtp = otpService.generateAndSaveOtp(phone);
otpService.generateAndSaveOtp(phone);
ArgumentCaptor<String> contentCaptor = ArgumentCaptor.forClass(
String.class
);
verify(smsApiMock).sendSMS(
eq(phone),
contentCaptor.capture(),
anyInt(),
anyString()
);
// 4. Trích xuất mã OTP từ nội dung tin nhắn đã bắt được
// Giả sử tin nhắn có định dạng: "... là 123456"
String sentMessage = contentCaptor.getValue();
String generatedOtp = sentMessage.substring(sentMessage.length() - 6);
// Xác thực với OTP vừa tạo ra
boolean verifyResult = otpService.verifyOtp(phone, generatedOtp);
@@ -68,10 +108,26 @@ public class OtpServiceTest {
// 5. Test case: Đảm bảo OTP chỉ được sử dụng 1 lần (đã bị xóa sau khi verify thành công bằng lệnh getdel)
@Test
public void testVerifyOtpConsumedAfterSuccess() {
public void testVerifyOtpConsumedAfterSuccess() throws IOException {
String phone =
"+8477" + String.format("%07d", new Random().nextInt(10000000));
String generatedOtp = otpService.generateAndSaveOtp(phone);
otpService.generateAndSaveOtp(phone);
ArgumentCaptor<String> contentCaptor = ArgumentCaptor.forClass(
String.class
);
verify(smsApiMock).sendSMS(
eq(phone),
contentCaptor.capture(),
anyInt(),
anyString()
);
// 4. Trích xuất mã OTP từ nội dung tin nhắn đã bắt được
// Giả sử tin nhắn có định dạng: "... là 123456"
String sentMessage = contentCaptor.getValue();
String generatedOtp = sentMessage.substring(sentMessage.length() - 6);
// Lần 1: Xác thực thành công
assertTrue(otpService.verifyOtp(phone, generatedOtp));
@@ -8,3 +8,6 @@ quarkus.http.host=0.0.0.0
## Orm
%test.quarkus.hibernate-orm.database.generation=drop-and-create
# Sms API
sms.api.token=your_sms_api_token_here
+5
View File
@@ -37,6 +37,11 @@ spec:
secretKeyRef:
name: postgres-credentials
key: url
- name: SMS_API_TOKEN
valueFrom:
secretKeyRef:
name: sms-api-token
key: value
resources:
requests:
memory: "32Mi"