def setUp(self): class DumpContext(BaseContext): def read(self): return [{'username': '******', 'content': 'hello world'}, {'username': '******', 'content': 'hello world2'} ] def load_more(self): return [{'username': '******', 'content': 'next1'}, {'username': '******', 'content': 'next2'} ] def fetch(self): return [{'username': '******', 'content': 'update1'}, {'username': '******', 'content': 'update2'}, {'username': '******', 'content': 'update2'} ] def reload(self): return [{'username': '******', 'content': 'reload 1'}, {'username': '******', 'content': 'reload 2'}, {'username': '******', 'content': 'reload 3'} ] self.app = Application() context = DumpContext('DumpContext') self.buffer = TweetsBuffer(self.app, context)
class TestTweetsBuffer(object): """docstring for TestTweetsBuffer""" def setUp(self): class DumpContext(BaseContext): def read(self): return [{'username': '******', 'content': 'hello world'}, {'username': '******', 'content': 'hello world2'} ] def load_more(self): return [{'username': '******', 'content': 'next1'}, {'username': '******', 'content': 'next2'} ] def fetch(self): return [{'username': '******', 'content': 'update1'}, {'username': '******', 'content': 'update2'}, {'username': '******', 'content': 'update2'} ] def reload(self): return [{'username': '******', 'content': 'reload 1'}, {'username': '******', 'content': 'reload 2'}, {'username': '******', 'content': 'reload 3'} ] self.app = Application() context = DumpContext('DumpContext') self.buffer = TweetsBuffer(self.app, context) def test_app_ref(self): assert isinstance(self.buffer.app, Application) def test_tweets_buffer_read(self): assert len(self.buffer.adapter.elements) == 2 def test_tweets_buffer_reload(self): self.buffer.reload() assert len(self.buffer.adapter.elements) == 2 def test_tweets_buffer_next(self): self.buffer.load_more() print(self.buffer.adapter.text) assert len(self.buffer.adapter.elements) == 4 def test_tweets_buffer_update(self): self.buffer.fetch() assert len(self.buffer.adapter.elements) == 5 def test_tweets_buffer_next_update(self): self.buffer.load_more() self.buffer.fetch() print(self.buffer.text) print(len(self.buffer.adapter.elements)) assert len(self.buffer.adapter.elements) == 7 def test_current_element(self): print(self.buffer.buffer.document.cursor_position_row) self.buffer.buffer.cursor_down(8) print(self.buffer.buffer.document.cursor_position_row) self.buffer.load_more() self.buffer.load_more() print(self.buffer.text) self.buffer.buffer.cursor_down(8) index = self.buffer.current_element() print(index) assert 1 == 2