@@ -723,6 +723,12 @@ func TestEntitlements(t *testing.T) {
723723return true
724724})).
725725Return (int64 (175 ),nil )
726+ mDB .EXPECT ().
727+ GetWorkspaces (gomock .Any (),gomock .Any ()).
728+ Return ([]database.GetWorkspacesRow {},nil )
729+ mDB .EXPECT ().
730+ GetTemplatesWithFilter (gomock .Any (),gomock .Any ()).
731+ Return ([]database.Template {},nil )
726732
727733entitlements ,err := license .Entitlements (context .Background (),mDB ,1 ,0 ,coderdenttest .Keys ,all )
728734require .NoError (t ,err )
@@ -766,6 +772,7 @@ func TestLicenseEntitlements(t *testing.T) {
766772codersdk .FeatureUserRoleManagement :true ,
767773codersdk .FeatureAccessControl :true ,
768774codersdk .FeatureControlSharedPorts :true ,
775+ codersdk .FeatureWorkspaceExternalAgent :true ,
769776}
770777
771778legacyLicense := func ()* coderdenttest.LicenseOptions {
@@ -1109,6 +1116,32 @@ func TestLicenseEntitlements(t *testing.T) {
11091116assert .Equal (t ,int64 (200 ),* feature .Actual )
11101117},
11111118},
1119+ {
1120+ Name :"ExternalWorkspace" ,
1121+ Licenses : []* coderdenttest.LicenseOptions {
1122+ enterpriseLicense ().UserLimit (100 ),
1123+ },
1124+ Arguments : license.FeatureArguments {
1125+ ExternalWorkspaceCount :1 ,
1126+ },
1127+ AssertEntitlements :func (t * testing.T ,entitlements codersdk.Entitlements ) {
1128+ assert .Equal (t ,codersdk .EntitlementEntitled ,entitlements .Features [codersdk .FeatureWorkspaceExternalAgent ].Entitlement )
1129+ assert .True (t ,entitlements .Features [codersdk .FeatureWorkspaceExternalAgent ].Enabled )
1130+ },
1131+ },
1132+ {
1133+ Name :"ExternalTemplate" ,
1134+ Licenses : []* coderdenttest.LicenseOptions {
1135+ enterpriseLicense ().UserLimit (100 ),
1136+ },
1137+ Arguments : license.FeatureArguments {
1138+ ExternalTemplateCount :1 ,
1139+ },
1140+ AssertEntitlements :func (t * testing.T ,entitlements codersdk.Entitlements ) {
1141+ assert .Equal (t ,codersdk .EntitlementEntitled ,entitlements .Features [codersdk .FeatureWorkspaceExternalAgent ].Entitlement )
1142+ assert .True (t ,entitlements .Features [codersdk .FeatureWorkspaceExternalAgent ].Enabled )
1143+ },
1144+ },
11121145}
11131146
11141147for _ ,tc := range testCases {