guangtouwangba/chatgpt-api-proxy

chatgpt-api-proxy's Issues

Sweep: Add test for findOne function in repository, ensure all package are imported correctly including internal pkg

Checklist

• tests/repository/usage_repository_test.go

• Import the necessary packages including the Go's built-in testing package, the go-sqlmock package for mocking the database, and the GormOpenAIUsageRepository struct from the internal/db/repository package.
• Create a setup function that initializes a mock database and a GormOpenAIUsageRepository instance.
• Write a test function that checks if the findOne function returns the correct record when given a valid openAIID and identityID. Use the mock database to simulate this scenario.
• Write a test function that checks if the findOne function returns an error when given an openAIID and identityID that do not exist in the database. Use the mock database to simulate this scenario.
• Write a test function that checks if the findOne function returns an error when there is a problem with the database connection. Use the mock database to simulate this scenario.