Skip to content

Commit 835f0a5

Browse files
committed
feat: extend MembershipPivot class from PivotModel for improved attribute handling
1 parent d657145 commit 835f0a5

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/base/relationships.spec.ts

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,11 @@
1-
import { ArkormCollection, QueryBuilder, createPrismaDatabaseAdapter } from '../../src'
1+
import { ArkormCollection, PivotModel, QueryBuilder, createPrismaDatabaseAdapter } from '../../src'
22
import { Comment, Image, Post, Profile, Role, Tag, User, setupCoreRuntime } from './helpers/core-fixtures'
33
import { beforeEach, describe, expect, expectTypeOf, it, vi } from 'vitest'
44

55
import { createCoreClient } from './helpers/core-fixtures'
66

77
describe('Model relationships', () => {
8-
class MembershipPivot {
9-
public constructor(private readonly attributes: Record<string, unknown> = {}) {
10-
}
11-
8+
class MembershipPivot extends PivotModel {
129
public getAttribute (key: string): unknown {
1310
return this.attributes[key]
1411
}

0 commit comments

Comments
 (0)