fix: add --key (or --keys) param for import/export commands

This commit is contained in:
Lance Edgar 2024-12-06 09:32:24 -06:00
parent 15b2cb07ba
commit a73896b75d
3 changed files with 22 additions and 12 deletions

View file

@ -82,8 +82,10 @@ class TestImporter(DataTestCase):
model = self.app.model
imp = self.make_importer(model_class=model.Setting)
self.assertEqual(imp.get_keys(), ['name'])
imp.key = 'value'
self.assertEqual(imp.get_keys(), ['value'])
with patch.multiple(imp, create=True, key='value'):
self.assertEqual(imp.get_keys(), ['value'])
with patch.multiple(imp, create=True, keys=['foo', 'bar']):
self.assertEqual(imp.get_keys(), ['foo', 'bar'])
def test_process_data(self):
model = self.app.model