From 21168a4965c5a0d444c4531e9f9e7a65ce734741 Mon Sep 17 00:00:00 2001
From: Hans Moog <3293976+hmoog@users.noreply.github.com>
Date: Tue, 3 Oct 2023 19:02:29 +0200
Subject: [PATCH] Refactor: reverted more code

---
 pkg/protocol/engine/mempool/tests/transaction.go | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/pkg/protocol/engine/mempool/tests/transaction.go b/pkg/protocol/engine/mempool/tests/transaction.go
index d1fda2aa6..3e0aae669 100644
--- a/pkg/protocol/engine/mempool/tests/transaction.go
+++ b/pkg/protocol/engine/mempool/tests/transaction.go
@@ -21,7 +21,7 @@ func (s *SignedTransaction) String() string {
 
 type Transaction struct {
 	id                 iotago.TransactionID
-	utxoInputs         []iotago.Input
+	inputs             []mempool.StateReference
 	outputCount        uint16
 	invalidTransaction bool
 }
@@ -36,7 +36,7 @@ func NewSignedTransaction(transaction mempool.Transaction) *SignedTransaction {
 func NewTransaction(outputCount uint16, inputs ...mempool.StateReference) *Transaction {
 	return &Transaction{
 		id:          tpkg.RandTransactionID(),
-		utxoInputs:  inputs,
+		inputs:      inputs,
 		outputCount: outputCount,
 	}
 }
@@ -46,15 +46,7 @@ func (t *Transaction) ID() (iotago.TransactionID, error) {
 }
 
 func (t *Transaction) Inputs() ([]mempool.StateReference, error) {
-	return append(t.utxoInputs, t.contextInputs...), nil
-}
-
-func (t *Transaction) UTXOInputs() ([]iotago.Input, error) {
-	return t.utxoInputs, nil
-}
-
-func (t *Transaction) ContextInputs() ([]iotago.Input, error) {
-	return t.contextInputs, nil
+	return t.inputs, nil
 }
 
 func (t *Transaction) String() string {