Has anyone tried to unit test @resource_manager? ...
# ask-communitys
Seth Birkholz
12/16/2021, 4:53 PMHas anyone tried to unit test @resource_manager? I don’t see any documentation on that. I can test normal flows but this one stumps me.
k
Kevin Kho
12/16/2021, 4:58 PM
Hi @Seth Birkholz, this is the first time I’ve seen this asked. I don’t know an immadiate answer. What are you trying to test it for?
Kevin Kho
12/16/2021, 4:59 PM
Are you using it to spin up a cluster?
s
Seth Birkholz
12/16/2021, 5:02 PMno its just a wrapper around a database connection.
Seth Birkholz
12/16/2021, 5:03 PM Copy code
@resource_manager
class SnowflakeConnection:
def __init__(self, account, user, password, database, schema, warehouse, role):Seth Birkholz
12/16/2021, 5:03 PMI am trying to test the setup and cleanup methods in this class
k
Kevin Kho
12/16/2021, 5:04 PM
Will have to ask the team for ideas
s
Seth Birkholz
12/16/2021, 5:04 PMPlease do. I have been successful at testing everything else but this one looks tricky.
a
Anna Geller
12/16/2021, 5:08 PM
maybe this can help https://github.com/PrefectHQ/prefect/blob/master/tests/tasks/core/test_resource_manager.py
👀 1
s
Seth Birkholz
12/16/2021, 6:12 PMThis is great! Thanks for pointing it out. 🙂
👍 2
6 Views