From 13f7b783f29d3d3f9c3fce30c28db6c38152337c Mon Sep 17 00:00:00 2001
From: Mike Naberezny <mike@naberezny.com>
Date: Sat, 4 Apr 2009 23:10:22 -0700
Subject: [PATCH] Added tests for ObservableMemory.

---
 src/py65/tests/test_memory.py | 151 ++++++++++++++++++++++++++++++++++
 1 file changed, 151 insertions(+)
 create mode 100644 src/py65/tests/test_memory.py

diff --git a/src/py65/tests/test_memory.py b/src/py65/tests/test_memory.py
new file mode 100644
index 0000000..5eabca9
--- /dev/null
+++ b/src/py65/tests/test_memory.py
@@ -0,0 +1,151 @@
+import unittest
+import sys
+import re
+import os
+from py65.memory import ObservableMemory
+
+class ObservableMemoryTests(unittest.TestCase):
+
+    # __setitem__
+
+    def test___setitem__with_no_listeners_changes_memory(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+ 
+        mem[0xC000] = 0xAB
+        self.assertEqual(0xAB, subject[0xC000])
+ 
+    def test___setitem__ignores_listeners_returning_none(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def listener_1(address, value):
+            return None
+
+        def listener_2(address, value):
+            return None
+        
+        mem.register_listener([0xC000], listener_1)
+        mem.register_listener([0xC000], listener_2)
+        
+        mem[0xC000] = 0xAB
+        self.assertEqual(0xAB, subject[0xC000])
+
+    def test___setitem__uses_result_of_last_listener(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def listener_1(address, value):
+            return 0x01
+
+        def listener_2(address, value):
+            return 0x02
+        
+        mem.register_listener([0xC000], listener_1)
+        mem.register_listener([0xC000], listener_2)
+        
+        mem[0xC000] = 0xAB
+        self.assertEqual(0x02, subject[0xC000])
+
+    # register_listener
+
+    def test_register_listener_covers_all_addresses_in_range(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def listener(address, value):
+            return 0xAB
+        
+        mem.register_listener(range(0xC000, 0xC001+1), listener)
+
+        mem[0xC000] = 0xAB
+        mem[0xC001] = 0xAB
+        self.assertEqual(0xAB, subject[0xC001])
+        self.assertEqual(0xAB, subject[0xC001])
+
+    def test_register_listener_does_not_register_same_listener_twice(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+        
+        try:   
+            self._count = 0
+
+            def listener(address, value):
+                self._count += 1
+        
+            mem.register_listener([0xC000], listener)
+            mem.register_listener([0xC000], listener)
+
+            mem[0xC000] = 0xAB
+            self.assertEqual(1, self._count)
+        finally:
+            delattr(self, '_count')
+   
+    # __getitem__
+    
+    def test___getitem__with_no_providers_changes_memory(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+ 
+        subject[0xC000] = 0xAB
+        self.assertEqual(0xAB, mem[0xC000])
+
+    def test___getitem__ignores_providers_returning_none(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def provider_1(address):
+            return None
+
+        def provider_2(address):
+            return None
+        
+        mem.register_provider([0xC000], provider_1)
+        mem.register_provider([0xC000], provider_2)
+        
+        mem[0xC000] = 0xAB
+        self.assertEqual(0xAB, subject[0xC000])
+
+    def test___getitem__uses_result_of_last_provider(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def provider_1(address):
+            return None
+
+        def provider_2(address):
+            return None
+        
+        mem.register_provider([0xC000], provider_1)
+        mem.register_provider([0xC000], provider_2)
+
+        subject[0xC000] = 0xAB
+        self.assertEqual(0xAB, mem[0xC000])
+
+    # __getattr__
+
+    def test__getattr__proxies_subject(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+        self.assertEqual(subject.count, mem.count)
+
+    # write
+    
+    def test_write_directly_writes_values_to_subject(self):
+        subject = self._make_subject()
+        mem = ObservableMemory(subject=subject)
+
+        def listener(address, value):
+            return 0xFF
+        mem.register_listener([0xC000,0xC001], listener)
+        
+        mem.write(0xC000, [0x01, 002])
+        self.assertEqual(0x01, subject[0xC000])
+        self.assertEqual(0x02, subject[0xC001])
+
+    # Test Helpers
+    
+    def _make_subject(self):    
+        subject = 0x10000 * [0x00] 
+        return subject
+