From 68fabe146465919eba37f220716ac52d3e9642a6 Mon Sep 17 00:00:00 2001
From: Daniel Larimer <dlarimer@invictus-innovations.com>
Date: Thu, 2 Jul 2015 11:18:52 -0400
Subject: [PATCH] Progress #127 - importing balances

---
 libraries/chain/operations.cpp  | 1 +
 libraries/wallet/wallet.cpp     | 1 +
 tests/tests/operation_tests.cpp | 1 -
 3 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/libraries/chain/operations.cpp b/libraries/chain/operations.cpp
index c970d181..1da28524 100644
--- a/libraries/chain/operations.cpp
+++ b/libraries/chain/operations.cpp
@@ -953,6 +953,7 @@ void  balance_claim_operation::get_required_auth(flat_set<account_id_type>& acti
 void  balance_claim_operation::validate()const
 {
    FC_ASSERT( fee == asset() );
+   FC_ASSERT( balance_owner_key != public_key_type() );
 }
 
 
diff --git a/libraries/wallet/wallet.cpp b/libraries/wallet/wallet.cpp
index c87749aa..e95eebe1 100644
--- a/libraries/wallet/wallet.cpp
+++ b/libraries/wallet/wallet.cpp
@@ -2262,6 +2262,7 @@ signed_transaction wallet_api::import_balance( string name_or_id, const vector<s
          {
             op.total_claimed = b.vesting_policy.valid()? asset(0, b.balance.asset_id) : b.balance;
             op.balance_to_claim = b.id;
+            op.balance_owner_key = keys[b.owner].get_public_key();
             trx.operations.push_back(op);
             required_addrs.insert(b.owner);
          }
diff --git a/tests/tests/operation_tests.cpp b/tests/tests/operation_tests.cpp
index ba098858..e32e35bd 100644
--- a/tests/tests/operation_tests.cpp
+++ b/tests/tests/operation_tests.cpp
@@ -1107,7 +1107,6 @@ BOOST_AUTO_TEST_CASE( witness_withdraw_pay_test )
 
    const asset_object* core = &asset_id_type()(db);
    const account_object* nathan = &get_account("nathan");
-   idump((*nathan));
    enable_fees(105000000);
    BOOST_CHECK_GT(db.current_fee_schedule().membership_lifetime_fee, 0);
    // Based on the size of the reserve fund later in the test, the witness budget will be set to this value