Handle repo change and status change events separately

Fix the SetActiveRepository method to fire PropertyChanged not StatusChanged.

Author: jcansdale

src/GitHub.InlineReviews/Services/PullRequestSessionManager.cs

@@ -67,12 +67,10 @@ public PullRequestSessionManager(
             this.modelServiceFactory = modelServiceFactory;
 
             Observable.FromEventPattern(teamExplorerContext, nameof(teamExplorerContext.StatusChanged))
-                .StartWith((EventPattern<object>)null)
                 .ObserveOn(RxApp.MainThreadScheduler)
-                .Subscribe(_ => RepoChanged(teamExplorerContext.ActiveRepository).Forget());
+                .Subscribe(_ => StatusChanged().Forget());
 
             teamExplorerContext.WhenAnyValue(x => x.ActiveRepository)
-                .Skip(1)
                 .ObserveOn(RxApp.MainThreadScheduler)
                 .Subscribe(x => RepoChanged(x).Forget());
         }
@@ -159,8 +157,8 @@ public async Task<IPullRequestSession> GetSession(IPullRequestModel pullRequest)
             {
                 // The branch for the PR was not previously marked with the PR number in the git
                 // config so we didn't pick up that the current branch is a PR branch. That has
-                // now been corrected, so call RepoChanged to make sure everything is up-to-date.
-                await RepoChanged(repository);
+                // now been corrected, so call StatusChanged to make sure everything is up-to-date.
+                await StatusChanged();
             }
 
             return await GetSessionInternal(pullRequest);
@@ -191,28 +189,23 @@ public PullRequestTextBufferInfo GetTextBufferInfo(ITextBuffer buffer)
 
         async Task RepoChanged(ILocalRepositoryModel localRepositoryModel)
         {
-            try
-            {
-                if (localRepositoryModel != repository)
-                {
-                    repository = localRepositoryModel;
-                    CurrentSession = null;
-                    sessions.Clear();
-
-                    if (localRepositoryModel == null)
-                    {
-                        return;
-                    }
+            repository = localRepositoryModel;
+            CurrentSession = null;
+            sessions.Clear();
 
-                    if (string.IsNullOrWhiteSpace(localRepositoryModel.CloneUrl))
-                    {
-                        return;
-                    }
-                }
+            if (localRepositoryModel != null)
+            {
+                await StatusChanged();
+            }
+        }
 
+        async Task StatusChanged()
+        {
+            try
+            {
                 var session = CurrentSession;
 
-                var pr = await service.GetPullRequestForCurrentBranch(localRepositoryModel).FirstOrDefaultAsync();
+                var pr = await service.GetPullRequestForCurrentBranch(repository).FirstOrDefaultAsync();
                 if (pr != null)
                 {
                     var changePR =
@@ -222,7 +215,7 @@ async Task RepoChanged(ILocalRepositoryModel localRepositoryModel)
                     if (changePR)
                     {
                         var modelService = await connectionManager.GetModelService(repository, modelServiceFactory);
-                        var pullRequest = await modelService?.GetPullRequest(pr.Item1, localRepositoryModel.Name, pr.Item2);
+                        var pullRequest = await modelService?.GetPullRequest(pr.Item1, repository.Name, pr.Item2);
                         if (pullRequest != null)
                         {
                             var newSession = await GetSessionInternal(pullRequest);

test/GitHub.InlineReviews.UnitTests/Services/PullRequestSessionManagerTests.cs

@@ -18,6 +18,7 @@
 using Microsoft.VisualStudio.Utilities;
 using NSubstitute;
 using NUnit.Framework;
+using System.ComponentModel;
 
 namespace GitHub.InlineReviews.UnitTests.Services
 {
@@ -1066,7 +1067,8 @@ static ITeamExplorerContext CreateTeamExplorerContext(ILocalRepositoryModel repo
         static void SetActiveRepository(ITeamExplorerContext teamExplorerContext, ILocalRepositoryModel localRepositoryModel)
         {
             teamExplorerContext.ActiveRepository.Returns(localRepositoryModel);
-            teamExplorerContext.StatusChanged += Raise.Event();
+            var eventArgs = new PropertyChangedEventArgs(nameof(teamExplorerContext.ActiveRepository));
+            teamExplorerContext.PropertyChanged += Raise.Event<PropertyChangedEventHandler>(teamExplorerContext, eventArgs);
         }
     }
 }