1
=== modified file 'desktopcouch/records/couchwidget.py'
2
--- desktopcouch/records/couchwidget.py	2009-08-18 23:05:18 +0000
3
+++ desktopcouch/records/couchwidget.py	2009-08-19 23:01:09 +0000
4
@@ -67,7 +67,7 @@
5
67
    def headings(self, headings):
67
    def headings(self, headings):
6
68
        self.__headings = headings
68
        self.__headings = headings
7
69
        if self.record_type != None:
69
        if self.record_type != None:
9
70
            self.__populate_treeview()(self.record_type)
70
            self.__populate_treeview()
10
71
71
11
72
    @property
72
    @property
12
73
    def editable(self):
73
    def editable(self):
13
@@ -119,7 +119,7 @@
14
119
119
15
120
    def __populate_treeview(self):
120
    def __populate_treeview(self):
16
121
        #if the database is not set up, just return
121
        #if the database is not set up, just return
18
122
        if self.__db == None:
122
        if self.__db == None or self.__headings == None:
19
123
            return
123
            return
20
124
124
21
125
        #refresh the model structure
125
        #refresh the model structure
22
126
126
23
=== added file 'desktopcouch/records/tests/test_couchwidget.py'
24
--- desktopcouch/records/tests/test_couchwidget.py	1970-01-01 00:00:00 +0000
25
+++ desktopcouch/records/tests/test_couchwidget.py	2009-08-19 23:01:09 +0000
26
@@ -0,0 +1,94 @@
27
1
# Copyright 2009 Canonical Ltd.
28
2
#
29
3
# This file is part of desktopcouch.
30
4
#
31
5
#  desktopcouch is free software: you can redistribute it and/or modify
32
6
# it under the terms of the GNU Lesser General Public License version 3
33
7
# as published by the Free Software Foundation.
34
8
#
35
9
# desktopcouch is distributed in the hope that it will be useful,
36
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
37
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
38
12
# GNU Lesser General Public License for more details.
39
13
#
40
14
# You should have received a copy of the GNU Lesser General Public License
41
15
# along with desktopcouch.  If not, see <http://www.gnu.org/licenses/>.
42
16
#
43
17
# Authors: Rick Spencer <rick.spencer@canonical.com>
44
18
45
19
"""Tests for the couchwidget object"""
46
20
47
21
from testtools import TestCase
48
22
from desktopcouch.records.server import CouchDatabase
49
23
from desktopcouch.records.couchwidget import CouchWidget
50
24
51
25
class TestCouchWidget(TestCase):
52
26
    """Test the CouchWidget functionality"""
53
27
    def setUp(self):
54
28
        self.dbname = "couch_widget_test"
55
29
        self.record_type = "test_record_type"
56
30
57
31
    def test_headings_first(self):
58
32
        #create a test widget with test database values
59
33
        cw = CouchWidget()
60
34
        cw.database = self.dbname
61
35
62
36
        #allow editing
63
37
        cw.editable = True
64
38
65
39
        #create headers/keys
66
40
        cw.headings = ["Key1","Key2","Key3","Key4"]
67
41
68
42
        #set the record_type for the TreeView
69
43
        #it will not populate without this value being set
70
44
        cw.record_type = self.record_type
71
45
72
46
        #create a row with all four columns set
73
47
        cw.append_row(["val1","val2","val3","val4"])
74
48
75
49
        #create a row with only the second column set
76
50
        cw.append_row(["","val2"])
77
51
78
52
        #create an empty row (which will not be saved until the user edits it)
79
53
        cw.append_row([])
80
54
81
55
        self.assertEqual(cw.record_type, self.record_type)
82
56
        self.__delete_db()
83
57
84
58
    def test_record_type_first(self):
85
59
        #create a test widget with test database values
86
60
        cw = CouchWidget()
87
61
        cw.database = self.dbname
88
62
89
63
        #allow editing
90
64
        cw.editable = True
91
65
92
66
        #set the record_type for the TreeView
93
67
        #it will not populate without this value being set
94
68
        cw.record_type = self.record_type
95
69
96
70
        #create headers/keys
97
71
        cw.headings = ["Key1","Key2","Key3","Key4"]
98
72
99
73
        #create a row with all four columns set
100
74
        cw.append_row(["val1","val2","val3","val4"])
101
75
102
76
        #create a row with only the second column set
103
77
        cw.append_row(["","val2"])
104
78
105
79
        #create an empty row (which will not be saved until the user edits it)
106
80
        cw.append_row([])
107
81
108
82
        self.assertEqual(cw.record_type, self.record_type)
109
83
        self.__delete_db()        
110
84
111
85
    def tearDown(self):
112
86
        """tear down each test"""
113
87
        self.__delete_db()
114
88
        #delete the database
115
89
116
90
    def __delete_db(self):
117
91
        db = CouchDatabase(self.dbname, create=True)
118
92
        del db._server[self.dbname]
119
93
120
94
Reviewer	Review Type	Date Requested	Status
Elliot Murphy (community)		2009-08-19	Approve on 2009-08-20
Review via email: mp+10432@code.qastaging.launchpad.net