mirror of
https://github.com/urbit/shrub.git
synced 2024-12-21 09:51:36 +03:00
a14b6e06d3
This makes the comet mining code actually work. You can now run king with `new --comet` to mine a new comet and get it on the network. Mining appears to be significantly faster; I've had to wait up to 20 minutes with vere, but I've never needed to wait more than 30 seconds with king.
89 lines
2.9 KiB
Haskell
89 lines
2.9 KiB
Haskell
module DawnTests (tests) where
|
|
|
|
import Arvo.Event
|
|
import Noun.Conversions
|
|
import UrbitPrelude
|
|
|
|
import Test.Tasty
|
|
import Test.Tasty.HUnit
|
|
|
|
import qualified Urbit.Ob as Ob
|
|
import qualified Vere.Dawn as Dawn
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
-- These golden cases generated in Urbit from entropy to make sure our +mix,
|
|
-- +shas, +shaf, etc. were actually calculated correctly.
|
|
|
|
cordToAtomBytes :: Text -> ByteString
|
|
cordToAtomBytes t = cordToAtom t ^. atomBytes
|
|
|
|
cordToAtom :: Text -> Atom
|
|
cordToAtom t = case cordToUW (Cord t) of
|
|
Nothing -> error "Couldn't parse constant embedded in file."
|
|
Just (UW a) -> a
|
|
|
|
testString = cordToAtomBytes $ concat
|
|
[ "0w1.XqnKc.onYJK.0zVOU.Uw142.jNx3C.oWV83.TYt6T.kmHUg.cnoq1.zla6B.bKeNa"
|
|
, ".8wUZu.6ZLHJ.c1TKV.KPcb3.9lU3~.p2G8D"
|
|
]
|
|
|
|
testSalt = cordToAtomBytes "0wc.~cOwa.Kb-DI.BrjVW.i0U37"
|
|
|
|
mixByteStrings = (Dawn.mix testSalt testString) @?= expected
|
|
where
|
|
expected = cordToAtomBytes $ concat
|
|
[ "0w1.XqnKc.onYJK.0zVOU.Uw142.jNx3C.oWV83.TYt6T.kmHUg.cnoq1.zla6B.bKeNa"
|
|
, ".8wUZu.6ZLHx.Pd5eP.0UOIL.IeHW5.b2ibw"
|
|
]
|
|
|
|
shasByteStrings = (Dawn.shas testSalt testString) @?= expected
|
|
where
|
|
expected = cordToAtomBytes
|
|
"0wfKW.mXzrj.c~IBb.lKd6k.2njoG.bRLcD.9eszA.gSSs8.mHRah"
|
|
|
|
shafByteStrings = (Dawn.shaf testSalt testString) @?= expected
|
|
where
|
|
expected = cordToAtomBytes "0w3h.Bg1Qh.ZZjoJ.23J~p.PHg-D"
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
cometShip :: Ship
|
|
cometShip = case Ob.parsePatp cometStr of
|
|
Left x -> error "Invalid ship name"
|
|
Right p -> Ship $ fromIntegral $ Ob.fromPatp p
|
|
where
|
|
cometStr = "~radmes-dilsec-sovlup-lagwep--tonred-waldeb-tocseg-marzod"
|
|
|
|
cometPass :: Pass
|
|
cometPass = case fromNoun (Atom cometPassAtom) of
|
|
Nothing -> error "Keyfile does not seem to contain a seed."
|
|
Just s -> s
|
|
where
|
|
cometPassAtom = cordToAtom $ concat
|
|
[ "0w99.P80w4.rL7Qt.0i5-h.8yta7.RhgHI.nrXjO.xBCix.Pxx5a.sJ6bv.a-Iwo.OeVBr"
|
|
, ".x8-Gs.1LLG~.FgDRk.GML3Y.X3qFZ.jtlpy"
|
|
]
|
|
|
|
cometRawBS = cordToAtomBytes "0w39.q35g-.hrd3f.q9UWK.Zxg40"
|
|
|
|
-- Verifies the internal usage of +shaf in comet derivation gets the right
|
|
-- answer.
|
|
testCometFingerprintBS = (Dawn.cometFingerprintBS cometPass) @?= cometRawBS
|
|
|
|
-- Tests that the real public interface for fingerprint generation does the
|
|
-- byte-munging correctly.
|
|
testCometFingerprint = (Dawn.cometFingerprint cometPass) @?= cometShip
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
tests :: TestTree
|
|
tests =
|
|
testGroup "Dawn"
|
|
[ testCase "Mix bytestrings of different length" $ mixByteStrings
|
|
, testCase "Shas bytestrings" $ shasByteStrings
|
|
, testCase "Shaf bytestrings" $ shafByteStrings
|
|
, testCase "Fingerprint bytestring derivation" $ testCometFingerprintBS
|
|
, testCase "Fingerprint total derivation" $ testCometFingerprint
|
|
]
|