mirror of https://github.com/koreader/koreader
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.0 KiB
Lua
46 lines
1.0 KiB
Lua
7 years ago
|
local MockTime = {
|
||
|
original_os_time = os.time,
|
||
|
original_util_time = nil,
|
||
|
value = os.time(),
|
||
|
}
|
||
|
|
||
|
function MockTime:install()
|
||
|
assert(self ~= nil)
|
||
|
local util = require("ffi/util")
|
||
|
if self.original_util_time == nil then
|
||
|
self.original_util_time = util.gettime
|
||
|
assert(self.original_util_time ~= nil)
|
||
|
end
|
||
|
os.time = function() return self.value end
|
||
|
util.gettime = function() return self.value, 0 end
|
||
|
end
|
||
|
|
||
|
function MockTime:uninstall()
|
||
|
assert(self ~= nil)
|
||
|
local util = require("ffi/util")
|
||
|
os.time = self.original_os_time
|
||
|
if self.original_util_time ~= nil then
|
||
|
util.gettime = self.original_util_time
|
||
|
end
|
||
|
end
|
||
|
|
||
|
function MockTime:set(value)
|
||
|
assert(self ~= nil)
|
||
|
if type(value) ~= "number" then
|
||
|
return false
|
||
|
end
|
||
|
self.value = math.floor(value)
|
||
|
return true
|
||
|
end
|
||
|
|
||
|
function MockTime:increase(value)
|
||
|
assert(self ~= nil)
|
||
|
if type(value) ~= "number" then
|
||
|
return false
|
||
|
end
|
||
|
self.value = math.floor(self.value + value)
|
||
|
return true
|
||
|
end
|
||
|
|
||
|
return MockTime
|