@@ -23,17 +23,31 @@ def test_version_endpoint_returns_json(self, client):
2323 assert response .status_code == 200
2424 assert response .content_type == 'application/json'
2525
26+ def test_version_endpoint_returns_array (self , client ):
27+ """Test that /version returns array for Grafana Infinity datasource."""
28+ response = client .get ('/version' )
29+ data = json .loads (response .data )
30+ assert isinstance (data , list )
31+ assert len (data ) == 1
32+
2633 def test_version_endpoint_contains_version_field (self , client ):
2734 """Test that /version response contains version field."""
2835 response = client .get ('/version' )
2936 data = json .loads (response .data )
30- assert 'version' in data
37+ assert 'version' in data [ 0 ]
3138
3239 def test_version_endpoint_contains_build_ts_field (self , client ):
3340 """Test that /version response contains build_ts field."""
3441 response = client .get ('/version' )
3542 data = json .loads (response .data )
36- assert 'build_ts' in data
43+ assert 'build_ts' in data [0 ]
44+
45+ def test_version_endpoint_contains_display_field (self , client ):
46+ """Test that /version response contains pre-formatted display field."""
47+ response = client .get ('/version' )
48+ data = json .loads (response .data )
49+ assert 'display' in data [0 ]
50+ assert 'PostgresAI v' in data [0 ]['display' ]
3751
3852
3953class TestReadVersionFile :
0 commit comments