def test_get(self): registry = ModelRegistry() t = TestModel() conn = ElasticOSMConnection().connect(settings) registry.register_model(t) t = TestModel() assign_test_values_to_fields(t) t.save() # give it a second to add to the index sleep(1) queried_model = TestModel.get(id=t.id) self.assertEqual(t.id, queried_model.id)