December 22, 2022
The Ultimate Guide to Testing with Prisma: Mocking Prisma Client
As your applications grow, automated testing becomes more and more important. In this article, you will learn how to mock Prisma Client so you can test functions with database interactions without hitting an actual database.
Table Of Contents
- Table Of Contents
- Introduction
- Prerequisites
- What is a mock?
- Set up a Prisma project
- Set up Vitest
- Why mock Prisma Client?
- Mock Prisma Client
- Using the mocked client
- Spy on methods
- Why Vitest?
- Summary & What's next
Introduction
Testing is becoming increasingly important in applications as it allows developers to be more confident in the code they write and iterate on their products more efficiently.
Being able to work confidently and efficiently are, as one might imagine, important aspects of any developer's workflow. So... why doesn't every developer write tests for their applications? The answer to this question often is: Writing tests, especially when a database is involved, can be tricky!
Warning: bad advice 👆🏻In this series, you will learn how to perform different types of tests against various applications that interact with a database.
This article specifically will dive into the topic of mocking and walk through how to mock Prisma Client. Then, you will take a look at what you can do with the mocked client.
Technologies you will use
Prerequisites
Assumed knowledge
The following would be helpful to have coming in to this series:
- Basic knowledge of JavaScript or TypeScript
- Basic knowledge of Prisma Client and its functionalities
Development environment
To follow along with the examples provided, you will be expected to have:
What is a mock?
The first concept you will look at in this series is mocking. This term refers to the practice of creating a controlled replacement for an object that acts similarly to the real object it replaces.
The goal of mocking is typically to allow a developer to replace any external dependencies a function may require so they can effectively write unit tests against that function. This way tests can be isolated to the function's behavior without worrying about the behavior of external modules that aren't directly related.
Note: You will take a closer look at unit tests in the next article of this series.
To illustrate this, consider the following function:
This function does three things:
- Checks to make sure a valid email address is provided
- Throws an error if an invalid address was provided
- Sends an email via an imaginary
mailer
service
To write a test to validate this function behaves as expected, you would likely start by testing the scenario where the function is provided an invalid email address and verifying an error is thrown.
The function, however, relies on two external pieces of code: isValidEmail
and mailer
. Because these are separate pieces of code and technically unrelated to the function you are testing, you would not want to have to worry about whether these imports function properly. Instead, these should be assumed to be functional and tested independently.
You also likely would not want an actual email to be sent during your test when mailer.send()
is called, as that functionality is independent of the function you are testing.
In situations like this, it is common practice to mock those dependencies instead, replacing the real imported object with a "fake" that returns a controlled value. In doing so, you gain the ability to trigger specific states in the test's target function without having to consider the behavior of another module.
This is a fairly basic scenario that illustrates how mocking can be useful, however the rest of this article will dive deeper into the different patterns and tools you can use to mock modules and use those mocks to test specific scenarios.
Set up a Prisma project
Before jumping into writing tests, you will need a project to experiment with. To set one up you will use try-prisma
, a tool that allows you to quickly set up a sample project that with Prisma.
Run the following command in a terminal:
Once that finishes, a starter project should have been set up in your current working directory in a folder named mocking_playground
.
You will also see additional output in your terminal with instructions on next steps. Follow those instructions to enter your project and run your first Prisma migration:
A SQLite database has now been generated, your schema applied, and Prisma Client has been generated. You are ready to begin working in your project!
Set up Vitest
In order to create tests and mocks, you will need a testing framework. In this series, you will use the increasingly popular Vitest testing framework which provides a set of tooling that allows you to build and run tests, as well as create mocks of modules.
Note: Vitest also does a ton of other super cool things! Give their docs a look if you're curious.
Run this command in your project to install the Vitest framework and its CLI tools:
Next, create a new folder in the root of your project named test
where all of your tests will live:
Note: It is not required by Vitest to put your tests in a
/test
folder. Vitest will by default detect test files based on these naming conventions.
Finally, in package.json
, add a new script named test
that simply runs the command vitest
:
You can now use npm run test
to run your tests. You can also run npm t
for short. Currently, your tests will fail because there are no test files.
Create a new file inside of the /test
directory named sample.test.ts
:
Add the following test so that you can verify Vitest is set up correctly:
Now that there is a valid test, running npm t
should result in a success! Vitest is set up and ready to be put to use.
Why mock Prisma Client?
The best way to illustrate why mocking Prisma Client is useful in unit testing is to write a function that uses Prisma Client and write a test for that function that does not use a mocked client.
In the root of your project, create a new folder named libs
. Then create a file within that folder named prisma.ts
:
Add the following snippet to that new file:
The code above instantiates Prisma Client and exports it as a singleton instance. This is the "real" Prisma Client instance.
Now that there is a usable instance of Prisma Client available, write a function that makes use of it.
Replace the contents of script.ts
with the following:
The createUser
function does the following:
- Takes in a
user
argument - Passes
user
along to theprisma.user.create
function - Returns the response, which should be the new user object
Next you will write a test for that new function. This test will ensure the createUser
returns the expected data when provided a valid user: the new user.
Update test/sample.test.ts
so that it matches the snippet below:
Note: The test above is not using a mocked Prisma Client. It is using the real client instance to demonstrate the problem you may run into when testing against a real database.
Assuming your database had not yet contained any users records, this test should pass the first time you run it. There are a few problems though:
- The next time you run this test, the
id
of the created user will not be1
, causing the test to fail. - The
email
field has a@unique
attribute in your Prisma schema, signifying that column has a unique index in the database. This will cause an error to occur on subsequent runs of the test. - This test assumes you are running against a development database and requires a database to be available. Every time you run this test a record will be added to your database.
In situations such as unit testing which focus on a single function, the best practice is to assume your database operations will behave correctly and use a mocked version of your client or driver instead, allowing you to focus on testing the specific behavior of the function you are targeting.
Note: There are scenarios where you may want to test against a database and actually perform operations on it. Integration and end-to-end tests are good example of these cases. These tests may rely on multiple database operations occurring across multiple functions and areas of your application.
Mock Prisma Client
For the reasons outlined in the previous section, it is considered best practice to create a mock of your client to properly unit test your functions that use Prisma Client. This mock will replace the imported module that your function would normally use.
To accomplish this, you will make use of Vitest's mocking tools and an external library named vitest-mock-extended
.
First off, install vitest-mock-extended
in your project:
Next, head over to the test/sample.test.ts
file and make the following changes to let Vitest know it should mock the libs/prisma.ts
module:
The mock
function available in the vi
object lets Vitest know it should mock the module found at a provided file path. There are a few different ways the mock
function can decide how to mock the target module, as described in the documentation.
Currently, Vitest will attempt to mock the module found at '../libs/prisma'
, however it will not be able to automatically mock the "deep", or "nested", properties of the prisma
object. For example, prisma.user.create()
will not be mocked properly as it is a deeply nested property of the Prisma Client instance. This causes the tests to fail as the function will still be run as it normally against the real database.
To solve this problem, you need to let Vitest know how exactly you want that module to be mocked and provide it the value that should be returned when the mocked module is imported, which should include mocked versions of the deeply nested properties.
Create a new folder within the libs
directory named __mocks__
:
The folder name __mocks__
is a common convention in testing frameworks where you may place any manually created mocks of modules. The __mocks__
folder must be directly adjacent to the module you are mocking, which is why we created the folder next to the libs/prisma.ts
file.
Within that new folder, create a file named prisma.ts
:
Notice this file has the same name as the "real" file, prisma.ts
. By following this convention, Vitest will know when it mocks the module via vi.mock
that it should use that file to find the mocked version of the client.
With that structure in place, you will now create the manual mock.
In the new libs/__mocks__/prisma.ts
file, add the following:
The snippet above does the following:
- Imports all of the tools needed to create the mocked client.
- Lets Vitest know that between each individual test the mock should be reset to its original state.
- Creates and exports a "deep mock" of Prisma Client using the
vitest-mock-extended
library'smockDeep
function which ensures all properties of the object, even deeply nested ones, are mocked.
Note: Essentially,
mockDeep
will set every Prisma Client function's value to the Vitest helper function:vi.fn()
.
At this point, if you run your test with npm t
again you should see you no longer receiving the same error as before! But there is still a problem...
This error actually occurs because the mock has been put in place correctly. Your prisma.user.create
invocation in script.ts
is no longer hitting the database. Currently, that function essentially does nothing and returns undefined
.
You need to tell Vitest what prisma.user.create
should do by mocking its behavior. Now that you have a proper mocked version of Prisma Client, this requires a simple change to your test.
In test/sample.test.ts
file, add the following to tell Vitest how that function should behave during the course of that individual test:
Above, the "fake" client was imported as it exports the deep mock of Prisma Client.
On this object you will notice a new set of functions attached to each Prisma Client property and function:
The one used in the snippet above, mockResolvedValue
, replaces the normal prisma.user.create
function with a function that returns the provided value. For the course of that single test, that function will behave as if you performed the following assignment:
Note: Later in this article you will dive in to some of the helpful functions available to your mocked Prisma Client and how you might use them.
You can now run functions that use Prisma Client by mocking the client's behaviors beforehand to ensure a desired outcome. This way, rather than worrying about the individual queries, you can focus on the function's actual business logic.
If you now run your tests again, you should finally see that all of your tests have passed! ✅
Using the mocked client
So you've got a mocked Prisma Client instance and have the ability to manipulate the client to generate the query results you need to test specific scenarios in your functions... what next?
The remainder of this article will dive into many of the functions your mocked client and Vitest have available and how they might be used in different scenarios to enable your testing experience.
Note: The examples below will not be viable, full-blown unit tests. Rather, they will be functional samples of the tools available via your mocked client. The next article in this series will cover unit testing in-depth.
Mocking query responses
One of the most common things you will use your mocked client for is mocking the responses of queries. You already mocked the response of the create
method previously in this article, however there are multiple ways to do this that each have their own use-cases.
Take this scenario, for example:
Note: The usage of
toStrictEqual
here is important. When comparing objects,toStrictEqual
ensures the objects have the same structure and type.
Although this test passes successfully, it doesn't make much sense. When prisma.post.findMany.mockResolvedValue
is invoked, the value provided to that function is used as the response of prisma.post.findMany
for the remainder of the test. More specifically, until the mockReset
function is called in libs/__mocks__/prisma.ts
.
As a result, the unpublished
and published
arrays will contain the exact same values, including the true
value in the published
property.
In order to generate a more realistic response in this scenario, you can make use of another function: mockResolvedValueOnce
. This function can be called multiple times to mock the responses of a function and the responses of subsequent invocations.
In your getPosts
function, you can use mockResolvedValueOnce
to mock the first and second responses that function should return.
Note: Many functions available via Vitest have a
mockXValueOnce
method along withmockXValue
. Refer to the documentation for more details.
Triggering and capturing errors
Another scenario you may want to test for is a case where a query fails and returns or throws an error. A great example of where this may be useful is Prisma Client's findUniqueOrThrow
function.
This function searches for a unique record but throws an error if a record is not found. Because your Prisma Client's functions are mocked, however, the findUniqueOrThrow
function no longer behaves that way. You must manually trigger the errored state. An example of how you might test for this behavior is shown below:
The mockImplementation
allows you to provide a function that replaces the behavior of the mocked function. In the case above, the replacement function simply throws an error.
While this may seem a bit tedious at first glance, the need to manually define the behavior of a function in this case is actually an added benefit. This allows you to have fine-grain control of what the output of your function will be in different states, even errored ones.
Along the same lines as above, if the method you are testing is intended to throw an actual error rather than return some message related to the error, you can also test for that!
By using the rejects
keyword on the response of the expect
function, Vitest knows to resolve the Promise
given to expect
and look for an errored response. Once the Promise
resolves, the toThrow
and toThrowError
functions allow you to check for specific details about the error.
Mocking transactions
Another piece of Prisma Client you may need to mock is a $transaction
.
There are different types of transactions: sequential operations and interactive transactions. The way you mock these will depend greatly on the goal of your test and the context in which you are using the $transaction
. There are, however, two general ways in which you will mock this function.
For both sequential operations and interactive transactions, the result of the completed transaction is eventually returned from the $transaction
function. If your test only cares about the result of the transaction, your test will look very similar to the tests above where you mocked a function's response.
An example might look something like this:
In the test above you:
- Mocked out the data for the post you intended to create.
- Mocked what the response from
$transaction
should look like. - Invoked the function after the Prisma Client methods had been mocked.
- Ensured the returned value from your function matched what you would have expected.
By mocking the response of the $transaction
function itself, you did not have to worry about what went on within the transaction's sequential actions (or the interactive transaction if that were the case).
What if you want to test an interactive transaction that has important business logic you need to validate? This method would not work as it completely forgoes the inner workings of the transaction.
To test an interactive transaction with important business logic, you may write a test that looks like the following:
This test is a little bit more involved, as there are a lot of different moving pieces to consider.
Here is what happens:
- The post and response objects are mocked.
- The responses of the
create
andcount
methods are mocked. - The
$transaction
function's implementation is mocked so that you can provide the mocked Prisma Client to the interactive transaction function rather than the actual client instance. - The
addPost
method is invoked. - The values of the response are validated to ensure the business logic within the interactive transaction worked. More specifically, it ensures the new post's
published
flag is set totrue
.
Spy on methods
The last concept you will explore is spying. Vitest, via a package named TinySpy, gives you the ability to spy on a function. Spying allows you to observe a function during the course of the code's execution and determine things such as: how many times it was invoked, what parameters were passed to it, the value it returned, and more.
Note: Spying on a function allows you to observe details about the function as your code is executed without modifying the target function or its behavior.
You may spy on an un-mocked function using vi.spyOn()
, however a mocked function with vi.fn()
has all of the spying functionalities available to it by default. Because Prisma Client has been mocked, every function should be capable of being spied on.
Below is a quick example of what a test might look like using a spy:
These spy functions are especially useful when you are attempting to ensure certain scenarios are triggered based on various inputs.
Why Vitest?
You may be curious about why this article focuses on Vitest as a testing framework rather than a more established and popular framework like Jest.
The reasoning behind this decision has to do with the different tools' compatibility with Node.js, specifically when dealing with Error
objects. Matteo Collina, a member of the Node.js Technical Steering Committee among other awesome achievements, describes this issue very well on a recent livestream of his.
The problem in a nutshell is that Jest cannot out of the box determine whether an error is an instance of the Error
class.
This may cause various unexpected problems as you write tests for different cases in your application.
How are they different?
Fortunately, for the most part every testing framework is very similar and the concepts transfer fairly seamlessly. For example, if you are accustomed to working with Jest and are considering a move to something like Vitest or node-tap
(another testing framework), the knowledge you already have will be very transferrable to the new technology.
Very minor adjustments will be required: things like function naming conventions and configuration.
Should you ever use Jest?
Yes! Jest is a fantastic tool written by very capable people. While Vitest may be the "best tool for the job" when testing a backend application in Node.js, Jest is still more than capable for testing frontend JavaScript applications.
Summary & What's next
In this article, you focused on the concepts of mocking and spying, both of which play a major role in unit testing an application. Specifically, you explored:
- What mocking is and why it is useful
- How to set up a project with Vitest and Prisma configured
- How to mock Prisma Client
- How to use a mocked Prisma Client instance
With this knowledge and context into the world of testing, you now have the tool set required to unit test an application. In the next article of this series you will do exactly that!
We hope you'll join along in the next parts of this series as we explore the various ways you can test your applications that use Prisma Client.
Don’t miss the next post!
Sign up for the Prisma Newsletter