From 3e3408e738b3fa42d7054b2d7d49b3421fa04644 Mon Sep 17 00:00:00 2001
From: Travis Briggs <audiodude@gmail.com>
Date: Thu, 2 Nov 2023 17:07:03 -0700
Subject: [PATCH] Add tests for existing handler methods

---
 rainfall/main.py      |  1 +
 rainfall/main_test.py | 70 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 rainfall/main_test.py

diff --git a/rainfall/main.py b/rainfall/main.py
index f59e044..1687bf9 100644
--- a/rainfall/main.py
+++ b/rainfall/main.py
@@ -27,6 +27,7 @@ def index():
 
   @app.route('/api/v1/user')
   def get_user():
+    print(flask.session)
     user_id = flask.session.get('user_id')
     if user_id is None:
       return flask.jsonify(status=404, error='No signed in user'), 404
diff --git a/rainfall/main_test.py b/rainfall/main_test.py
new file mode 100644
index 0000000..1bc1c6c
--- /dev/null
+++ b/rainfall/main_test.py
@@ -0,0 +1,70 @@
+from unittest.mock import patch
+
+import flask
+import pytest
+import uuid
+
+from rainfall.db import db
+from rainfall.models.user import User
+
+BASIC_USER_ID = uuid.UUID('06543f11-12b6-71ea-8000-e026c63c22e2')
+
+
+@pytest.fixture
+def basic_user(app):
+  with app.app_context():
+    basic_user = User(id=BASIC_USER_ID,
+                      google_id='1234',
+                      name='Jane Doe',
+                      email='janedoe@email.fake',
+                      picture_url='https://pictures.fake/1234')
+    db.session.add(basic_user)
+    db.session.commit()
+
+  return basic_user
+
+
+class MainTest:
+
+  def test_get_user(self, app, basic_user):
+    with app.test_client() as client:
+      with client.session_transaction() as sess:
+        sess['user_id'] = BASIC_USER_ID
+
+      rv = client.get('/api/v1/user')
+      assert rv.json == {
+          'id': str(BASIC_USER_ID),
+          'google_id': '1234',
+          'name': 'Jane Doe',
+          'email': 'janedoe@email.fake',
+          'picture_url': 'https://pictures.fake/1234',
+      }
+
+  def test_get_user_404(self, app):
+    with app.test_client() as client:
+
+      rv = client.get('/api/v1/user')
+      assert rv.status == '404 NOT FOUND'
+      assert rv.json == {'error': 'No signed in user', 'status': 404}
+
+  @patch('rainfall.main.check_csrf', return_value=None)
+  @patch('rainfall.main.id_token.verify_oauth2_token')
+  def test_login(self, mock_verify, mock_check_csrf, app):
+    mock_verify.return_value = {
+        'sub': '1234',
+        'name': 'Jane Doe',
+        'email': 'janedoe@email.fake',
+        'picture': 'https://pictures.fake/photo-1234'
+    }
+
+    with app.test_client() as client:
+      rv = client.post('/api/v1/login')
+      assert rv.status == '302 FOUND'
+      assert rv.headers['location'] == 'http://localhost:5173/welcome'
+
+      user_id = flask.session['user_id']
+      user = db.session.get(User, user_id)
+      assert user.google_id == '1234'
+      assert user.name == 'Jane Doe'
+      assert user.email == 'janedoe@email.fake'
+      assert user.picture_url == 'https://pictures.fake/photo-1234'