update tests

jonaslagoni · Mar 5, 2024 · 321df0a · 321df0a
1 parent 1819b52
commit 321df0a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/test/generators/python/PythonConstrainer.spec.ts b/test/generators/python/PythonConstrainer.spec.ts
@@ -53,11 +53,18 @@ describe('PythonConstrainer', () => {
   describe('Any', () => {
     test('should render type', () => {
       const model = new ConstrainedAnyModel('test', undefined, {}, '');
+      const dependencyManager = new PythonDependencyManager(
+        PythonGenerator.defaultOptions
+      );
       const type = PythonDefaultTypeMapping.Any({
         constrainedModel: model,
-        ...defaultOptions
+        ...defaultOptions,
+        dependencyManager
       });
       expect(type).toEqual('Any');
+      expect(dependencyManager.dependencies).toEqual([
+        'from typing import Any'
+      ]);
     });
   });
   describe('Float', () => {

diff --git a/test/generators/python/PythonGenerator.spec.ts b/test/generators/python/PythonGenerator.spec.ts
@@ -131,12 +131,10 @@ describe('PythonGenerator', () => {
           }
         ]
       });
-      const expectedDependencies: string[] = [];
 
       const models = await generator.generate(doc);
       expect(models).toHaveLength(1);
       expect(models[0].result).toMatchSnapshot();
-      expect(models[0].dependencies).toEqual(expectedDependencies);
     });
     test('should work with empty objects', async () => {
       const doc = {
@@ -145,12 +143,10 @@ describe('PythonGenerator', () => {
         additionalProperties: false
       };
       generator = new PythonGenerator();
-      const expectedDependencies: string[] = [];
 
       const models = await generator.generate(doc);
       expect(models).toHaveLength(1);
       expect(models[0].result).toMatchSnapshot();
-      expect(models[0].dependencies).toEqual(expectedDependencies);
     });
   });
 });