Test coverage for UserMentionTooltipContainer

Author: smashwilson

lib/containers/user-mention-tooltip-container.js

@@ -4,7 +4,7 @@ import PropTypes from 'prop-types';
 
 import Octicon from '../atom/octicon';
 
-class UserMentionTooltip extends React.Component {
+export class BareUserMentionTooltipContainer extends React.Component {
   static propTypes = {
     repositoryOwner: PropTypes.shape({
       login: PropTypes.string.isRequired,
@@ -47,7 +47,7 @@ class UserMentionTooltip extends React.Component {
   }
 }
 
-export default createFragmentContainer(UserMentionTooltip, {
+export default createFragmentContainer(BareUserMentionTooltipContainer, {
   repositoryOwner: graphql`
     fragment userMentionTooltipContainer_repositoryOwner on RepositoryOwner {
       login

test/builder/graphql/user.js

@@ -1,6 +1,10 @@
-import {createSpecBuilderClass} from './base';
+import {createSpecBuilderClass, createConnectionBuilderClass, createUnionBuilderClass, defer} from './base';
 import {nextID} from '../id-sequence';
 
+const RepositoryBuilder = defer('../repository', 'RepositoryBuilder');
+
+export const RepositoryConnectionBuilder = createConnectionBuilderClass('Repository', RepositoryBuilder);
+
 export const UserBuilder = createSpecBuilderClass('User', {
   __typename: {default: 'User'},
   id: {default: nextID},
@@ -10,8 +14,35 @@ export const UserBuilder = createSpecBuilderClass('User', {
     const login = f.login || 'login';
     return `https://github.com/${login}`;
   }},
+  company: {default: 'GitHub'},
+  repositories: {linked: RepositoryConnectionBuilder},
+},
+'Node & Actor & RegistryPackageOwner & RegistryPackageSearch & ProjectOwner ' +
+  '& RepositoryOwner & UniformResourceLocatable',
+);
+
+export const OrganizationMemberConnectionBuilder = createConnectionBuilderClass('OrganizationMember', UserBuilder);
+
+export const OrganizationBuilder = createSpecBuilderClass('Organization', {
+  login: {default: 'someone'},
+  avatarUrl: {default: 'https://avatars3.githubusercontent.com/u/17565?s=32&v=4'},
+  repositories: {linked: RepositoryConnectionBuilder},
+  membersWithRole: {linked: OrganizationMemberConnectionBuilder},
+},
+'Node & Actor & RegistryPackageOwner & RegistryPackageSearch & ProjectOwner ' +
+  '& RepositoryOwner & UniformResourceLocatable & MemberStatusable',
+);
+
+export const RepositoryOwnerBuilder = createUnionBuilderClass('RepositoryOwner', {
+  beUser: UserBuilder,
+  beOrganization: OrganizationBuilder,
+  default: 'beUser',
 });
 
 export function userBuilder(...nodes) {
   return UserBuilder.onFragmentQuery(nodes);
 }
+
+export function organizationBuilder(...nodes) {
+  return OrganizationBuilder.onFragmentQuery(nodes);
+}

test/containers/user-mention-tooltip-container.test.js

@@ -0,0 +1,47 @@
+import React from 'react';
+import {shallow} from 'enzyme';
+
+import {BareUserMentionTooltipContainer} from '../../lib/containers/user-mention-tooltip-container';
+import {userBuilder, organizationBuilder} from '../builder/graphql/user';
+
+import ownerQuery from '../../lib/containers/__generated__/userMentionTooltipContainer_repositoryOwner.graphql';
+
+describe('UserMentionTooltipContainer', function() {
+  function buildApp(override = {}) {
+    const props = {
+      ...override,
+    };
+
+    return <BareUserMentionTooltipContainer {...props} />;
+  }
+
+  it('renders information about a User', function() {
+    const wrapper = shallow(buildApp({
+      repositoryOwner: userBuilder(ownerQuery)
+        .login('someone')
+        .avatarUrl('https://i.redd.it/03xcogvbr6v21.jpg')
+        .company('Infinity, Ltd.')
+        .repositories(conn => conn.totalCount(5))
+        .build(),
+    }));
+
+    assert.strictEqual(wrapper.find('img').prop('src'), 'https://i.redd.it/03xcogvbr6v21.jpg');
+    assert.isTrue(wrapper.find('span').someWhere(s => s.text() === 'Infinity, Ltd.'));
+    assert.isTrue(wrapper.find('span').someWhere(s => s.text() === '5 repositories'));
+  });
+
+  it('renders information about an Organization', function() {
+    const wrapper = shallow(buildApp({
+      repositoryOwner: organizationBuilder(ownerQuery)
+        .login('acme')
+        .avatarUrl('https://i.redd.it/eekf8onik0v21.jpg')
+        .membersWithRole(conn => conn.totalCount(10))
+        .repositories(conn => conn.totalCount(5))
+        .build(),
+    }));
+
+    assert.strictEqual(wrapper.find('img').prop('src'), 'https://i.redd.it/eekf8onik0v21.jpg');
+    assert.isTrue(wrapper.find('span').someWhere(s => s.text() === '10 members'));
+    assert.isTrue(wrapper.find('span').someWhere(s => s.text() === '5 repositories'));
+  });
+});