Use new NoABIFound error

Author: kclowes

Diff for: tests/core/contracts/test_contract_call_interface.py

@@ -24,6 +24,7 @@
     BlockNumberOutofRange,
     InvalidAddress,
     MismatchedABI,
+    NoABIFound,
     NoABIFunctionsFound,
     ValidationError,
 )
@@ -527,7 +528,7 @@ def test_function_multiple_possible_encodings(web3):
 
 def test_function_no_abi(web3):
     contract = web3.eth.contract()
-    with pytest.raises(NoABIFunctionsFound):
+    with pytest.raises(NoABIFound):
         contract.functions.thisFunctionDoesNotExist().call()
 
 

Diff for: web3/contract.py

@@ -106,18 +106,21 @@ class ContractFunctions:
     """
 
     def __init__(self, abi, web3, address=None):
-        if abi:
-            self.abi = abi
+        self.abi = abi
+        self.web3 = web3
+        self.address = address
+
+        if self.abi:
             self._functions = filter_by_type('function', self.abi)
             for func in self._functions:
                 setattr(
                     self,
                     func['name'],
                     ContractFunction.factory(
                         func['name'],
-                        web3=web3,
+                        web3=self.web3,
                         contract_abi=self.abi,
-                        address=address,
+                        address=self.address,
                         function_identifier=func['name']))
 
     def __iter__(self):
@@ -128,6 +131,10 @@ def __iter__(self):
             yield func['name']
 
     def __getattr__(self, function_name):
+        if self.abi is None:
+            raise NoABIFound(
+                "There is no ABI found for this contract.",
+            )
         if '_functions' not in self.__dict__:
             raise NoABIFunctionsFound(
                 "The abi for this contract contains no function definitions. ",
@@ -1180,7 +1187,7 @@ def __init__(self,
         self.abi = abi
         self._functions = None
 
-        if abi:
+        if self.abi:
             if transaction is None:
                 transaction = {}