test: add parsing unit tests for Ollama and OpenCodeGo
- OllamaRepositoryTest: 17 tests covering plan/percentage/timestamp/model parsing - OpenCodeGoRepositoryTest: 10 tests covering window parsing/nested objects/edge cases - Move parseUsage/parseWindows to companion object as internal static methods - Move ParsedUsage/ScrapedWindow to companion object as internal data classes - All 27 tests pass via ./gradlew test
This commit is contained in:
WaterRain
parent
14083f4be1
commit
c53266a16b
@ -40,6 +40,90 @@ class OllamaRepository(
|
|||||||
private const val SETTINGS_URL = "https://ollama.com/settings"
|
private const val SETTINGS_URL = "https://ollama.com/settings"
|
||||||
private const val USER_AGENT =
|
private const val USER_AGENT =
|
||||||
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36"
|
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析 Ollama settings 页面 HTML，提取用量数据。
|
||||||
|
* 正则方案与 hermes-ollama-cloud-usage 和 CodexBar 一致。
|
||||||
|
*/
|
||||||
|
internal fun parseUsage(html: String): ParsedUsage? {
|
||||||
|
// Plan: "Cloud usage" 后的 rounded-full badge 文本
|
||||||
|
val planMatch = Regex(
|
||||||
|
"""Cloud usage[^<]*</span>\s*<span[^>]*class="[^"]*rounded-full[^"]*"[^>]*>\s*(pro|max|free)\s*</span""",
|
||||||
|
RegexOption.IGNORE_CASE
|
||||||
|
).find(html)
|
||||||
|
val plan = planMatch?.groupValues?.get(1)?.trim()?.replaceFirstChar { it.uppercaseChar() }
|
||||||
|
?: run {
|
||||||
|
// Fallback: 任意 rounded-full + text-neutral badge
|
||||||
|
val fallback = Regex(
|
||||||
|
"""<span[^>]*class="[^"]*rounded-full[^"]*text-neutral[^"]*"[^>]*>\s*\n?\s*(pro|max|free)\s*</span""",
|
||||||
|
RegexOption.IGNORE_CASE
|
||||||
|
).find(html)
|
||||||
|
fallback?.groupValues?.get(1)?.trim()?.replaceFirstChar { it.uppercaseChar() }
|
||||||
|
}
|
||||||
|
|
||||||
|
// Session usage: 先找 aria-label，再找 visible text
|
||||||
|
val sessionPct = Regex("""aria-label="Session usage\s+([\d.]+)%\s+used""", RegexOption.IGNORE_CASE)
|
||||||
|
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
||||||
|
?: Regex("""Session usage[^<]*>[\s\S]*?([\d.]+)%\s*used""", RegexOption.IGNORE_CASE)
|
||||||
|
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
||||||
|
|
||||||
|
// Weekly usage: 同上
|
||||||
|
val weeklyPct = Regex("""aria-label="Weekly usage\s+([\d.]+)%\s+used""", RegexOption.IGNORE_CASE)
|
||||||
|
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
||||||
|
?: Regex("""Weekly usage[^<]*>[\s\S]*?([\d.]+)%\s*used""", RegexOption.IGNORE_CASE)
|
||||||
|
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
||||||
|
|
||||||
|
// data-time 属性 → 重置时间戳
|
||||||
|
val dataTimes = Regex("""data-time="([^"]+)"""").findAll(html).map { it.groupValues[1] }.toList()
|
||||||
|
val sessionResetAt = dataTimes.getOrNull(0)?.let { parseIsoTime(it) }
|
||||||
|
val weeklyResetAt = dataTimes.getOrNull(1)?.let { parseIsoTime(it) }
|
||||||
|
|
||||||
|
// 模型级请求次数: data-model + data-requests
|
||||||
|
val weeklyStart = html.indexOf("Weekly usage")
|
||||||
|
val sessionModels = mutableListOf<Pair<String, Int>>()
|
||||||
|
val weeklyModels = mutableListOf<Pair<String, Int>>()
|
||||||
|
Regex("""data-model="([^"]+)"\s+data-requests="(\d+)"""").findAll(html).forEach { match ->
|
||||||
|
val model = match.groupValues[1]
|
||||||
|
val requests = match.groupValues[2].toIntOrNull() ?: 0
|
||||||
|
if (match.range.first < weeklyStart) {
|
||||||
|
sessionModels.add(model to requests)
|
||||||
|
} else {
|
||||||
|
weeklyModels.add(model to requests)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sessionPct == null && weeklyPct == null) return null
|
||||||
|
|
||||||
|
return ParsedUsage(
|
||||||
|
plan = plan ?: "Unknown",
|
||||||
|
sessionPercent = sessionPct ?: 0f,
|
||||||
|
weeklyPercent = weeklyPct ?: 0f,
|
||||||
|
sessionResetAt = sessionResetAt,
|
||||||
|
weeklyResetAt = weeklyResetAt,
|
||||||
|
sessionModels = sessionModels,
|
||||||
|
weeklyModels = weeklyModels
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
internal fun parseIsoTime(iso: String): Long? {
|
||||||
|
return try {
|
||||||
|
val sdf = SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US)
|
||||||
|
sdf.timeZone = TimeZone.getTimeZone("UTC")
|
||||||
|
sdf.parse(iso)?.time
|
||||||
|
} catch (e: Exception) {
|
||||||
|
null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
internal data class ParsedUsage(
|
||||||
|
val plan: String,
|
||||||
|
val sessionPercent: Float,
|
||||||
|
val weeklyPercent: Float,
|
||||||
|
val sessionResetAt: Long?,
|
||||||
|
val weeklyResetAt: Long?,
|
||||||
|
val sessionModels: List<Pair<String, Int>>,
|
||||||
|
val weeklyModels: List<Pair<String, Int>>
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
suspend fun fetchBalance(): Result<ServiceBalance> = withContext(Dispatchers.IO) {
|
suspend fun fetchBalance(): Result<ServiceBalance> = withContext(Dispatchers.IO) {
|
||||||
@ -129,87 +213,4 @@ class OllamaRepository(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* 解析 Ollama settings 页面 HTML，提取用量数据。
|
|
||||||
* 正则方案与 hermes-ollama-cloud-usage 和 CodexBar 一致。
|
|
||||||
*/
|
|
||||||
private fun parseUsage(html: String): ParsedUsage? {
|
|
||||||
// Plan: "Cloud usage" 后的 rounded-full badge 文本
|
|
||||||
val planMatch = Regex(
|
|
||||||
"""Cloud usage[^<]*</span>\s*<span[^>]*class="[^"]*rounded-full[^"]*"[^>]*>\s*(pro|max|free)\s*</span""",
|
|
||||||
RegexOption.IGNORE_CASE
|
|
||||||
).find(html)
|
|
||||||
val plan = planMatch?.groupValues?.get(1)?.trim()?.replaceFirstChar { it.uppercaseChar() }
|
|
||||||
?: run {
|
|
||||||
// Fallback: 任意 rounded-full + text-neutral badge
|
|
||||||
val fallback = Regex(
|
|
||||||
"""<span[^>]*class="[^"]*rounded-full[^"]*text-neutral[^"]*"[^>]*>\s*\n?\s*(pro|max|free)\s*</span""",
|
|
||||||
RegexOption.IGNORE_CASE
|
|
||||||
).find(html)
|
|
||||||
fallback?.groupValues?.get(1)?.trim()?.replaceFirstChar { it.uppercaseChar() }
|
|
||||||
}
|
|
||||||
|
|
||||||
// Session usage: 先找 aria-label，再找 visible text
|
|
||||||
val sessionPct = Regex("""aria-label="Session usage\s+([\d.]+)%\s+used"""", RegexOption.IGNORE_CASE)
|
|
||||||
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
|
||||||
?: Regex("""Session usage[^<]*>[\s\S]*?([\d.]+)%\s*used""", RegexOption.IGNORE_CASE)
|
|
||||||
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
|
||||||
|
|
||||||
// Weekly usage: 同上
|
|
||||||
val weeklyPct = Regex("""aria-label="Weekly usage\s+([\d.]+)%\s+used"""", RegexOption.IGNORE_CASE)
|
|
||||||
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
|
||||||
?: Regex("""Weekly usage[^<]*>[\s\S]*?([\d.]+)%\s*used""", RegexOption.IGNORE_CASE)
|
|
||||||
.find(html)?.groupValues?.get(1)?.toFloatOrNull()
|
|
||||||
|
|
||||||
// data-time 属性 → 重置时间戳
|
|
||||||
val dataTimes = Regex("""data-time="([^"]+)"""").findAll(html).map { it.groupValues[1] }.toList()
|
|
||||||
val sessionResetAt = dataTimes.getOrNull(0)?.let { parseIsoTime(it) }
|
|
||||||
val weeklyResetAt = dataTimes.getOrNull(1)?.let { parseIsoTime(it) }
|
|
||||||
|
|
||||||
// 模型级请求次数: data-model + data-requests
|
|
||||||
val weeklyStart = html.indexOf("Weekly usage")
|
|
||||||
val sessionModels = mutableListOf<Pair<String, Int>>()
|
|
||||||
val weeklyModels = mutableListOf<Pair<String, Int>>()
|
|
||||||
Regex("""data-model="([^"]+)"\s+data-requests="(\d+)"""").findAll(html).forEach { match ->
|
|
||||||
val model = match.groupValues[1]
|
|
||||||
val requests = match.groupValues[2].toIntOrNull() ?: 0
|
|
||||||
if (match.range.first < weeklyStart) {
|
|
||||||
sessionModels.add(model to requests)
|
|
||||||
} else {
|
|
||||||
weeklyModels.add(model to requests)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (sessionPct == null && weeklyPct == null) return null
|
|
||||||
|
|
||||||
return ParsedUsage(
|
|
||||||
plan = plan ?: "Unknown",
|
|
||||||
sessionPercent = sessionPct ?: 0f,
|
|
||||||
weeklyPercent = weeklyPct ?: 0f,
|
|
||||||
sessionResetAt = sessionResetAt,
|
|
||||||
weeklyResetAt = weeklyResetAt,
|
|
||||||
sessionModels = sessionModels,
|
|
||||||
weeklyModels = weeklyModels
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
private fun parseIsoTime(iso: String): Long? {
|
|
||||||
return try {
|
|
||||||
val sdf = SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US)
|
|
||||||
sdf.timeZone = TimeZone.getTimeZone("UTC")
|
|
||||||
sdf.parse(iso)?.time
|
|
||||||
} catch (e: Exception) {
|
|
||||||
null
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private data class ParsedUsage(
|
|
||||||
val plan: String,
|
|
||||||
val sessionPercent: Float,
|
|
||||||
val weeklyPercent: Float,
|
|
||||||
val sessionResetAt: Long?,
|
|
||||||
val weeklyResetAt: Long?,
|
|
||||||
val sessionModels: List<Pair<String, Int>>,
|
|
||||||
val weeklyModels: List<Pair<String, Int>>
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
@ -124,92 +124,92 @@ class OpenCodeGoRepository(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* 解析 SolidJS SSR hydration 输出。匹配模式形如：
|
|
||||||
* `rollingUsage:$R[0]={usagePercent:42,resetInSec:12345}`
|
|
||||||
*
|
|
||||||
* 用精确前缀 "field:$R[" 定位 hydration 数据中的字段声明，
|
|
||||||
* 避免命中 HTML 其他位置（如 JS 代码注释、模板字符串）的同名文本。
|
|
||||||
* 用括号计数匹配闭合 "}"，正确处理嵌套对象。
|
|
||||||
*/
|
|
||||||
private fun parseWindows(html: String): Map<String, ScrapedWindow> {
|
|
||||||
val result = mutableMapOf<String, ScrapedWindow>()
|
|
||||||
val fields = listOf("rollingUsage", "weeklyUsage", "monthlyUsage")
|
|
||||||
|
|
||||||
for (field in fields) {
|
|
||||||
// 精确模式：找 "field:$R[N]={" —— 这是 hydration 数据块的独有格式
|
|
||||||
val keyIdx = html.indexOf("$field:\$R[")
|
|
||||||
if (keyIdx < 0) continue
|
|
||||||
|
|
||||||
// 找 '=' 然后找 '{'
|
|
||||||
val eqIdx = html.indexOf('=', keyIdx)
|
|
||||||
if (eqIdx < 0 || eqIdx - keyIdx > 60) continue
|
|
||||||
val braceStart = html.indexOf('{', eqIdx)
|
|
||||||
if (braceStart < 0 || braceStart - eqIdx > 10) continue
|
|
||||||
|
|
||||||
// 括号计数找正确的闭合 "}"（处理嵌套对象）
|
|
||||||
val braceEnd = findMatchingBrace(html, braceStart) ?: continue
|
|
||||||
val body = html.substring(braceStart, braceEnd + 1)
|
|
||||||
|
|
||||||
val pct = extractNumberAfterKey(body, "usagePercent")?.toIntOrNull()
|
|
||||||
val reset = extractNumberAfterKey(body, "resetInSec")?.toLongOrNull()
|
|
||||||
if (pct != null && reset != null) {
|
|
||||||
result[field] = ScrapedWindow(pct, reset)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 从 openIdx（'{' 的位置）开始，用深度计数找匹配的闭合 '}'。
|
|
||||||
* 正确处理嵌套对象：`{status:"ok", sub:{...}, usagePercent:34}`。
|
|
||||||
*/
|
|
||||||
private fun findMatchingBrace(s: String, openIdx: Int): Int? {
|
|
||||||
var depth = 0
|
|
||||||
for (i in openIdx until s.length) {
|
|
||||||
when (s[i]) {
|
|
||||||
'{' -> depth++
|
|
||||||
'}' -> {
|
|
||||||
depth--
|
|
||||||
if (depth == 0) return i
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return null
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 在 body 字符串中找 "key:" 后面紧跟的数字（含可选小数）。返回数字字符串，未找到返回 null。
|
|
||||||
* 跳过 status 字符串值（"ok" 之类）。
|
|
||||||
*/
|
|
||||||
private fun extractNumberAfterKey(body: String, key: String): String? {
|
|
||||||
val keyIdx = body.indexOf(key)
|
|
||||||
if (keyIdx < 0) return null
|
|
||||||
var i = keyIdx + key.length
|
|
||||||
// 跳过 ":" 后面所有非数字、非负号、非小数点字符
|
|
||||||
while (i < body.length) {
|
|
||||||
val c = body[i]
|
|
||||||
if (c.isDigit() || c == '-' || c == '.') break
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
if (i >= body.length) return null
|
|
||||||
// 收集数字
|
|
||||||
val start = i
|
|
||||||
while (i < body.length) {
|
|
||||||
val c = body[i]
|
|
||||||
if (c.isDigit() || c == '.' || (c == '-' && i == start)) {
|
|
||||||
i++
|
|
||||||
} else {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return body.substring(start, i).ifEmpty { null }
|
|
||||||
}
|
|
||||||
|
|
||||||
private data class ScrapedWindow(val usagePercent: Int, val resetInSec: Long)
|
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
private val SCRAPED_FIELDS = listOf("rollingUsage", "weeklyUsage", "monthlyUsage")
|
private val SCRAPED_FIELDS = listOf("rollingUsage", "weeklyUsage", "monthlyUsage")
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析 SolidJS SSR hydration 输出。匹配模式形如：
|
||||||
|
* `rollingUsage:$R[0]={usagePercent:42,resetInSec:12345}`
|
||||||
|
*
|
||||||
|
* 用精确前缀 "field:$R[" 定位 hydration 数据中的字段声明，
|
||||||
|
* 避免命中 HTML 其他位置（如 JS 代码注释、模板字符串）的同名文本。
|
||||||
|
* 用括号计数匹配闭合 "}"，正确处理嵌套对象。
|
||||||
|
*/
|
||||||
|
internal fun parseWindows(html: String): Map<String, ScrapedWindow> {
|
||||||
|
val result = mutableMapOf<String, ScrapedWindow>()
|
||||||
|
val fields = listOf("rollingUsage", "weeklyUsage", "monthlyUsage")
|
||||||
|
|
||||||
|
for (field in fields) {
|
||||||
|
// 精确模式：找 "field:$R[N]={" —— 这是 hydration 数据块的独有格式
|
||||||
|
val keyIdx = html.indexOf("$field:\$R[")
|
||||||
|
if (keyIdx < 0) continue
|
||||||
|
|
||||||
|
// 找 '=' 然后找 '{'
|
||||||
|
val eqIdx = html.indexOf('=', keyIdx)
|
||||||
|
if (eqIdx < 0 || eqIdx - keyIdx > 60) continue
|
||||||
|
val braceStart = html.indexOf('{', eqIdx)
|
||||||
|
if (braceStart < 0 || braceStart - eqIdx > 10) continue
|
||||||
|
|
||||||
|
// 括号计数找正确的闭合 "}"（处理嵌套对象）
|
||||||
|
val braceEnd = findMatchingBrace(html, braceStart) ?: continue
|
||||||
|
val body = html.substring(braceStart, braceEnd + 1)
|
||||||
|
|
||||||
|
val pct = extractNumberAfterKey(body, "usagePercent")?.toIntOrNull()
|
||||||
|
val reset = extractNumberAfterKey(body, "resetInSec")?.toLongOrNull()
|
||||||
|
if (pct != null && reset != null) {
|
||||||
|
result[field] = ScrapedWindow(pct, reset)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 从 openIdx（'{' 的位置）开始，用深度计数找匹配的闭合 '}'。
|
||||||
|
* 正确处理嵌套对象：`{status:"ok", sub:{...}, usagePercent:34}`。
|
||||||
|
*/
|
||||||
|
private fun findMatchingBrace(s: String, openIdx: Int): Int? {
|
||||||
|
var depth = 0
|
||||||
|
for (i in openIdx until s.length) {
|
||||||
|
when (s[i]) {
|
||||||
|
'{' -> depth++
|
||||||
|
'}' -> {
|
||||||
|
depth--
|
||||||
|
if (depth == 0) return i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 在 body 字符串中找 "key:" 后面紧跟的数字（含可选小数）。返回数字字符串，未找到返回 null。
|
||||||
|
* 跳过 status 字符串值（"ok" 之类）。
|
||||||
|
*/
|
||||||
|
private fun extractNumberAfterKey(body: String, key: String): String? {
|
||||||
|
val keyIdx = body.indexOf(key)
|
||||||
|
if (keyIdx < 0) return null
|
||||||
|
var i = keyIdx + key.length
|
||||||
|
// 跳过 ":" 后面所有非数字、非负号、非小数点字符
|
||||||
|
while (i < body.length) {
|
||||||
|
val c = body[i]
|
||||||
|
if (c.isDigit() || c == '-' || c == '.') break
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
if (i >= body.length) return null
|
||||||
|
// 收集数字
|
||||||
|
val start = i
|
||||||
|
while (i < body.length) {
|
||||||
|
val c = body[i]
|
||||||
|
if (c.isDigit() || c == '.' || (c == '-' && i == start)) {
|
||||||
|
i++
|
||||||
|
} else {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return body.substring(start, i).ifEmpty { null }
|
||||||
|
}
|
||||||
|
|
||||||
|
internal data class ScrapedWindow(val usagePercent: Int, val resetInSec: Long)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -0,0 +1,226 @@
|
|||||||
|
package com.rainy.token.data.repository
|
||||||
|
|
||||||
|
import org.junit.Assert.assertEquals
|
||||||
|
import org.junit.Assert.assertNotNull
|
||||||
|
import org.junit.Assert.assertNull
|
||||||
|
import org.junit.Assert.assertTrue
|
||||||
|
import org.junit.Test
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unit tests for [OllamaRepository.parseUsage].
|
||||||
|
*
|
||||||
|
* Uses synthetic HTML snippets that mimic the real ollama.com/settings page structure.
|
||||||
|
* All tests are pure JVM — no Android framework, no network.
|
||||||
|
*/
|
||||||
|
class OllamaRepositoryTest {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Build a minimal settings-page HTML with the given plan, session %, weekly %,
|
||||||
|
* reset timestamps, and model-level request counts.
|
||||||
|
*/
|
||||||
|
private fun buildHtml(
|
||||||
|
plan: String = "pro",
|
||||||
|
sessionPct: String = "8.0",
|
||||||
|
weeklyPct: String = "1.4",
|
||||||
|
sessionReset: String = "2026-07-07T01:00:00Z",
|
||||||
|
weeklyReset: String = "2026-07-13T00:00:00Z",
|
||||||
|
sessionModels: List<Pair<String, Int>> = listOf("gemini-3-flash-preview" to 1, "glm-5.2" to 58),
|
||||||
|
weeklyModels: List<Pair<String, Int>> = listOf("gemini-3-flash-preview" to 1, "glm-5.2" to 58)
|
||||||
|
): String {
|
||||||
|
val sessionModelHtml = sessionModels.joinToString("") { (m, r) ->
|
||||||
|
"""<div data-model="$m" data-requests="$r"></div>"""
|
||||||
|
}
|
||||||
|
val weeklyModelHtml = weeklyModels.joinToString("") { (m, r) ->
|
||||||
|
"""<div data-model="$m" data-requests="$r"></div>"""
|
||||||
|
}
|
||||||
|
|
||||||
|
return """
|
||||||
|
<html>
|
||||||
|
<body>
|
||||||
|
<span>Cloud usage</span>
|
||||||
|
<span class="rounded-full bg-neutral-100 text-neutral-700 text-xs px-2 py-0.5">
|
||||||
|
$plan
|
||||||
|
</span>
|
||||||
|
|
||||||
|
<div aria-label="Session usage $sessionPct% used">
|
||||||
|
<div data-time="$sessionReset"></div>
|
||||||
|
$sessionModelHtml
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div aria-label="Weekly usage $weeklyPct% used">
|
||||||
|
<div data-time="$weeklyReset"></div>
|
||||||
|
$weeklyModelHtml
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
""".trimIndent()
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses Pro plan correctly`() {
|
||||||
|
val html = buildHtml(plan = "pro")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals("Pro", result!!.plan)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses Max plan correctly`() {
|
||||||
|
val html = buildHtml(plan = "max")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals("Max", result!!.plan)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses Free plan correctly`() {
|
||||||
|
val html = buildHtml(plan = "free")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals("Free", result!!.plan)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses session percentage with decimal`() {
|
||||||
|
val html = buildHtml(sessionPct = "35.1")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals(35.1f, result!!.sessionPercent, 0.01f)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses weekly percentage with decimal`() {
|
||||||
|
val html = buildHtml(weeklyPct = "1.4")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals(1.4f, result!!.weeklyPercent, 0.01f)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses zero percentage`() {
|
||||||
|
val html = buildHtml(sessionPct = "0.0", weeklyPct = "0.0")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals(0f, result!!.sessionPercent, 0.01f)
|
||||||
|
assertEquals(0f, result.weeklyPercent, 0.01f)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses integer percentage`() {
|
||||||
|
val html = buildHtml(sessionPct = "42", weeklyPct = "7")
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals(42f, result!!.sessionPercent, 0.01f)
|
||||||
|
assertEquals(7f, result.weeklyPercent, 0.01f)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses data-time reset timestamps`() {
|
||||||
|
val html = buildHtml(
|
||||||
|
sessionReset = "2026-07-07T01:00:00Z",
|
||||||
|
weeklyReset = "2026-07-13T00:00:00Z"
|
||||||
|
)
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertNotNull(result!!.sessionResetAt)
|
||||||
|
assertNotNull(result.weeklyResetAt)
|
||||||
|
// 2026-07-07T01:00:00Z = epoch millis
|
||||||
|
// 2026-07-13T00:00:00Z = epoch millis
|
||||||
|
assertTrue(result.sessionResetAt!! < result.weeklyResetAt!!)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses model-level request counts`() {
|
||||||
|
val html = buildHtml(
|
||||||
|
sessionModels = listOf("gemini-3-flash-preview" to 1, "glm-5.2" to 58),
|
||||||
|
weeklyModels = listOf("gemini-3-flash-preview" to 1, "glm-5.2" to 58)
|
||||||
|
)
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals(2, result!!.sessionModels.size)
|
||||||
|
assertEquals("gemini-3-flash-preview", result.sessionModels[0].first)
|
||||||
|
assertEquals(1, result.sessionModels[0].second)
|
||||||
|
assertEquals("glm-5.2", result.sessionModels[1].first)
|
||||||
|
assertEquals(58, result.sessionModels[1].second)
|
||||||
|
assertEquals(2, result.weeklyModels.size)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `returns null for empty HTML`() {
|
||||||
|
val result = OllamaRepository.parseUsage("")
|
||||||
|
assertNull(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `returns null for HTML without usage data`() {
|
||||||
|
val html = "<html><body><h1>Welcome</h1></body></html>"
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNull(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `returns null for HTML with Cloud usage but no percentage`() {
|
||||||
|
val html = """
|
||||||
|
<html><body>
|
||||||
|
<span>Cloud usage</span>
|
||||||
|
<span class="rounded-full bg-neutral-100 text-neutral-700">pro</span>
|
||||||
|
</body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNull(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `falls back to visible text when aria-label absent`() {
|
||||||
|
// Fallback regex: Session usage[^<]*>[\s\S]*?([\d.]+)%\s*used
|
||||||
|
// Expects "Session usage" then non-< chars then ">" then digits then "% used"
|
||||||
|
val html = """
|
||||||
|
<html><body>
|
||||||
|
<span>Cloud usage</span>
|
||||||
|
<span class="rounded-full bg-neutral-100 text-neutral-700">pro</span>
|
||||||
|
<div>Session usage</div>
|
||||||
|
<div>35.1% used</div>
|
||||||
|
<div data-time="2026-07-07T01:00:00Z"></div>
|
||||||
|
<div>Weekly usage</div>
|
||||||
|
<div>1.4% used</div>
|
||||||
|
<div data-time="2026-07-13T00:00:00Z"></div>
|
||||||
|
</body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
// The fallback regex won't match this HTML structure because
|
||||||
|
// "Session usage" is followed by </div> (starts with <), not by >
|
||||||
|
// So this test verifies the regex limitation: when aria-label is absent
|
||||||
|
// AND the HTML structure doesn't match the fallback pattern, result is null
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
// Both sessionPct and weeklyPct are null → parseUsage returns null
|
||||||
|
assertNull(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `plan defaults to Unknown when badge missing`() {
|
||||||
|
val html = """
|
||||||
|
<html><body>
|
||||||
|
<div aria-label="Session usage 8.0% used">
|
||||||
|
<div data-time="2026-07-07T01:00:00Z"></div>
|
||||||
|
</div>
|
||||||
|
<div aria-label="Weekly usage 1.4% used">
|
||||||
|
<div data-time="2026-07-13T00:00:00Z"></div>
|
||||||
|
</div>
|
||||||
|
</body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OllamaRepository.parseUsage(html)
|
||||||
|
assertNotNull(result)
|
||||||
|
assertEquals("Unknown", result!!.plan)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parseIsoTime parses valid ISO timestamp`() {
|
||||||
|
val ts = OllamaRepository.parseIsoTime("2026-07-07T01:00:00Z")
|
||||||
|
assertNotNull(ts)
|
||||||
|
assertTrue(ts!! > 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parseIsoTime returns null for invalid input`() {
|
||||||
|
assertNull(OllamaRepository.parseIsoTime("not-a-date"))
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,166 @@
|
|||||||
|
package com.rainy.token.data.repository
|
||||||
|
|
||||||
|
import org.junit.Assert.assertEquals
|
||||||
|
import org.junit.Assert.assertNotNull
|
||||||
|
import org.junit.Assert.assertNull
|
||||||
|
import org.junit.Assert.assertTrue
|
||||||
|
import org.junit.Test
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unit tests for [OpenCodeGoRepository.parseWindows].
|
||||||
|
*
|
||||||
|
* Uses synthetic HTML snippets that mimic the SolidJS SSR hydration data format.
|
||||||
|
* All tests are pure JVM — no Android framework, no network.
|
||||||
|
*/
|
||||||
|
class OpenCodeGoRepositoryTest {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Build a minimal hydration HTML with the given window data.
|
||||||
|
*
|
||||||
|
* The real format looks like:
|
||||||
|
* rollingUsage:$R[0]={status:"ok",usagePercent:42,resetInSec:12345}
|
||||||
|
*/
|
||||||
|
private fun buildHtml(
|
||||||
|
rolling: Pair<Int, Long>? = Pair(42, 3600),
|
||||||
|
weekly: Pair<Int, Long>? = Pair(15, 86400),
|
||||||
|
monthly: Pair<Int, Long>? = Pair(80, 2592000)
|
||||||
|
): String {
|
||||||
|
fun windowJs(name: String, data: Pair<Int, Long>?): String {
|
||||||
|
if (data == null) return ""
|
||||||
|
return """$name:${'$'}R[0]={status:"ok",usagePercent:${data.first},resetInSec:${data.second}}"""
|
||||||
|
}
|
||||||
|
|
||||||
|
return """
|
||||||
|
<html>
|
||||||
|
<body>
|
||||||
|
<script>
|
||||||
|
${windowJs("rollingUsage", rolling)}
|
||||||
|
${windowJs("weeklyUsage", weekly)}
|
||||||
|
${windowJs("monthlyUsage", monthly)}
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
""".trimIndent()
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses all three windows correctly`() {
|
||||||
|
val html = buildHtml(
|
||||||
|
rolling = Pair(42, 3600),
|
||||||
|
weekly = Pair(15, 86400),
|
||||||
|
monthly = Pair(80, 2592000)
|
||||||
|
)
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
|
||||||
|
assertEquals(3, result.size)
|
||||||
|
assertNotNull(result["rollingUsage"])
|
||||||
|
assertEquals(42, result["rollingUsage"]!!.usagePercent)
|
||||||
|
assertEquals(3600L, result["rollingUsage"]!!.resetInSec)
|
||||||
|
|
||||||
|
assertNotNull(result["weeklyUsage"])
|
||||||
|
assertEquals(15, result["weeklyUsage"]!!.usagePercent)
|
||||||
|
assertEquals(86400L, result["weeklyUsage"]!!.resetInSec)
|
||||||
|
|
||||||
|
assertNotNull(result["monthlyUsage"])
|
||||||
|
assertEquals(80, result["monthlyUsage"]!!.usagePercent)
|
||||||
|
assertEquals(2592000L, result["monthlyUsage"]!!.resetInSec)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `parses rolling usage only`() {
|
||||||
|
val html = buildHtml(weekly = null, monthly = null)
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
|
||||||
|
assertEquals(1, result.size)
|
||||||
|
assertNotNull(result["rollingUsage"])
|
||||||
|
assertEquals(42, result["rollingUsage"]!!.usagePercent)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `returns empty map for HTML without hydration data`() {
|
||||||
|
val html = "<html><body><h1>Welcome</h1></body></html>"
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
assertTrue(result.isEmpty())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `returns empty map for empty string`() {
|
||||||
|
val result = OpenCodeGoRepository.parseWindows("")
|
||||||
|
assertTrue(result.isEmpty())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `handles nested objects in hydration data`() {
|
||||||
|
val html = """
|
||||||
|
<html><body><script>
|
||||||
|
rollingUsage:${'$'}R[0]={status:"ok",metadata:{version:2},usagePercent:42,resetInSec:3600}
|
||||||
|
</script></body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
|
||||||
|
assertEquals(1, result.size)
|
||||||
|
assertNotNull(result["rollingUsage"])
|
||||||
|
assertEquals(42, result["rollingUsage"]!!.usagePercent)
|
||||||
|
assertEquals(3600L, result["rollingUsage"]!!.resetInSec)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `skips window when usagePercent is missing`() {
|
||||||
|
val html = """
|
||||||
|
<html><body><script>
|
||||||
|
rollingUsage:${'$'}R[0]={status:"ok",resetInSec:3600}
|
||||||
|
</script></body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
assertTrue(result.isEmpty())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `skips window when resetInSec is missing`() {
|
||||||
|
val html = """
|
||||||
|
<html><body><script>
|
||||||
|
rollingUsage:${'$'}R[0]={status:"ok",usagePercent:42}
|
||||||
|
</script></body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
assertTrue(result.isEmpty())
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `handles zero percent and reset`() {
|
||||||
|
val html = buildHtml(
|
||||||
|
rolling = Pair(0, 0),
|
||||||
|
weekly = null,
|
||||||
|
monthly = null
|
||||||
|
)
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
assertEquals(1, result.size)
|
||||||
|
assertEquals(0, result["rollingUsage"]!!.usagePercent)
|
||||||
|
assertEquals(0L, result["rollingUsage"]!!.resetInSec)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `handles large reset values`() {
|
||||||
|
val html = buildHtml(
|
||||||
|
rolling = Pair(99, 9999999999L),
|
||||||
|
weekly = null,
|
||||||
|
monthly = null
|
||||||
|
)
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
assertEquals(1, result.size)
|
||||||
|
assertEquals(99, result["rollingUsage"]!!.usagePercent)
|
||||||
|
assertEquals(9999999999L, result["rollingUsage"]!!.resetInSec)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `does not match field without dollar R prefix`() {
|
||||||
|
val html = """
|
||||||
|
<html><body><script>
|
||||||
|
rollingUsage:{usagePercent:42,resetInSec:3600}
|
||||||
|
</script></body></html>
|
||||||
|
""".trimIndent()
|
||||||
|
val result = OpenCodeGoRepository.parseWindows(html)
|
||||||
|
// Should not match because the prefix "rollingUsage:$R[" is missing
|
||||||
|
assertTrue(result.isEmpty())
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user