Patch up user-agent matching and add simple tests for them

This commit is contained in:
Max Wofford
2025-05-09 11:55:39 -04:00
parent 02b472baac
commit a3c885463f
3 changed files with 121 additions and 3 deletions

View File

@@ -0,0 +1,73 @@
require "test_helper"
class WakatimeServiceTest < Minitest::Test
# Since parse_user_agent is a pure function that doesn't need database access,
# we can test it without loading any fixtures
def setup
ActiveRecord::FixtureSet.reset_cache
end
def test_parse_user_agent_with_vscode_wakatime_client
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 vscode/1.0.0 vscode-wakatime/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "vscode", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_GitHub_Desktop
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 github-desktop/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "github-desktop", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_Figma
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 figma/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "figma", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_Terminal
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 terminal/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "terminal", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_vim
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 vim/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "vim", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_Windows
user_agent = "wakatime/v1.0.0 (windows-x86_64) go1.0.0 vscode/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "windows", result[:os]
assert_equal "vscode", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_Cursor
user_agent = "wakatime/v1.0.0 (darwin-arm64) go1.0.0 cursor/1.0.0"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "darwin", result[:os]
assert_equal "cursor", result[:editor]
assert_nil result[:error]
end
def test_parse_user_agent_with_invalid_user_agent
user_agent = "invalid-user-agent"
result = WakatimeService.parse_user_agent(user_agent)
assert_equal "", result[:os]
assert_equal "", result[:editor]
assert_equal "failed to parse user agent string", result[:err]
end
end