|
28 | 28 | QgsTransactionGroup,
|
29 | 29 | QgsField,
|
30 | 30 | QgsFieldConstraints,
|
| 31 | + QgsDataProvider, |
31 | 32 | NULL
|
32 | 33 | )
|
33 | 34 | from qgis.gui import QgsEditorWidgetRegistry
|
@@ -84,9 +85,17 @@ def partiallyCompiledFilters(self):
|
84 | 85 |
|
85 | 86 | # HERE GO THE PROVIDER SPECIFIC TESTS
|
86 | 87 | def testDefaultValue(self):
|
87 |
| - self.assertEqual(self.provider.defaultValue(0), 'nextval(\'qgis_test."someData_pk_seq"\'::regclass)') |
| 88 | + self.provider.setProviderProperty(QgsDataProvider.EvaluateDefaultValues, True) |
| 89 | + self.assertTrue(isinstance(self.provider.defaultValue(0), int)) |
88 | 90 | self.assertEqual(self.provider.defaultValue(1), NULL)
|
89 |
| - self.assertEqual(self.provider.defaultValue(2), '\'qgis\'::text') |
| 91 | + self.assertEqual(self.provider.defaultValue(2), 'qgis') |
| 92 | + self.provider.setProviderProperty(QgsDataProvider.EvaluateDefaultValues, False) |
| 93 | + |
| 94 | + def testDefaultValueClause(self): |
| 95 | + self.provider.setProviderProperty(QgsDataProvider.EvaluateDefaultValues, False) |
| 96 | + self.assertEqual(self.provider.defaultValueClause(0), 'nextval(\'qgis_test."someData_pk_seq"\'::regclass)') |
| 97 | + self.assertFalse(self.provider.defaultValueClause(1)) |
| 98 | + self.assertEqual(self.provider.defaultValueClause(2), '\'qgis\'::text') |
90 | 99 |
|
91 | 100 | def testDateTimeTypes(self):
|
92 | 101 | vl = QgsVectorLayer('%s table="qgis_test"."date_times" sql=' % (self.dbconn), "testdatetimes", "postgres")
|
@@ -248,7 +257,7 @@ def testPktMapInsert(self):
|
248 | 257 | vl = QgsVectorLayer('{} table="qgis_test"."{}" key="obj_id" sql='.format(self.dbconn, 'oid_serial_table'), "oid_serial", "postgres")
|
249 | 258 | self.assertTrue(vl.isValid())
|
250 | 259 | f = QgsFeature(vl.fields())
|
251 |
| - f['obj_id'] = vl.dataProvider().defaultValue(0) |
| 260 | + f['obj_id'] = vl.dataProvider().defaultValueClause(0) |
252 | 261 | f['name'] = 'Test'
|
253 | 262 | r, f = vl.dataProvider().addFeatures([f])
|
254 | 263 | self.assertTrue(r)
|
@@ -553,17 +562,17 @@ def testKey(lyr, key, kfnames):
|
553 | 562 | oflds = olyr.fields()
|
554 | 563 | if key is None:
|
555 | 564 | # if the pkey was not given, it will create a pkey
|
556 |
| - self.assertEquals(oflds.size(), flds.size() + 1) |
557 |
| - self.assertEquals(oflds[0].name(), kfnames[0]) |
| 565 | + self.assertEqual(oflds.size(), flds.size() + 1) |
| 566 | + self.assertEqual(oflds[0].name(), kfnames[0]) |
558 | 567 | for i in range(flds.size()):
|
559 | 568 | self.assertEqual(oflds[i + 1].name(), flds[i].name())
|
560 | 569 | else:
|
561 | 570 | # pkey was given, no extra field generated
|
562 |
| - self.assertEquals(oflds.size(), flds.size()) |
| 571 | + self.assertEqual(oflds.size(), flds.size()) |
563 | 572 | for i in range(oflds.size()):
|
564 | 573 | self.assertEqual(oflds[i].name(), flds[i].name())
|
565 | 574 | pks = olyr.pkAttributeList()
|
566 |
| - self.assertEquals(len(pks), len(kfnames)) |
| 575 | + self.assertEqual(len(pks), len(kfnames)) |
567 | 576 | for i in range(0, len(kfnames)):
|
568 | 577 | self.assertEqual(oflds[pks[i]].name(), kfnames[i])
|
569 | 578 |
|
|
0 commit comments