wallet: Reserve UTXOs used in build_utxos

This commit is contained in:
Christian Decker
2017-05-31 16:16:59 +02:00
committed by Rusty Russell
parent 19a4e7f542
commit bdb6b1002f
4 changed files with 82 additions and 8 deletions

View File

@@ -6,7 +6,7 @@
#include <stdio.h>
#include <unistd.h>
static bool test_wallet_add_utxo(void)
static bool test_wallet_outputs(void)
{
char filename[] = "/tmp/ldb-XXXXXX";
struct utxo u;
@@ -16,26 +16,52 @@ static bool test_wallet_add_utxo(void)
close(fd);
w->db = db_open(w, filename);
CHECK_MSG(w->db,"Failed opening the db");
CHECK_MSG(w->db, "Failed opening the db");
CHECK_MSG(db_migrate(w->db), "DB migration failed");
memset(&u, 0, sizeof(u));
/* Should work, it's the first time we add it */
CHECK_MSG(wallet_add_utxo(w, &u, p2sh_wpkh), "wallet_add_utxo failed on first add");
CHECK_MSG(wallet_add_utxo(w, &u, p2sh_wpkh),
"wallet_add_utxo failed on first add");
/* Should fail, we already have that UTXO */
CHECK_MSG(!wallet_add_utxo(w, &u, p2sh_wpkh), "wallet_add_utxo succeeded on second add");
CHECK_MSG(!wallet_add_utxo(w, &u, p2sh_wpkh),
"wallet_add_utxo succeeded on second add");
/* Attempt to reserve the utxo */
CHECK_MSG(wallet_update_output_status(w, &u.txid, u.outnum,
output_state_available,
output_state_reserved),
"could not reserve available output");
/* Reserving twice should fail */
CHECK_MSG(!wallet_update_output_status(w, &u.txid, u.outnum,
output_state_available,
output_state_reserved),
"could reserve already reserved output");
/* Un-reserving should work */
CHECK_MSG(wallet_update_output_status(w, &u.txid, u.outnum,
output_state_reserved,
output_state_available),
"could not unreserve reserved output");
/* Switching from any to something else */
CHECK_MSG(wallet_update_output_status(w, &u.txid, u.outnum,
output_state_any,
output_state_spent),
"could not change output state ignoring oldstate");
tal_free(w);
return true;
}
int main(void)
{
bool ok = true;
ok &= test_wallet_add_utxo();
ok &= test_wallet_outputs();
return !ok;
}