108 lines
3.3 KiB
Lua
108 lines
3.3 KiB
Lua
describe('cp.cache', function()
|
|
local cache
|
|
|
|
before_each(function()
|
|
cache = require('cp.cache')
|
|
cache.load()
|
|
end)
|
|
|
|
after_each(function()
|
|
cache.clear_contest_data('atcoder', 'test_contest')
|
|
cache.clear_contest_data('codeforces', 'test_contest')
|
|
cache.clear_contest_data('cses', 'test_contest')
|
|
end)
|
|
|
|
describe('load and save', function()
|
|
it('loads without error when cache file exists', function()
|
|
assert.has_no_errors(function()
|
|
cache.load()
|
|
end)
|
|
end)
|
|
|
|
it('saves and persists data', function()
|
|
local problems = { { id = 'A', name = 'Problem A' } }
|
|
|
|
assert.has_no_errors(function()
|
|
cache.set_contest_data('atcoder', 'test_contest', problems)
|
|
end)
|
|
|
|
local result = cache.get_contest_data('atcoder', 'test_contest')
|
|
assert.is_not_nil(result)
|
|
assert.equals('A', result.problems[1].id)
|
|
end)
|
|
end)
|
|
|
|
describe('contest data', function()
|
|
it('stores and retrieves contest data', function()
|
|
local problems = {
|
|
{ id = 'A', name = 'First Problem' },
|
|
{ id = 'B', name = 'Second Problem' },
|
|
}
|
|
|
|
cache.set_contest_data('codeforces', 'test_contest', problems)
|
|
local result = cache.get_contest_data('codeforces', 'test_contest')
|
|
|
|
assert.is_not_nil(result)
|
|
assert.equals(2, #result.problems)
|
|
assert.equals('A', result.problems[1].id)
|
|
assert.equals('Second Problem', result.problems[2].name)
|
|
end)
|
|
|
|
it('returns nil for missing contest', function()
|
|
local result = cache.get_contest_data('atcoder', 'nonexistent_contest')
|
|
assert.is_nil(result)
|
|
end)
|
|
|
|
it('clears contest data', function()
|
|
local problems = { { id = 'A' } }
|
|
cache.set_contest_data('atcoder', 'test_contest', problems)
|
|
|
|
cache.clear_contest_data('atcoder', 'test_contest')
|
|
local result = cache.get_contest_data('atcoder', 'test_contest')
|
|
|
|
assert.is_nil(result)
|
|
end)
|
|
|
|
it('handles cses expiry correctly', function()
|
|
local problems = { { id = 'A' } }
|
|
cache.set_contest_data('cses', 'test_contest', problems)
|
|
|
|
local result = cache.get_contest_data('cses', 'test_contest')
|
|
assert.is_not_nil(result)
|
|
assert.is_not_nil(result.expires_at)
|
|
end)
|
|
end)
|
|
|
|
describe('test cases', function()
|
|
it('stores and retrieves test cases', function()
|
|
local test_cases = {
|
|
{ index = 1, input = '1 2', expected = '3' },
|
|
{ index = 2, input = '4 5', expected = '9' },
|
|
}
|
|
|
|
cache.set_test_cases('atcoder', 'test_contest', 'A', test_cases)
|
|
local result = cache.get_test_cases('atcoder', 'test_contest', 'A')
|
|
|
|
assert.is_not_nil(result)
|
|
assert.equals(2, #result)
|
|
assert.equals('1 2', result[1].input)
|
|
assert.equals('9', result[2].expected)
|
|
end)
|
|
|
|
it('handles contest-level test cases', function()
|
|
local test_cases = { { input = 'test', expected = 'output' } }
|
|
|
|
cache.set_test_cases('cses', 'test_contest', nil, test_cases)
|
|
local result = cache.get_test_cases('cses', 'test_contest', nil)
|
|
|
|
assert.is_not_nil(result)
|
|
assert.equals(1, #result)
|
|
assert.equals('test', result[1].input)
|
|
end)
|
|
|
|
it('returns nil for missing test cases', function()
|
|
local result = cache.get_test_cases('atcoder', 'nonexistent', 'A')
|
|
assert.is_nil(result)
|
|
end)
|
|
end)
|
|
end)
|