add some tests.

Author: sspickle

tests/conftest.py

@@ -0,0 +1,32 @@
+
+import pytest
+from unittest.mock import MagicMock, patch
+from main import app as flask_app
+
+@pytest.fixture
+def app():
+    """Create and configure a new app instance for each test."""
+    # Overwrite the GRL value to True so we can bypass some of the checks
+    # that are not relevant for testing.
+    with patch('ide.auth.GRL', True):
+        flask_app.config.update({
+            "TESTING": True,
+        })
+        yield flask_app
+
+@pytest.fixture
+def client(app):
+    """A test client for the app."""
+    return app.test_client()
+
+@pytest.fixture(autouse=True)
+def mock_google_oauth(mocker):
+    """Mock Google OAuth2."""
+    mocker.patch('ide.auth.is_logged_in', return_value=True)
+    mocker.patch('ide.auth.get_user_info', return_value={'email': 'test@example.com'})
+
+@pytest.fixture(autouse=True)
+def mock_datastore(mocker):
+    """Mock Google Cloud Datastore."""
+    mocker.patch('google.cloud.ndb.Client', return_value=MagicMock())
+    mocker.patch('google.cloud.ndb.Key', return_value=MagicMock())

tests/test_api.py

@@ -0,0 +1,153 @@
+
+import pytest
+from unittest.mock import patch, MagicMock
+
+def test_api_login_not_logged_in(client):
+    with patch('ide.auth.is_logged_in', return_value=False):
+        response = client.get('/api/login')
+        assert response.status_code == 200
+        assert response.json == {'state': 'not_logged_in', 'login_url': '/google/login'}
+
+def test_api_login_logged_in_existing_user(client):
+    with patch('ide.models.User.query') as mock_query:
+        mock_user = MagicMock()
+        mock_user.key.id.return_value = 'testuser'
+        mock_user.secret = 'secret'
+        mock_query.return_value.filter.return_value.get.return_value = mock_user
+        response = client.get('/api/login')
+        assert response.status_code == 200
+        assert response.json == {'state': 'logged_in', 'username': 'testuser', 'secret': 'secret', 'logout_url': '/google/logout'}
+
+def test_api_login_logged_in_new_user(client):
+    with patch('ide.models.User.query') as mock_query:
+        mock_query.return_value.filter.return_value.get.return_value = None
+        response = client.get('/api/login')
+        assert response.status_code == 200
+        assert response.json == {'state': 'new_user', 'suggested_name': 'test'}
+
+def test_api_user_get_existing(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse:
+        mock_parse.return_value = (['testuser'], MagicMock(), 'test@example.com')
+        response = client.get('/api/user/testuser')
+        assert response.status_code == 200
+        assert response.json == {}
+
+def test_api_user_get_non_existing(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse:
+        mock_parse.return_value = (['testuser'], None, 'test@example.com')
+        response = client.get('/api/user/testuser')
+        assert response.status_code == 404
+
+def test_api_user_put_new(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.User') as mock_user_model, \
+         patch('ide.routes.Folder') as mock_folder_model:
+        mock_parse.return_value = (['testuser'], None, 'test@example.com')
+        response = client.put('/api/user/testuser')
+        assert response.status_code == 200
+        mock_user_model.assert_called_once()
+        assert mock_folder_model.call_count == 2
+
+
+def test_api_user_put_existing(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse:
+        mock_parse.return_value = (['testuser'], MagicMock(), 'test@example.com')
+        response = client.put('/api/user/testuser')
+        assert response.status_code == 403
+
+def test_api_user_folders_get(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.Folder.query') as mock_query:
+        mock_parse.return_value = (['testuser'], MagicMock(), 'test@example.com')
+        mock_folder = MagicMock()
+        mock_folder.key.id.return_value = 'MyPrograms'
+        mock_folder.isPublic = True
+        mock_query.return_value = [mock_folder]
+        response = client.get('/api/user/testuser/folder/')
+        assert response.status_code == 200
+        assert response.json == {'user': 'testuser', 'folders': ['MyPrograms'], 'publics': [True]}
+
+def test_api_user_folder_put(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.authorize_user', return_value=True), \
+         patch('ide.routes.Folder') as mock_folder_model:
+        mock_parse.return_value = (['testuser', 'MyNewFolder'], MagicMock(), 'test@example.com')
+        response = client.put('/api/user/testuser/folder/MyNewFolder')
+        assert response.status_code == 200
+        mock_folder_model.assert_called_once()
+
+def test_api_user_folder_delete_non_empty(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.ndb.Key') as mock_key, \
+         patch('ide.routes.Program.query') as mock_program_query:
+        mock_parse.return_value = (['testuser', 'MyPrograms'], MagicMock(), 'test@example.com')
+        mock_key.return_value.get.return_value = MagicMock()
+        mock_program_query.return_value.__iter__.return_value = [MagicMock()]
+        response = client.delete('/api/user/testuser/folder/MyPrograms')
+        assert response.status_code == 409
+
+def test_api_user_folder_delete_empty(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.ndb.Key') as mock_key, \
+         patch('ide.routes.Program.query') as mock_program_query:
+        mock_parse.return_value = (['testuser', 'MyPrograms'], MagicMock(), 'test@example.com')
+        mock_folder = MagicMock()
+        mock_key.return_value.get.return_value = mock_folder
+        mock_program_query.return_value.__iter__.return_value = []
+        response = client.delete('/api/user/testuser/folder/MyPrograms')
+        assert response.status_code == 200
+        mock_folder.key.delete.assert_called_once()
+
+def test_api_user_folder_programs_get(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.ndb.Key') as mock_key, \
+         patch('ide.routes.Program.query') as mock_program_query:
+        mock_parse.return_value = (['testuser', 'MyPrograms'], MagicMock(), 'test@example.com')
+        mock_folder = MagicMock()
+        mock_folder.isPublic = True
+        mock_key.return_value.get.return_value = mock_folder
+        mock_program = MagicMock()
+        mock_program.key.id.return_value = 'test_program'
+        mock_program.screenshot = b''
+        mock_program.datetime = '2025-07-10'
+        mock_program_query.return_value = [mock_program]
+        response = client.get('/api/user/testuser/folder/MyPrograms/program/')
+        assert response.status_code == 200
+        assert response.json['programs'][0]['name'] == 'test_program'
+
+def test_api_user_folder_program_get(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.ndb.Key') as mock_key:
+        mock_parse.return_value = (['testuser', 'MyPrograms', 'test_program'], MagicMock(), 'test@example.com')
+        mock_folder = MagicMock()
+        mock_folder.isPublic = True
+        mock_program = MagicMock()
+        mock_program.source = 'print("Hello, World!")'
+        mock_program.screenshot = b''
+        mock_program.datetime = '2025-07-10'
+        mock_key.return_value.get.side_effect = [mock_folder, mock_program]
+        response = client.get('/api/user/testuser/folder/MyPrograms/program/test_program')
+        assert response.status_code == 200
+        assert response.json['source'] == 'print("Hello, World!")'
+
+def test_api_user_folder_program_put(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.authorize_user', return_value=True), \
+         patch('ide.routes.ndb.Key') as mock_key, \
+         patch('ide.routes.Program') as mock_program_model:
+        mock_parse.return_value = (['testuser', 'MyPrograms', 'test_program'], MagicMock(), 'test@example.com')
+        mock_key.return_value.get.side_effect = [None, MagicMock()]
+        response = client.put('/api/user/testuser/folder/MyPrograms/program/test_program', data={'program': '{}'})
+        assert response.status_code == 200
+        mock_program_model.assert_called_once()
+
+def test_api_user_folder_program_delete(client):
+    with patch('ide.routes.parseUrlPath') as mock_parse, \
+         patch('ide.routes.authorize_user', return_value=True), \
+         patch('ide.routes.ndb.Key') as mock_key:
+        mock_parse.return_value = (['testuser', 'MyPrograms', 'test_program'], MagicMock(), 'test@example.com')
+        mock_program = MagicMock()
+        mock_key.return_value.get.return_value = mock_program
+        response = client.delete('/api/user/testuser/folder/MyPrograms/program/test_program')
+        assert response.status_code == 200
+        mock_program.key.delete.assert_called_once()