Files
@ e6fc18e4e15a
Branch filter:
Location: rattail-project/rattail/tests/autocomplete/test_base.py
e6fc18e4e15a
6.3 KiB
text/x-python
Add new `RattailConfiguration` class based on python-configuration
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 | # -*- coding: utf-8; -*-
from __future__ import unicode_literals, absolute_import
from unittest import TestCase
import sqlalchemy as sa
from rattail.autocomplete import base as mod
from rattail.config import make_config
from rattail.db import Session
class TestAutocompleter(TestCase):
def setUp(self):
self.config = self.make_config()
def make_config(self):
return make_config([], extend=False)
def test_constructor(self):
# cannot instantiate autocompleter with no key
class BadAutocompleter(mod.Autocompleter):
pass
self.assertRaises(NotImplementedError, BadAutocompleter, self.config)
# but works okay if key is defined
class GoodAutocompleter(mod.Autocompleter):
autocompleter_key = 'stores'
autocompleter = GoodAutocompleter(self.config)
self.assertIsNotNone(autocompleter)
def test_get_model_class(self):
model = self.config.get_model()
# no model class by default; hence error
class BadAutocompleter(mod.Autocompleter):
autocompleter_key = 'stores'
autocompleter = BadAutocompleter(self.config)
self.assertRaises(AttributeError, autocompleter.get_model_class)
# but if one is set, it is returned
class GoodAutocompleter(mod.Autocompleter):
autocompleter_key = 'stores'
model_class = model.Store
autocompleter = GoodAutocompleter(self.config)
result = autocompleter.get_model_class()
self.assertIs(result, model.Store)
def test_autocomplete_fieldname(self):
# no fieldname by default; hence error
class BadAutocompleter(mod.Autocompleter):
autocompleter_key = 'stores'
autocompleter = BadAutocompleter(self.config)
self.assertRaises(NotImplementedError, getattr,
autocompleter, 'autocomplete_fieldname')
# but works okay if class defines it
class GoodAutocompleter(mod.Autocompleter):
autocompleter_key = 'stores'
autocomplete_fieldname = 'name'
autocompleter = GoodAutocompleter(self.config)
self.assertEqual(autocompleter.autocomplete_fieldname, 'name')
def test_autocomplete(self):
engine = sa.create_engine('sqlite://')
model = self.config.get_model()
model.Base.metadata.create_all(bind=engine)
session = Session(bind=engine)
class StoreAutocompleter(mod.Autocompleter):
autocompleter_key = 'store'
model_class = model.Store
autocomplete_fieldname = 'name'
# first create a few stores
store001 = model.Store(id='001', name="Acme Store #1")
session.add(store001)
store002 = model.Store(id='002', name="Acme Store #2")
session.add(store002)
store999 = model.Store(id='999', name="Warehouse")
session.add(store999)
# and our autocompleter
autocompleter = StoreAutocompleter(self.config)
# searching for nothing yields no results
result = autocompleter.autocomplete(session, '')
self.assertEqual(len(result), 0)
# search for 'acme' yields stores 1, 2
result = autocompleter.autocomplete(session, 'acme')
self.assertEqual(len(result), 2)
uuids = [info['value'] for info in result]
self.assertIn(store001.uuid, uuids)
self.assertIn(store002.uuid, uuids)
# search for 'warehouse' yields store 999
result = autocompleter.autocomplete(session, 'warehouse')
self.assertEqual(len(result), 1)
self.assertEqual(result[0]['value'], store999.uuid)
# search for 'bogus' yields no results
result = autocompleter.autocomplete(session, 'bogus')
self.assertEqual(len(result), 0)
# search for 's' yields all 3 stores
result = autocompleter.autocomplete(session, 's')
self.assertEqual(len(result), 3)
# unless we cap the max results
autocompleter.max_results = 2
result = autocompleter.autocomplete(session, 's')
self.assertEqual(len(result), 2)
# removing cap should get all 3 again
autocompleter.max_results = None
result = autocompleter.autocomplete(session, 's')
self.assertEqual(len(result), 3)
class TestPhoneMagicMixin(TestCase):
def setUp(self):
self.config = self.make_config()
def make_config(self):
return make_config([], extend=False)
def test_autocomplete(self):
engine = sa.create_engine('sqlite://')
model = self.config.get_model()
model.Base.metadata.create_all(bind=engine)
session = Session(bind=engine)
class StoreMagicAutocompleter(mod.PhoneMagicMixin, mod.Autocompleter):
autocompleter_key = 'store'
model_class = model.Store
autocomplete_fieldname = 'name'
phone_model_class = model.StorePhoneNumber
# first create a few stores, with phones
store001 = model.Store(id='001', name="Acme Store #1")
store001.add_phone_number('417-555-0001')
session.add(store001)
store002 = model.Store(id='002', name="Acme Store #2")
store002.add_phone_number('417-555-0002')
session.add(store002)
store999 = model.Store(id='999', name="Warehouse")
session.add(store999)
# and our autocompleter
autocompleter = StoreMagicAutocompleter(self.config)
# searching for nothing yields no results
result = autocompleter.autocomplete(session, '')
self.assertEqual(len(result), 0)
# search for 'acme' yields stores 1, 2
result = autocompleter.autocomplete(session, 'acme')
self.assertEqual(len(result), 2)
uuids = [info['value'] for info in result]
self.assertIn(store001.uuid, uuids)
self.assertIn(store002.uuid, uuids)
# search for 'warehouse' yields store 999
result = autocompleter.autocomplete(session, 'warehouse')
self.assertEqual(len(result), 1)
self.assertEqual(result[0]['value'], store999.uuid)
# TODO: need to test the actual phone number magic, but sqlite
# does not have a regexp_replace() function built-in. so
# either need to code some workaround, or require the 're'
# extension be loaded for sqlite, etc.
|