sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Explicitly set billing to off in the test for being able to index a large set of data

Browse source
This commit is contained in:
sabaimran 2023-11-25 20:48:32 -08:00
parent b2afbaa315
commit 73e38fccf3
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_client.py
View file

@ -128,8 +128,8 @@ def test_regenerate_with_invalid_content_type(client):
# ---------------------------------------------------------------------------------------------------- # ----------------------------------------------------------------------------------------------------
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
def test_index_update_big_files(client): def test_index_update_big_files(client):
state.billing_enabled = True
# Arrange # Arrange
state.billing_enabled = True
files = get_big_size_sample_files_data() files = get_big_size_sample_files_data()
headers = {"Authorization": "Bearer kk-secret"} headers = {"Authorization": "Bearer kk-secret"}
@ -143,6 +143,7 @@ def test_index_update_big_files(client):
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
def test_index_update_big_files_no_billing(client): def test_index_update_big_files_no_billing(client):
# Arrange # Arrange
state.billing_enabled = False
files = get_big_size_sample_files_data() files = get_big_size_sample_files_data()
headers = {"Authorization": "Bearer kk-secret"} headers = {"Authorization": "Bearer kk-secret"}