added more network package coverage

StorjOld · Sep 30, 2015 · b29e672 · b29e672
1 parent 87807cb
commit b29e672
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/network/test_package.py b/tests/network/test_package.py
@@ -29,6 +29,27 @@ def test_parse_normal(self):
                                self.btctxstore.testnet)
         self.assertTrue(parsed != None)
 
+    def test_ignores_nondata_package_with_data(self):
+        package_bytes = package._create(package._TYPE_DATA, self.wif, 
+                                        b"F483", self.btctxstore.testnet)
+        self.assertTrue(package_bytes != None)
+        package_bytes = b'0' + package_bytes[1:]  # hack type
+
+        parsed = package.parse(package_bytes, self.address, 2, 
+                               self.btctxstore.testnet)
+        self.assertEqual(parsed, None)
+
+
+    def test_ignores_dcc_address_mismatch(self):
+        package_bytes = package._create(package._TYPE_DATA, self.wif, 
+                                        b"F483", self.btctxstore.testnet)
+        self.assertTrue(package_bytes != None)
+
+        address = self.btctxstore.get_address(self.btctxstore.create_key())
+        parsed = package.parse(package_bytes, address, 2, 
+                               self.btctxstore.testnet)
+        self.assertEqual(parsed, None)
+
     def test_ignores_stale_package(self):
         package_bytes = package._create(package._TYPE_DATA, self.wif, 
                                         b"F483", self.btctxstore.testnet)