tka: reduce boilerplate code in the tests

Updates #cleanup Change-Id: Id69d509f5e470fb5fb50b5c5c4ca61f000389c53 Signed-off-by: Alex Chan <alexc@tailscale.com>
2026-04-28 10:25:55 +01:00
parent cb239808a6
commit 0ac09721df
7 changed files with 63 additions and 109 deletions
@@ -320,6 +320,22 @@ func optTemplate(name string, template AUM) testchainOpt {
 	}
 }

+func genesisTemplate(key Key) testchainOpt {
+	return optTemplate("genesis", AUM{MessageKind: AUMCheckpoint, State: &State{
+		Keys:              []Key{key},
+		DisablementValues: [][]byte{DisablementKDF([]byte{1, 2, 3})},
+	}})
+}
+
+func checkpointTemplate() testchainOpt {
+	fakeState := &State{
+		Keys:              []Key{{Kind: Key25519, Votes: 1}},
+		DisablementValues: [][]byte{bytes.Repeat([]byte{1}, 32)},
+	}
+
+	return optTemplate("checkpoint", AUM{MessageKind: AUMCheckpoint, State: fakeState})
+}
+
 func optKey(name string, key Key, priv ed25519.PrivateKey) testchainOpt {
 	return testchainOpt{
 		Name:    name,